Add statistics dashboard

2026-01-17 15:48:05 +01:00
parent 6b73ca4b21
commit 207c5bbf7a
2 changed files with 90 additions and 9 deletions
--- a/server/server.go
+++ b/server/server.go
@@ -1,6 +1,7 @@
 package main

 import (
+	"fmt"
 	"log"
 	"net/http"
 	"sync"
@@ -8,11 +9,50 @@ import (
 	"github.com/gorilla/websocket"
 )

-var upgrader = websocket.Upgrader{
-	CheckOrigin: func(r *http.Request) bool {
-		return true
-	},
-}
+var (
+	upgrader = websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool {
+			return true
+		},
+	}
+
+	dashboardHTML = `
+<!DOCTYPE html>
+<html lang="en">
+<head>
+	<meta charset="UTF-8">
+	<title>Clipboard On The Go - Stats</title>
+	<meta name="viewport" content="width=device-width, initial-scale=1.0">
+	<script src="https://cdn.tailwindcss.com"></script>
+</head>
+<body class="bg-gray-100 min-h-screen flex flex-col items-center justify-center">
+	<div class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4 w-full max-w-md">
+		<h1 class="text-2xl font-bold mb-4 text-center">Clipboard On The Go - Server Stats</h1>
+		<div class="mb-2 flex justify-between">
+			<span class="font-semibold">Connected Clients:</span>
+			<span id="clients" class="font-mono">0</span>
+		</div>
+		<div class="mb-2 flex justify-between">
+			<span class="font-semibold">Messages Relayed:</span>
+			<span id="messages" class="font-mono">0</span>
+		</div>
+	</div>
+	<script>
+		function updateStats() {
+			fetch('/stats/data')
+				.then(r => r.json())
+				.then(d => {
+					document.getElementById('clients').textContent = d.clients;
+					document.getElementById('messages').textContent = d.messages;
+				});
+		}
+		setInterval(updateStats, 1000);
+		updateStats();
+	</script>
+</body>
+</html>
+`
+)

 type Client struct {
 	conn           *websocket.Conn
@@ -26,6 +66,10 @@ type Server struct {
 	unregister chan *Client
 	broadcast  chan *Message
 	mu         sync.RWMutex
+
+	// Statistics
+	statsEnabled    bool
+	messagesRelayed uint64
 }

 type Message struct {
@@ -36,10 +80,11 @@ type Message struct {

 func newServer() *Server {
 	return &Server{
-		clients:    make(map[*Client]bool),
-		register:   make(chan *Client),
-		unregister: make(chan *Client),
-		broadcast:  make(chan *Message),
+		clients:      make(map[*Client]bool),
+		register:     make(chan *Client),
+		unregister:   make(chan *Client),
+		broadcast:    make(chan *Message),
+		statsEnabled: EmbeddedServerStats,
 	}
 }

@@ -74,6 +119,11 @@ func (s *Server) run() {
 				}
 			}
 			s.mu.RUnlock()
+			if s.statsEnabled {
+				s.mu.Lock()
+				s.messagesRelayed++
+				s.mu.Unlock()
+			}
 		}
 	}
 }
@@ -157,7 +207,26 @@ func main() {
 		handleWebSocket(server, w, r)
 	})

+	// Serve stats dashboard if enabled
+	if EmbeddedServerStats {
+		http.HandleFunc("/stats", func(w http.ResponseWriter, r *http.Request) {
+			w.Header().Set("Content-Type", "text/html; charset=utf-8")
+			fmt.Fprint(w, dashboardHTML)
+		})
+		http.HandleFunc("/stats/data", func(w http.ResponseWriter, r *http.Request) {
+			server.mu.RLock()
+			clients := len(server.clients)
+			messages := server.messagesRelayed
+			server.mu.RUnlock()
+			w.Header().Set("Content-Type", "application/json")
+			fmt.Fprintf(w, `{"clients":%d,"messages":%d}`, clients, messages)
+		})
+	}
+
 	log.Printf("Server starting on port %s", port)
+	if EmbeddedServerStats {
+		log.Printf("Stats dashboard enabled at /stats")
+	}
 	err := http.ListenAndServe(":"+port, nil)
 	if err != nil {
 		log.Fatal("ListenAndServe error:", err)