Docs

2021-11-15 20:23:10 -03:00
parent 217325edf0
commit a722ce0db4
3 changed files with 160 additions and 23 deletions
@@ -210,27 +210,21 @@ func (c *MQConn) Write(m *types.Msg) error {
 	return c.conn.WriteJSON(m)
 }

-func Serve() {
-	var upgrader = websocket.Upgrader{}
-	server := NewServer()
-	http.HandleFunc("/ws", func(writer http.ResponseWriter, request *http.Request) {
-		id := random.Str(32)
-		con, err := upgrader.Upgrade(writer, request, nil)
-		if err != nil {
-			http.Error(writer, err.Error(), http.StatusInternalServerError)
-			return
-		}
-		mqconn := server.Put(id, con)
-		con.SetCloseHandler(func(code int, text string) error {
-			mqconn.Close()
-			return nil
-		})
-		mqconn.Loop()
-	})
-	http.HandleFunc("/stats", func(writer http.ResponseWriter, request *http.Request) {
-		//server.Range(func(m *MQConn) bool {
-		//	m.
-		//	return true
-		//})
+var upgrader = websocket.Upgrader{}
+var server *MQServer = NewServer()
+
+func Serve(writer http.ResponseWriter, request *http.Request) {
+
+	id := random.Str(32)
+	con, err := upgrader.Upgrade(writer, request, nil)
+	if err != nil {
+		http.Error(writer, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	mqconn := server.Put(id, con)
+	con.SetCloseHandler(func(code int, text string) error {
+		mqconn.Close()
+		return nil
 	})
+	mqconn.Loop()
 }