mongo.go重构

utils/utils.go

@@ -60,40 +60,44 @@ func GetMonthToday(t time.Time, month int) time.Time {
 	return target.AddDate(0, 0, t.Day()-1)
 }
 func GetUserInfo(ctx context.Context) (user do.User, err error) {
-	redisAddr := g.Cfg().MustGet(ctx, "redis.default.address").String()
-	gft := gftoken.NewGfToken(
-		gftoken.WithCacheKey("gfToken:"),
-		gftoken.WithTimeout(20),
-		gftoken.WithMaxRefresh(10),
-		gftoken.WithMultiLogin(true),
-		//gftoken.WithExcludePaths(g.SliceStr{"/excludeDemo"}),
-		gftoken.WithGRedisConfig(&gredis.Config{
-			Address: redisAddr,
-			Db:      1,
-		}))
 	r := g.RequestFromCtx(ctx)
-	if r == nil {
-		return
-	}
-	// 解析 token
-	data, err := gft.ParseToken(g.RequestFromCtx(ctx))
-	if err != nil {
-		return user, gerror.Wrap(err, "token 解析失败")
-	}
+	if r != nil {
+		redisAddr := g.Cfg().MustGet(ctx, "redis.default.address").String()
+		gft := gftoken.NewGfToken(
+			gftoken.WithCacheKey("gfToken:"),
+			gftoken.WithTimeout(20),
+			gftoken.WithMaxRefresh(10),
+			gftoken.WithMultiLogin(true),
+			//gftoken.WithExcludePaths(g.SliceStr{"/excludeDemo"}),
+			gftoken.WithGRedisConfig(&gredis.Config{
+				Address: redisAddr,
+				Db:      1,
+			}))
+		// 解析 token
+		data, err := gft.ParseToken(g.RequestFromCtx(ctx))
+		if err != nil {
+			return user, gerror.Wrap(err, "token 解析失败")
+		}
 
-	// 检查 data 是否为 nil
-	if data == nil {
-		return user, gerror.New("token 数据为空")
-	}
+		// 检查 data 是否为 nil
+		if data == nil {
+			return user, gerror.New("token 数据为空")
+		}
 
-	// 检查 data.Data 是否为 nil
-	if data.Data == nil {
-		return user, gerror.New("用户信息为空")
-	}
+		// 检查 data.Data 是否为 nil
+		if data.Data == nil {
+			return user, gerror.New("用户信息为空")
+		}
 
-	dataMap := gconv.Map(data.Data)
-	user.UserName = dataMap["userName"]
-	user.TenantId = dataMap["tenantId"]
+		dataMap := gconv.Map(data.Data)
+		user.UserName = dataMap["userName"]
+		user.TenantId = dataMap["tenantId"]
+	} else {
+		user.TenantId = ctx.Value("tenantId")
+	}
+	if user.TenantId == nil {
+		return user, gerror.New("租户信息为空")
+	}
 	return
 }
 func OrderMap(m map[string]interface{}) map[string]interface{} {