Changeset 605 in code for trunk/server.go

Timestamp:

Oct 5, 2021, 5:13:53 PM (4 years ago)

Author:

contact

Message:

Add "server status" command

Right now, it prints the number of active users and number of
downstream connections.

File:

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

trunk/server.go

@@ -56 +56 @@
         Identd         *Identd // can be nil
 
-        db     Database
-        stopWG sync.WaitGroup
+        db        Database
+        stopWG    sync.WaitGroup
+        connCount int64 // atomic
 
         lock      sync.Mutex
@@ -166 +167 @@
 
 func (s *Server) handle(ic ircConn) {
+        atomic.AddInt64(&s.connCount, 1)
         id := atomic.AddUint64(&lastDownstreamID, 1)
         dc := newDownstreamConn(s, ic, id)
@@ -178 +180 @@
         }
         dc.Close()
+        atomic.AddInt64(&s.connCount, -1)
 }
 
@@ -250 +253 @@
         return params
 }
+
+type ServerStats struct {
+        Users       int
+        Downstreams int64
+}
+
+func (s *Server) Stats() *ServerStats {
+        var stats ServerStats
+        s.lock.Lock()
+        stats.Users = len(s.users)
+        s.lock.Unlock()
+        stats.Downstreams = atomic.LoadInt64(&s.connCount)
+        return &stats
+}