feat 模块增加after start

app/app.go

@@ -6,8 +6,10 @@ import (
 	"git.hlsq.asia/mmorpg/service-common/log"
 	"git.hlsq.asia/mmorpg/service-common/module"
 	"git.hlsq.asia/mmorpg/service-gateway/config"
+	"git.hlsq.asia/mmorpg/service-gateway/internal/global"
 	"git.hlsq.asia/mmorpg/service-gateway/internal/grpc_server"
 	"github.com/judwhite/go-svc"
+	"sync"
 )
 
 type Program struct {
@@ -23,8 +25,8 @@ func (p *Program) Init(_ svc.Environment) error {
 	p.moduleList = append(p.moduleList, (&module.DB{}).Bind(config.Get().DB))
 	p.moduleList = append(p.moduleList, &ModuleWebServer{})
 	p.moduleList = append(p.moduleList, &ModuleWebsocketServer{})
-	p.moduleList = append(p.moduleList, (&module.Grpc{}).Bind(grpc_server.NewServer(config.Get().Serve.Grpc)))
 	p.moduleList = append(p.moduleList, &ModuleLoginQueue{})
+	p.moduleList = append(p.moduleList, (&module.Grpc{}).Bind(grpc_server.NewServer(config.Get().Serve.Grpc)))
 	p.moduleList = append(p.moduleList, (&module.Prometheus{}).Bind(config.Get().Metric))
 	p.moduleList = append(p.moduleList, (&module.Tracer{}).Bind(config.Get().Metric, common.KeyDiscoverServiceNameGateway))
 	p.moduleList = append(p.moduleList, &module.Discover{})
@@ -42,11 +44,20 @@ func (p *Program) Init(_ svc.Environment) error {
 }
 
 func (p *Program) Start() error {
+	ready := &sync.WaitGroup{}
+	ready.Add(len(p.moduleList))
 	for _, m := range p.moduleList {
-		if err := m.Start(); err != nil {
+		if err := m.Start(ready); err != nil {
 			return err
 		}
 	}
+	ready.Wait()
+	for _, m := range p.moduleList {
+		if err := m.AfterStart(); err != nil {
+			return err
+		}
+	}
+	global.ServiceStatus = global.ServiceStatusReady
 
 	log.Infof(fmt.Sprintf("%v Start successful...", config.Get().App.Name))
 	return nil