wip: 0.2.0

1. websocket 连接,退出,消息
  2. 基本页面

internal/handler/local.go

@@ -0,0 +1,35 @@
+package handler
+
+import (
+	"github.com/gorilla/websocket"
+	"github.com/loveuer/nf"
+	"github.com/loveuer/nf/nft/log"
+	"github.com/loveuer/ushare/internal/controller"
+	"net/http"
+)
+
+func LocalRegistry() nf.HandlerFunc {
+	upgrader := websocket.Upgrader{
+		ReadBufferSize:  1024,
+		WriteBufferSize: 1024,
+		CheckOrigin: func(r *http.Request) bool {
+			return true
+		},
+	}
+
+	return func(c *nf.Ctx) error {
+
+		ip := c.IP(true)
+		ua := c.Get("User-Agent")
+
+		conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
+		if err != nil {
+			log.Error("LocalRegistry: failed to upgrade websocket connection, err = %s", err.Error())
+			return err
+		}
+
+		controller.RoomController.Register(conn, ip, ua)
+
+		return nil
+	}
+}