Changeset 708 in code for trunk/server.go

Timestamp:

Nov 17, 2021, 2:58:19 PM (4 years ago)

Author:

contact

Message:

Add basic active users and downstreams metrics

File:

• trunk/server.go (modified) (3 diffs)

trunk/server.go

@@ -16 +16 @@
 
         "github.com/prometheus/client_golang/prometheus"
+        "github.com/prometheus/client_golang/prometheus/promauto"
         "gopkg.in/irc.v3"
         "nhooyr.io/websocket"
@@ -112 +113 @@
 
 func (s *Server) Start() error {
+        s.registerMetrics()
+
         users, err := s.db.ListUsers(context.TODO())
         if err != nil {
@@ -124 +127 @@
 
         return nil
+}
+
+func (s *Server) registerMetrics() {
+        factory := promauto.With(s.MetricsRegistry)
+
+        factory.NewGaugeFunc(prometheus.GaugeOpts{
+                Name: "soju_users_active",
+                Help: "Current number of active users",
+        }, func() float64 {
+                s.lock.Lock()
+                n := len(s.users)
+                s.lock.Unlock()
+                return float64(n)
+        })
+
+        factory.NewGaugeFunc(prometheus.GaugeOpts{
+                Name: "soju_downstreams_active",
+                Help: "Current number of downstream connections",
+        }, func() float64 {
+                return float64(atomic.LoadInt64(&s.connCount))
+        })
 }