fix modules registeration

This commit is contained in:
medcl 2023-09-04 19:19:29 +08:00
parent 4093fcb3c1
commit 104462b49e
1 changed files with 13 additions and 13 deletions

26
main.go
View File

@ -61,16 +61,16 @@ func main() {
api := api2.GatewayAPI{} api := api2.GatewayAPI{}
modules := []module.Module{} modules := []module.ModuleItem{}
modules = append(modules, &stats.SimpleStatsModule{}) modules = append(modules, module.ModuleItem{Value: &stats.SimpleStatsModule{}, Priority: 1})
modules = append(modules, &elastic2.ElasticModule{}) modules = append(modules, module.ModuleItem{Value: &elastic2.ElasticModule{}, Priority: 1})
modules = append(modules, &queue2.DiskQueue{}) modules = append(modules, module.ModuleItem{Value: &queue2.DiskQueue{}, Priority: 1})
modules = append(modules, &redis.RedisModule{}) modules = append(modules, module.ModuleItem{Value: &redis.RedisModule{}, Priority: 1})
modules = append(modules, &pipeline.PipeModule{}) modules = append(modules, module.ModuleItem{Value: &pipeline.PipeModule{}, Priority: 1})
modules = append(modules, &task.TaskModule{}) modules = append(modules, module.ModuleItem{Value: &task.TaskModule{}, Priority: 1})
modules = append(modules, &metrics.MetricsModule{}) modules = append(modules, module.ModuleItem{Value: &metrics.MetricsModule{}, Priority: 1})
modules = append(modules, &security.Module{}) modules = append(modules, module.ModuleItem{Value: &security.Module{}, Priority: 1})
modules = append(modules, &agent.AgentModule{}) modules = append(modules, module.ModuleItem{Value: &agent.AgentModule{}, Priority: 100})
uiModule := &ui.UIModule{} uiModule := &ui.UIModule{}
@ -84,11 +84,11 @@ func main() {
if !global.Env().SetupRequired() { if !global.Env().SetupRequired() {
for _, v := range modules { for _, v := range modules {
module.RegisterSystemModule(v) module.RegisterModuleWithPriority(v.Value,v.Priority)
} }
} else { } else {
for _, v := range modules { for _, v := range modules {
v.Setup() v.Value.Setup()
} }
} }
@ -142,7 +142,7 @@ func main() {
if global.Env().SetupRequired() { if global.Env().SetupRequired() {
for _, v := range modules { for _, v := range modules {
v.Start() v.Value.Start()
} }
} }