61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package module
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"git.hlsq.asia/mmorpg/service-common/config"
|
|
"git.hlsq.asia/mmorpg/service-common/log"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"net/http"
|
|
"sync"
|
|
)
|
|
|
|
// Prometheus 普罗米修斯模块
|
|
type Prometheus struct {
|
|
DefaultModule
|
|
wg *sync.WaitGroup
|
|
server *http.Server
|
|
metricCfg *config.MetricConfig
|
|
}
|
|
|
|
func (m *Prometheus) Init() error {
|
|
m.wg = &sync.WaitGroup{}
|
|
return nil
|
|
}
|
|
|
|
func (m *Prometheus) Start(ready *sync.WaitGroup) error {
|
|
m.wg.Add(1)
|
|
go func() {
|
|
defer m.wg.Done()
|
|
m.server = &http.Server{
|
|
Addr: fmt.Sprintf("%v:%v", m.metricCfg.Prometheus.Address, m.metricCfg.Prometheus.Port),
|
|
Handler: promhttp.Handler(),
|
|
}
|
|
ready.Done()
|
|
if err := m.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
log.Errorf("prometheus server failed: %v", err.Error())
|
|
}
|
|
log.Infof("prometheus server stop.")
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (m *Prometheus) Stop() error {
|
|
if err := m.server.Shutdown(context.Background()); err != nil {
|
|
log.Errorf("stop prometheus server failed: %v", err)
|
|
}
|
|
m.wg.Wait()
|
|
return nil
|
|
}
|
|
|
|
func (m *Prometheus) Bind(data ...any) Module {
|
|
if data == nil || len(data) == 0 {
|
|
return m
|
|
}
|
|
if mc, ok := data[0].(*config.MetricConfig); ok {
|
|
m.metricCfg = mc
|
|
}
|
|
return m
|
|
}
|