0085831ec5
Since we have some perf issues from time to time it is good to have pprof debugger. Disabled by default.
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package metrics
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Service serves metrics.
|
|
type Service struct {
|
|
*http.Server
|
|
config Config
|
|
serviceType string
|
|
}
|
|
|
|
// Config config used for monitoring.
|
|
type Config struct {
|
|
Enabled bool `yaml:"Enabled"`
|
|
Address string `yaml:"Address"`
|
|
Port string `yaml:"Port"`
|
|
}
|
|
|
|
// Start runs http service with exposed endpoint on configured port.
|
|
func (ms *Service) Start() {
|
|
if ms.config.Enabled {
|
|
log.WithFields(log.Fields{
|
|
"endpoint": ms.Addr,
|
|
"service": ms.serviceType,
|
|
}).Info("service running")
|
|
err := ms.ListenAndServe()
|
|
if err != nil && err != http.ErrServerClosed {
|
|
log.Warnf("%s service couldn't start on configured port", ms.serviceType)
|
|
}
|
|
} else {
|
|
log.Infof("%s service hasn't started since it's disabled", ms.serviceType)
|
|
}
|
|
}
|
|
|
|
// ShutDown stops service.
|
|
func (ms *Service) ShutDown() {
|
|
log.WithFields(log.Fields{
|
|
"endpoint": ms.Addr,
|
|
"service": ms.serviceType,
|
|
}).Info("shutting down service")
|
|
err := ms.Shutdown(context.Background())
|
|
if err != nil {
|
|
log.Fatalf("can't shut down %s service", ms.serviceType)
|
|
}
|
|
}
|