feat 模块增加after start

This commit is contained in:
2026-01-20 12:11:26 +08:00
parent 5aeff7c786
commit c0f6cd90c6
11 changed files with 43 additions and 38 deletions

View File

@@ -14,6 +14,7 @@ import (
// ModuleWebsocketServer Websocket服务模块
type ModuleWebsocketServer struct {
module.DefaultModule
wg *sync.WaitGroup
server *websocket.WSServer
}
@@ -28,10 +29,11 @@ func (m *ModuleWebsocketServer) Init() error {
return nil
}
func (m *ModuleWebsocketServer) Start() error {
func (m *ModuleWebsocketServer) Start(ready *sync.WaitGroup) error {
m.wg.Add(1)
go func() {
defer m.wg.Done()
ready.Done()
_ = m.server.Run(
fmt.Sprintf("tcp4://0.0.0.0:%v", config.Get().Serve.Socket.Web.Port),
true,
@@ -55,7 +57,3 @@ func (m *ModuleWebsocketServer) Stop() error {
m.wg.Wait()
return nil
}
func (m *ModuleWebsocketServer) Bind(_ ...any) module.Module {
return m
}