增加了一些功能

main.go

@@ -1,25 +1,44 @@
 package main
 
 import (
+	"Embalming_Girl_Server/admin"
+	"Embalming_Girl_Server/db"
 	"Embalming_Girl_Server/handler"
 	"log"
 	"net/http"
 )
 
 func main() {
+	if err := db.Init(); err != nil {
+		log.Fatal("Database init failed:", err)
+	}
+	log.Println("Database initialized")
+
 	hub := handler.NewHub()
 	hub.InitTestRooms()
 
-	http.HandleFunc("/ws", hub.ServeWS)
-	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/ws", hub.ServeWS)
+	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
-		w.Write([]byte("Embalming Girl Server v0.1.0"))
+		w.Write([]byte("Embalming Girl Server v0.2.0"))
+	})
+
+	admin.Register(mux, func() map[string]interface{} {
+		var totalUsers int
+		db.DB.QueryRow("SELECT COUNT(*) FROM users").Scan(&totalUsers)
+		return map[string]interface{}{
+			"total_users":    totalUsers,
+			"active_rooms":   len(hub.RoomMgr.ListRooms()),
+			"online_clients": hub.OnlineCount(),
+		}
 	})
 
 	addr := ":8080"
 	log.Printf("Server started on %s", addr)
+	log.Printf("Admin panel: http://localhost%s/admin", addr)
 	log.Printf("Test rooms: TEST3, TEST4, TEST5, TEST6")
-	if err := http.ListenAndServe(addr, nil); err != nil {
+	if err := http.ListenAndServe(addr, mux); err != nil {
 		log.Fatal("ListenAndServe:", err)
 	}
 }