76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
|
|||
|
|
commonMongo "gitea.com/red-future/common/db/mongo"
|
|||
|
|
"go.mongodb.org/mongo-driver/v2/bson"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// MongoDAO MongoDB原生查询（不需要token验证）
|
|||
|
|
var MongoDAO = new(mongoDAO)
|
|||
|
|
|
|||
|
|
type mongoDAO struct{}
|
|||
|
|
|
|||
|
|
// FindOne 原生查询单条记录（不需要token验证）
|
|||
|
|
// 未找到记录时返回 nil error，调用方需检查 result 是否为零值
|
|||
|
|
func (m *mongoDAO) FindOne(ctx context.Context, filter bson.M, result interface{}, collectionName string) error {
|
|||
|
|
db := commonMongo.GetDB()
|
|||
|
|
collection := db.Collection(collectionName)
|
|||
|
|
|
|||
|
|
err := collection.FindOne(ctx, filter).Decode(result)
|
|||
|
|
if err != nil {
|
|||
|
|
if err.Error() == "mongo: no documents in result" {
|
|||
|
|
return nil // 未找到记录，返回nil而不是错误
|
|||
|
|
}
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InsertOne 原生插入单条记录（不需要token验证）
|
|||
|
|
func (m *mongoDAO) InsertOne(ctx context.Context, document interface{}, collectionName string) (interface{}, error) {
|
|||
|
|
db := commonMongo.GetDB()
|
|||
|
|
collection := db.Collection(collectionName)
|
|||
|
|
|
|||
|
|
result, err := collection.InsertOne(ctx, document)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return result.InsertedID, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateOne 原生更新单条记录（不需要token验证）
|
|||
|
|
// 返回 matchedCount（匹配到的记录数）和 modifiedCount（实际修改的记录数）
|
|||
|
|
func (m *mongoDAO) UpdateOne(ctx context.Context, filter bson.M, update bson.M, collectionName string) (matchedCount int64, modifiedCount int64, err error) {
|
|||
|
|
db := commonMongo.GetDB()
|
|||
|
|
collection := db.Collection(collectionName)
|
|||
|
|
|
|||
|
|
result, err := collection.UpdateOne(ctx, filter, update)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, 0, err
|
|||
|
|
}
|
|||
|
|
return result.MatchedCount, result.ModifiedCount, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateMany 原生批量更新记录（不需要token验证）
|
|||
|
|
func (m *mongoDAO) UpdateMany(ctx context.Context, filter bson.M, update bson.M, collectionName string) (matchedCount int64, modifiedCount int64, err error) {
|
|||
|
|
db := commonMongo.GetDB()
|
|||
|
|
collection := db.Collection(collectionName)
|
|||
|
|
|
|||
|
|
result, err := collection.UpdateMany(ctx, filter, update)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, 0, err
|
|||
|
|
}
|
|||
|
|
return result.MatchedCount, result.ModifiedCount, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Delete 原生删除记录（不需要token验证，用于回滚操作）
|
|||
|
|
func (m *mongoDAO) Delete(ctx context.Context, filter bson.M, collectionName string) error {
|
|||
|
|
db := commonMongo.GetDB()
|
|||
|
|
collection := db.Collection(collectionName)
|
|||
|
|
|
|||
|
|
_, err := collection.DeleteOne(ctx, filter)
|
|||
|
|
return err
|
|||
|
|
}
|