mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.6 KiB
65 lines
1.6 KiB
// Copyright 2023 ROC. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/Masterminds/semver/v3"
|
|
"github.com/fatih/color"
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
|
"github.com/rocboss/paopao-ce/internal/dao"
|
|
"github.com/rocboss/paopao-ce/internal/dao/cache"
|
|
"github.com/rocboss/paopao-ce/internal/metrics/prometheus"
|
|
"github.com/rocboss/paopao-ce/internal/metrics/statistics"
|
|
)
|
|
|
|
var (
|
|
_ Service = (*metricsService)(nil)
|
|
)
|
|
|
|
type metricsService struct {
|
|
*baseHttpService
|
|
}
|
|
|
|
func (s *metricsService) Name() string {
|
|
return "MetricsService"
|
|
}
|
|
|
|
func (s *metricsService) Version() *semver.Version {
|
|
return semver.MustParse("v0.1.0")
|
|
}
|
|
|
|
func (s *metricsService) OnInit() error {
|
|
s.registerRoute(s, nil)
|
|
return nil
|
|
}
|
|
|
|
func (s *metricsService) String() string {
|
|
return fmt.Sprintf("listen on %s\n", color.GreenString("http://%s:%s", conf.MetricsServerSetting.HttpIp, conf.MetricsServerSetting.HttpPort))
|
|
}
|
|
|
|
func newMetricsService() Service {
|
|
addr := conf.MetricsServerSetting.HttpIp + ":" + conf.MetricsServerSetting.HttpPort
|
|
server := httpServers.from(addr, func() *httpServer {
|
|
ds, wc, mc := dao.DataService(), cache.NewWebCache(), statistics.NewMetricCache()
|
|
mux := http.NewServeMux()
|
|
mux.Handle("/metrics", prometheus.NewHandler(ds, wc, mc))
|
|
return &httpServer{
|
|
baseServer: newBaseServe(),
|
|
server: &http.Server{
|
|
Addr: addr,
|
|
Handler: mux,
|
|
},
|
|
}
|
|
})
|
|
return &metricsService{
|
|
baseHttpService: &baseHttpService{
|
|
server: server,
|
|
},
|
|
}
|
|
}
|