pagerduty-exporter/metrics_team.go at main · vrtdev/pagerduty-exporter · GitHub

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package main

import (
	"log/slog"

	"github.com/PagerDuty/go-pagerduty"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/webdevops/go-common/prometheus/collector"
)

type MetricsCollectorTeam struct {
	collector.Processor

	prometheus struct {
		team       *prometheus.GaugeVec
		teamMember *prometheus.GaugeVec
	}
}

func (m *MetricsCollectorTeam) Setup(collector *collector.Collector) {
	m.Processor.Setup(collector)

	m.prometheus.team = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "pagerduty_team_info",
			Help: "PagerDuty team",
		},
		[]string{
			"teamID",
			"teamName",
			"teamUrl",
		},
	)
	m.Collector.RegisterMetricList("pagerduty_team_info", m.prometheus.team, true)

	m.prometheus.teamMember = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "pagerduty_team_member_info",
			Help: "PagerDuty team member information",
		},
		[]string{
			"teamID", "userID", "role",
		},
	)
	m.Collector.RegisterMetricList("pagerduty_team_member_info", m.prometheus.teamMember, true)
}

func (m *MetricsCollectorTeam) Reset() {
}

func (m *MetricsCollectorTeam) Collect(callback chan<- func()) {
	listOpts := pagerduty.ListTeamOptions{}
	listOpts.Limit = PagerdutyListLimit
	listOpts.Offset = 0

	teamMetricList := m.Collector.GetMetricList("pagerduty_team_info")
	teamMembersMetricList := m.Collector.GetMetricList("pagerduty_team_member_info")

	for {
		m.Logger().Debug("fetch teams", slog.Uint64("offset", uint64(listOpts.Offset)), slog.Uint64("limit", uint64(listOpts.Limit)))

		list, err := PagerDutyClient.ListTeamsWithContext(m.Context(), listOpts)
		PrometheusPagerDutyApiCounter.WithLabelValues("ListTeams").Inc()

		if err != nil {
			panic(err)
		}

		for _, team := range list.Teams {
			teamMetricList.AddInfo(prometheus.Labels{
				"teamID":   team.ID,
				"teamName": team.Name,
				"teamUrl":  team.HTMLURL,
			})

			members, err := PagerDutyClient.ListTeamMembersPaginated(m.Context(), team.ID)
			PrometheusPagerDutyApiCounter.WithLabelValues("ListTeamMemberships").Inc()
			if err != nil {
				m.Logger().Error("error fetching team members", slog.String("team", team.ID), slog.Any("error", err))
				break
			}
			for _, member := range members {
				teamMembersMetricList.AddInfo(prometheus.Labels{
					"teamID": team.ID,
					"userID": member.User.ID,
					"role":   member.Role,
				})
			}
		}

		listOpts.Offset += list.Limit
		if stopPagerdutyPaging(list.APIListObject) {
			break
		}
	}
}