76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package util
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
"gitea.com/red-future/common/beans"
|
||
"gitea.com/red-future/common/db/mongo"
|
||
"gitea.com/red-future/common/redis"
|
||
"gitea.com/red-future/common/utils"
|
||
"github.com/gogf/gf/v2/errors/gerror"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/util/gconv"
|
||
"go.mongodb.org/mongo-driver/v2/bson"
|
||
)
|
||
|
||
// GetTenantInfo 获取租户信息
|
||
// 优先从 token 获取，失败则从请求参数 customerServiceId 查询 customer_service_account 表
|
||
func GetTenantInfo(ctx context.Context) (user *beans.User, err error) {
|
||
// 1. 优先从 token 获取
|
||
user, err = utils.GetUserInfo(ctx)
|
||
if err == nil {
|
||
return
|
||
}
|
||
|
||
// 2. token 获取失败，尝试从请求参数或context获取 accountName
|
||
var accountName string
|
||
|
||
// 2.1 尝试从request获取（HTTP请求场景）
|
||
req := g.RequestFromCtx(ctx)
|
||
if req != nil {
|
||
accountName = req.Get("accountName").String()
|
||
if accountName == "" {
|
||
accountName = req.Get("account_name").String()
|
||
}
|
||
}
|
||
|
||
// 2.2 request不存在或未获取到，尝试从context.Value获取（WebSocket场景）
|
||
if accountName == "" {
|
||
if val := ctx.Value("accountName"); val != nil {
|
||
if str, ok := val.(string); ok {
|
||
accountName = str
|
||
}
|
||
}
|
||
}
|
||
|
||
if accountName == "" {
|
||
return user, gerror.New("无法获取租户信息：无 token 且无 accountName 参数")
|
||
}
|
||
|
||
// 3. 先查Redis缓存（accountName -> tenantId映射）
|
||
cacheKey := fmt.Sprintf("tenant:account:%s", accountName)
|
||
cached, cacheErr := redis.RedisClient().Get(ctx, cacheKey)
|
||
if cacheErr == nil && !g.IsEmpty(cached) {
|
||
user.TenantId = gconv.Uint64(cached.Interface())
|
||
user.UserName = accountName
|
||
return user, nil
|
||
}
|
||
|
||
// 4. 缓存未命中，查询 customer_service_account 表
|
||
filter := bson.M{"accountName": accountName, "isDeleted": false}
|
||
var account struct {
|
||
TenantId uint64 `bson:"tenantId"`
|
||
}
|
||
if findErr := mongo.GetDB().Collection("customer_service_account").FindOne(ctx, filter).Decode(&account); findErr != nil {
|
||
return user, gerror.Newf("通过 accountName 查询租户失败: %v", findErr)
|
||
}
|
||
|
||
// 5. 写入缓存（3分钟过期，避免数据长时间不一致）
|
||
redis.RedisClient().SetEX(ctx, cacheKey, account.TenantId, 180)
|
||
|
||
user.TenantId = account.TenantId
|
||
user.UserName = accountName
|
||
return user, nil
|
||
}
|