first commit

middleware/check_login.go

@@ -0,0 +1,25 @@
+package middleware
+
+import (
+	"log/slog"
+	"slgserver/constant"
+	"slgserver/net"
+)
+
+func CheckLogin() net.MiddlewareFunc {
+	return func(next net.HandlerFunc) net.HandlerFunc {
+		return func(req *net.WsMsgReq, rsp *net.WsMsgRsp) {
+
+			_, err := req.Conn.GetProperty("uid")
+			if err != nil {
+		slog.Warn("connect not found uid",
+			"msgName", req.Body.Name)
+				rsp.Body.Code = constant.UserNotInConnect
+				req.Conn.Push("account.pleaseLogin", nil)
+				return
+			}
+
+			next(req, rsp)
+		}
+	}
+}

middleware/check_rid.go

@@ -0,0 +1,23 @@
+package middleware
+
+import (
+	"log/slog"
+	"slgserver/constant"
+	"slgserver/net"
+)
+
+func CheckRId() net.MiddlewareFunc {
+	return func(next net.HandlerFunc) net.HandlerFunc {
+		return func(req *net.WsMsgReq, rsp *net.WsMsgRsp) {
+
+			_, err := req.Conn.GetProperty("rid")
+			if err != nil {
+				rsp.Body.Code = constant.RoleNotInConnect
+		slog.Warn("connect not found role",
+			"msgName", req.Body.Name)
+				return
+			}
+			next(req, rsp)
+		}
+	}
+}

middleware/check_role.go

@@ -0,0 +1,23 @@
+package middleware
+
+import (
+	"log/slog"
+	"slgserver/constant"
+	"slgserver/net"
+)
+
+func CheckRole() net.MiddlewareFunc {
+	return func(next net.HandlerFunc) net.HandlerFunc {
+		return func(req *net.WsMsgReq, rsp *net.WsMsgRsp) {
+
+			_, err := req.Conn.GetProperty("role")
+			if err != nil {
+				rsp.Body.Code = constant.RoleNotInConnect
+		slog.Warn("connect not found role",
+			"msgName", req.Body.Name)
+				return
+			}
+			next(req, rsp)
+		}
+	}
+}

middleware/elapsed_time.go

@@ -0,0 +1,23 @@
+package middleware
+
+import (
+	"fmt"
+	"log/slog"
+	"slgserver/net"
+	"time"
+)
+
+func ElapsedTime() net.MiddlewareFunc {
+	return func(next net.HandlerFunc) net.HandlerFunc {
+		return func(req *net.WsMsgReq, rsp *net.WsMsgRsp) {
+			bt := time.Now().UnixNano()
+			next(req, rsp)
+			et := time.Now().UnixNano()
+			diff := (et - bt)/int64(time.Millisecond)
+
+		slog.Info("ElapsedTime:",
+			"msgName", req.Body.Name,
+			"cost", fmt.Sprintf("%dms", diff))
+		}
+	}
+}

middleware/log.go

@@ -0,0 +1,20 @@
+package middleware
+
+import (
+	"fmt"
+	"log/slog"
+	"slgserver/net"
+)
+
+func Log() net.MiddlewareFunc {
+	return func(next net.HandlerFunc) net.HandlerFunc {
+		return func(req *net.WsMsgReq, rsp *net.WsMsgRsp) {
+
+		slog.Info("client req",
+			"msgName", req.Body.Name,
+			"data", fmt.Sprintf("%v", req.Body.Msg))
+
+			next(req, rsp)
+		}
+	}
+}