38 lines
924 B
Go
38 lines
924 B
Go
package metrics
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/shared/mlog"
|
|
)
|
|
|
|
// Service prometheus to run the server.
|
|
type Service struct {
|
|
*http.Server
|
|
}
|
|
|
|
// NewMetricsServer factory method to create a new prometheus server.
|
|
func NewMetricsServer(address string, metricsService *Metrics, logger *mlog.Logger) *Service {
|
|
return &Service{
|
|
&http.Server{
|
|
Addr: address,
|
|
Handler: promhttp.HandlerFor(metricsService.registry, promhttp.HandlerOpts{
|
|
ErrorLog: logger.StdLogger(mlog.LvlError),
|
|
}),
|
|
},
|
|
}
|
|
}
|
|
|
|
// Run will start the prometheus server.
|
|
func (h *Service) Run() error {
|
|
return errors.Wrap(h.Server.ListenAndServe(), "prometheus ListenAndServe")
|
|
}
|
|
|
|
// Shutdown will shutdown the prometheus server.
|
|
func (h *Service) Shutdown() error {
|
|
return errors.Wrap(h.Server.Close(), "prometheus Close")
|
|
}
|