优化系统

server/middleware/jwt.go

@@ -4,7 +4,6 @@ import (
 	"strings"
 
 	"server/models"
-	"server/services"
 
 	"github.com/beego/beego/v2/server/web"
 	"github.com/beego/beego/v2/server/web/context"
@@ -61,17 +60,16 @@ func JWTAuthMiddleware() web.FilterFunc {
 			return
 		}
 
-	// 将用户信息存储在上下文
-	ctx.Input.SetData("userId", claims.UserID)
-	ctx.Input.SetData("username", claims.Username)
-	ctx.Input.SetData("tenantId", claims.TenantId)
-	
-	// 判断用户类型：检查userId是否在员工表中
-	// 如果userId在yz_tenant_employees表中存在，则为员工登录；否则为用户登录
-	userType := "user"
-	if services.IsEmployee(claims.UserID) {
-		userType = "employee"
+		// 将用户信息存储在上下文
+		ctx.Input.SetData("userId", claims.UserID)
+		ctx.Input.SetData("username", claims.Username)
+		ctx.Input.SetData("tenantId", claims.TenantId)
+
+		// 从token中获取用户类型（如果token中没有，则默认为"user"）
+		userType := claims.UserType
+		if userType == "" {
+			userType = "user"
+		}
+		ctx.Input.SetData("userType", userType)
 	}
-	ctx.Input.SetData("userType", userType)
-}
 }