feat 模块增加after start
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user