red-future/common
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 19 Wiki Activity

mongo.go重构

Browse Source
This commit is contained in:
张斌
2025-12-30 10:52:12 +08:00
parent 7d05387104
commit f06e050d78
2 changed files with 48 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
mongo/mongo.go
View File

@@ -11,7 +11,6 @@ import (
"github.com/gogf/gf/v2/container/gvar"
"gitee.com/red-future---jilin-g/common/consts"
"gitee.com/red-future---jilin-g/common/do"
"gitee.com/red-future---jilin-g/common/redis"
"gitee.com/red-future---jilin-g/common/utils"
"github.com/gogf/gf/v2/errors/gerror"
@@ -99,6 +98,13 @@ func connect() error {
return nil
}
// GetDB 获取 MongoDB 数据库实例
func GetDB() *mongo.Database {
mu.RLock()
defer mu.RUnlock()
return db
}
// healthCheck 健康检查协程
func healthCheck() {
ticker := time.NewTicker(30 * time.Second)
@@ -261,76 +267,12 @@ func oneOptionsToMap(ctx context.Context, opts ...options.Lister[options.FindOne
return
}
// getTenantInfo 获取租户信息
// 优先从 token 获取，失败则从请求参数 customerServiceId 查询 customer_service_account 表
func getTenantInfo(ctx context.Context) (user do.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()
}
// 兼容旧参数名
if accountName == "" {
accountName = req.Get("customerServiceId").String()
}
if accountName == "" {
accountName = req.Get("customer_service_id").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 == "" {
if val := ctx.Value("customerServiceId"); val != nil {
if str, ok := val.(string); ok {
accountName = str
}
}
}
}
if accountName == "" {
return user, gerror.New("无法获取租户信息：无 token 且无 accountName 参数")
}
// 3. 直接查询 customer_service_account 表获取 tenantId
filter := bson.M{"accountName": accountName, "isDeleted": false}
var account struct {
TenantId interface{} `bson:"tenantId"`
}
if findErr := db.Collection("customer_service_account").FindOne(ctx, filter).Decode(&account); findErr != nil {
return user, gerror.Newf("通过 accountName 查询租户失败: %v", findErr)
}
user.TenantId = account.TenantId
user.UserName = accountName
err = nil // 清空之前从token获取时的错误
return
}
// Find 查询多条记录
func (m *MongoDB) Find(ctx context.Context, filter bson.M, result interface{}, collection string, opts ...options.Lister[options.FindOptions]) (err error) {
if err = utils.ValidStructPtr(result); err != nil {
return
}
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}
@@ -379,7 +321,7 @@ func (m *MongoDB) FindOne(ctx context.Context, filter bson.M, result interface{}
if err = utils.ValidStructPtr(result); err != nil {
return
}
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}
@@ -456,7 +398,7 @@ func (m *MongoDB) Delete(ctx context.Context, filter bson.M, collection string,
err = gerror.New("缺少查询条件")
return
}
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}
@@ -477,7 +419,7 @@ func (m *MongoDB) Update(ctx context.Context, filter bson.M, update bson.M, coll
return
}
filter["isDeleted"] = false
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}
@@ -557,7 +499,7 @@ func (m *MongoDB) SaveOrUpdate(ctx context.Context, filter []bson.M, update []bs
err = gerror.New("查询条件和更新数据的数量必须一致")
return
}
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}
@@ -622,7 +564,7 @@ func (m *MongoDB) SaveOrUpdate(ctx context.Context, filter []bson.M, update []bs
// Insert 插入多条记录
func (m *MongoDB) Insert(ctx context.Context, documents []interface{}, collection string, opts ...options.Lister[options.InsertManyOptions]) (ids []interface{}, err error) {
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}
@@ -655,7 +597,7 @@ func (m *MongoDB) Insert(ctx context.Context, documents []interface{}, collectio
// Count 查询总数
func (m *MongoDB) Count(ctx context.Context, filter bson.M, collection string) (count int64, err error) {
user, err := getTenantInfo(ctx)
user, err := utils.GetUserInfo(ctx)
if err != nil {
return
}