Initial QSfera import

Server/vendor/github.com/libregraph/idm/server/metrics.go

@@ -0,0 +1,125 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright 2021 The LibreGraph Authors.
+ */
+
+package server
+
+import (
+	"github.com/libregraph/idm/pkg/ldapserver"
+	"github.com/prometheus/client_golang/prometheus"
+)
+
+const (
+	metricsSubsystemLDAPServer = "ldapserver"
+)
+
+// MustRegister registers all rtm metrics with the provided registerer and
+// panics upon the first registration that causes an error.
+func MustRegister(reg prometheus.Registerer, cs ...prometheus.Collector) {
+	reg.MustRegister(cs...)
+}
+
+type ldapServerCollector struct {
+	stats *ldapserver.Stats
+
+	connsTotalDesc   *prometheus.Desc
+	connsCurrentDesc *prometheus.Desc
+	connsMaxDesc     *prometheus.Desc
+
+	bindsDesc   *prometheus.Desc
+	unbindsDesc *prometheus.Desc
+	searchesDsc *prometheus.Desc
+}
+
+func NewLDAPServerCollector(s *ldapserver.Server) prometheus.Collector {
+	return &ldapServerCollector{
+		stats: s.Stats,
+
+		connsTotalDesc: prometheus.NewDesc(
+			prometheus.BuildFQName("", metricsSubsystemLDAPServer, "connections_total"),
+			"Total number of incoming LDAP connections",
+			nil,
+			nil,
+		),
+		connsCurrentDesc: prometheus.NewDesc(
+			prometheus.BuildFQName("", metricsSubsystemLDAPServer, "connections_current"),
+			"Current number of concurrent established incoming LDAP connections",
+			nil,
+			nil,
+		),
+		connsMaxDesc: prometheus.NewDesc(
+			prometheus.BuildFQName("", metricsSubsystemLDAPServer, "connections_max"),
+			"Maximum number of concurrent established incoming LDAP connections",
+			nil,
+			nil,
+		),
+		bindsDesc: prometheus.NewDesc(
+			prometheus.BuildFQName("", metricsSubsystemLDAPServer, "binds_total"),
+			"Total number of incoming LDAP bind requests",
+			nil,
+			nil,
+		),
+		unbindsDesc: prometheus.NewDesc(
+			prometheus.BuildFQName("", metricsSubsystemLDAPServer, "unbinds_total"),
+			"Total number of incoming LDAP unbind requests",
+			nil,
+			nil,
+		),
+		searchesDsc: prometheus.NewDesc(
+			prometheus.BuildFQName("", metricsSubsystemLDAPServer, "searches_total"),
+			"Total number of incoming LDAP search requests",
+			nil,
+			nil,
+		),
+	}
+}
+
+// Describe is implemented with DescribeByCollect. That's possible because the
+// Collect method will always return the same two metrics with the same two
+// descriptors.
+func (lc *ldapServerCollector) Describe(ch chan<- *prometheus.Desc) {
+	prometheus.DescribeByCollect(lc, ch)
+}
+
+// Collect first gathers the associated managers collectors managers data. Then
+// it creates constant metrics based on the returned data.
+func (lc *ldapServerCollector) Collect(ch chan<- prometheus.Metric) {
+	stats := lc.stats.Clone()
+
+	ch <- prometheus.MustNewConstMetric(
+		lc.connsTotalDesc,
+		prometheus.CounterValue,
+		float64(stats.Conns),
+	)
+
+	ch <- prometheus.MustNewConstMetric(
+		lc.connsCurrentDesc,
+		prometheus.GaugeValue,
+		float64(stats.ConnsCurrent),
+	)
+
+	ch <- prometheus.MustNewConstMetric(
+		lc.connsMaxDesc,
+		prometheus.CounterValue,
+		float64(stats.ConnsMax),
+	)
+
+	ch <- prometheus.MustNewConstMetric(
+		lc.bindsDesc,
+		prometheus.CounterValue,
+		float64(stats.Binds),
+	)
+
+	ch <- prometheus.MustNewConstMetric(
+		lc.unbindsDesc,
+		prometheus.CounterValue,
+		float64(stats.Unbinds),
+	)
+
+	ch <- prometheus.MustNewConstMetric(
+		lc.searchesDsc,
+		prometheus.CounterValue,
+		float64(stats.Searches),
+	)
+}