feat 模块增加after start

This commit is contained in:
2026-01-20 12:11:28 +08:00
parent c293f9688a
commit 95ccbb0676
6 changed files with 21 additions and 30 deletions

View File

@@ -12,6 +12,7 @@ import (
// ModulePprof pprof模块
type ModulePprof struct {
module.DefaultModule
wg *sync.WaitGroup
server *http.Server
}
@@ -21,13 +22,14 @@ func (m *ModulePprof) Init() error {
return nil
}
func (m *ModulePprof) Start() error {
func (m *ModulePprof) Start(ready *sync.WaitGroup) error {
m.wg.Add(1)
go func() {
defer m.wg.Done()
m.server = &http.Server{
Addr: "localhost:6060",
}
ready.Done()
if err := m.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
log.Errorf("pprof server failed: %v", err.Error())
}
@@ -43,7 +45,3 @@ func (m *ModulePprof) Stop() error {
m.wg.Wait()
return nil
}
func (m *ModulePprof) Bind(_ ...any) module.Module {
return m
}