service/metrics/service.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

package metrics

import (
	"context"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"net/http"
)

// ID declares public service name.
const ID = "metrics"

// Service to manage application metrics using Prometheus.
type Service struct {
	cfg  *Config
	http *http.Server
}

// Init service.
func (s *Service) Init(cfg *Config) (bool, error) {
	s.cfg = cfg
	return true, nil
}

// Enabled indicates that server is able to collect metrics.
func (s *Service) Enabled() bool {
	return s.cfg != nil
}

// Register new prometheus collector.
func (s *Service) Register(c prometheus.Collector) error {
	return prometheus.Register(c)
}

// MustRegister registers new collector or fails with panic.
func (s *Service) MustRegister(c prometheus.Collector) {
	if err := prometheus.Register(c); err != nil {
		panic(err)
	}
}

// Serve prometheus metrics service.
func (s *Service) Serve() error {
	s.http = &http.Server{Addr: s.cfg.Address, Handler: promhttp.Handler()}

	return s.http.ListenAndServe()
}

// Stop prometheus metrics service.
func (s *Service) Stop() {
	// gracefully stop server
	s.http.Shutdown(context.Background())
}