feat: implement user counting and return on /online

common/stats.go

@@ -0,0 +1,12 @@
+package common
+
+var OnlineUsers int
+
+func init(){ 
+    OnlineUsers = 0 
+}
+
+func OnlineStatUpdate(t int) {
+    OnlineUsers += t
+}
+

gpcm/main.go

@@ -122,6 +122,7 @@ func handleRequest(conn net.Conn) {
 	conn.Write([]byte(payload))
 
 	logging.Notice(session.ModuleName, "Connection established from", conn.RemoteAddr())
+  common.OnlineStatUpdate(1)
 
 	// Here we go into the listening loop
 	for {
@@ -132,10 +133,12 @@ func handleRequest(conn net.Conn) {
 			if errors.Is(err, io.EOF) {
 				// Client closed connection, terminate.
 				logging.Notice(session.ModuleName, "Client closed connection")
+        common.OnlineStatUpdate(-1)
 				return
 			}
 
 			logging.Error(session.ModuleName, "Connection lost")
+      common.OnlineStatUpdate(-1)
 			return
 		}
 

nas/route.go

@@ -10,6 +10,7 @@ import (
 	"strings"
 	"wwfc/logging"
 	"wwfc/sake"
+  "wwfc/common"
 )
 
 type Route struct {
@@ -75,10 +76,9 @@ func (route *Route) Handle() http.Handler {
 
     // Check for /online
 		if strings.HasPrefix(r.URL.String(), "/online") {
-      logging.Notice("test")
     	w.Header().Set("Content-Type", "text/plain")
-      w.Header().Set("Content-Length", strconv.Itoa(len("Success")))
-    	w.Write([]byte("Success"))
+      w.Header().Set("Content-Length", strconv.Itoa(len(strconv.Itoa(common.OnlineUsers))))
+    	w.Write([]byte(strconv.Itoa(common.OnlineUsers))) 
 			return
 		}