mongo.go重构
This commit is contained in:
@@ -13,16 +13,9 @@ import (
|
||||
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||
)
|
||||
|
||||
var AdPosition = &adPosition{
|
||||
NoCache: true,
|
||||
}
|
||||
var AdPosition = &adPosition{}
|
||||
|
||||
type adPosition struct {
|
||||
NoCache bool
|
||||
}
|
||||
|
||||
func (d *adPosition) SetNoCache() {
|
||||
AdPosition.NoCache = true
|
||||
}
|
||||
|
||||
// Insert 插入广告位
|
||||
@@ -36,7 +29,7 @@ func (d *adPosition) Insert(ctx context.Context, adPosition *entity.AdPosition)
|
||||
g.Log().Infof(ctx, "获取到stream消息: %v", streamMsg)
|
||||
}
|
||||
|
||||
_, err = mongo.Insert(ctx, []interface{}{adPosition}, entity.AdPositionCollection)
|
||||
_, err = mongo.DB().Insert(ctx, []interface{}{adPosition}, entity.AdPositionCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -124,7 +117,7 @@ func (d *adPosition) Update(ctx context.Context, req *dto.UpdateAdPositionReq) (
|
||||
|
||||
if len(updateFields) > 0 {
|
||||
update := bson.M{"$set": updateFields}
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdPositionCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
|
||||
}
|
||||
return
|
||||
}
|
||||
@@ -138,7 +131,7 @@ func (d *adPosition) UpdateStatus(ctx context.Context, id, status string) (err e
|
||||
filter := bson.M{"_id": objectId}
|
||||
update := bson.M{"$set": bson.M{"status": status}}
|
||||
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdPositionCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -151,7 +144,7 @@ func (d *adPosition) UpdateStatistics(ctx context.Context, id string, stats map[
|
||||
filter := bson.M{"_id": objectId}
|
||||
update := bson.M{"$set": stats}
|
||||
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdPositionCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -164,7 +157,7 @@ func (d *adPosition) GetOne(ctx context.Context, id string) (adPosition *entity.
|
||||
filter := bson.M{"_id": objectId}
|
||||
|
||||
adPosition = &entity.AdPosition{}
|
||||
err = mongo.FindOne(ctx, d.NoCache, filter, adPosition, entity.AdPositionCollection)
|
||||
err = mongo.DB().FindOne(ctx, filter, adPosition, entity.AdPositionCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -173,7 +166,7 @@ func (d *adPosition) GetByCode(ctx context.Context, code string) (adPosition *en
|
||||
filter := bson.M{"positionCode": code}
|
||||
|
||||
adPosition = &entity.AdPosition{}
|
||||
err = mongo.FindOne(ctx, d.NoCache, filter, adPosition, entity.AdPositionCollection)
|
||||
err = mongo.DB().FindOne(ctx, filter, adPosition, entity.AdPositionCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -215,7 +208,7 @@ func (d *adPosition) buildListFilter(req *dto.ListAdPositionReq) bson.M {
|
||||
|
||||
// checkTotalCount 检查总数
|
||||
func (d *adPosition) checkTotalCount(ctx context.Context, filter bson.M) (total int64, err error) {
|
||||
total, err = mongo.Count(ctx, d.NoCache, filter, entity.AdPositionCollection)
|
||||
total, err = mongo.DB().Count(ctx, filter, entity.AdPositionCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -248,7 +241,7 @@ func (d *adPosition) List(ctx context.Context, req *dto.ListAdPositionReq) (list
|
||||
|
||||
opts := options.Find().SetLimit(limit).SetSkip(skip).SetSort(sort)
|
||||
|
||||
err = mongo.Find(ctx, d.NoCache, filter, &list, entity.AdPositionCollection, opts)
|
||||
err = mongo.DB().Find(ctx, filter, &list, entity.AdPositionCollection, opts)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -260,6 +253,6 @@ func (d *adPosition) GetAvailableAdPositions(ctx context.Context) (list []*entit
|
||||
|
||||
opts := options.Find().SetSort(bson.M{"createdAt": -1})
|
||||
|
||||
err = mongo.Find(ctx, d.NoCache, filter, &list, entity.AdPositionCollection, opts)
|
||||
err = mongo.DB().Find(ctx, filter, &list, entity.AdPositionCollection, opts)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -4,46 +4,40 @@ import (
|
||||
"context"
|
||||
|
||||
"cid/model/entity"
|
||||
|
||||
"gitee.com/red-future---jilin-g/common/mongo"
|
||||
"go.mongodb.org/mongo-driver/v2/bson"
|
||||
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||
)
|
||||
|
||||
var AdSource = &adSourceDao{
|
||||
NoCache: true,
|
||||
}
|
||||
var AdSource = &adSourceDao{}
|
||||
|
||||
type adSourceDao struct {
|
||||
NoCache bool
|
||||
}
|
||||
|
||||
func (d *adSourceDao) SetNoCache() {
|
||||
AdSource.NoCache = true
|
||||
}
|
||||
|
||||
// GetByName 根据名称获取广告源
|
||||
func (d *adSourceDao) GetByName(ctx context.Context, name string) (adSource *entity.AdSource, err error) {
|
||||
err = mongo.FindOne(ctx, d.NoCache, bson.M{"name": name}, &adSource, "ad_sources")
|
||||
err = mongo.DB().FindOne(ctx, bson.M{"name": name}, &adSource, "ad_sources")
|
||||
return
|
||||
}
|
||||
|
||||
// GetAvailableSources 获取可用的广告源
|
||||
func (d *adSourceDao) GetAvailableSources(ctx context.Context) (list []*entity.AdSource, err error) {
|
||||
err = mongo.Find(ctx, d.NoCache, bson.M{"status": "active"}, &list, "ad_sources",
|
||||
err = mongo.DB().Find(ctx, bson.M{"status": "active"}, &list, "ad_sources",
|
||||
options.Find().SetSort(bson.M{"priority": -1, "createdAt": 1}))
|
||||
return
|
||||
}
|
||||
|
||||
// GetSourcesByProvider 根据提供商获取广告源
|
||||
func (d *adSourceDao) GetSourcesByProvider(ctx context.Context, provider string) (list []*entity.AdSource, err error) {
|
||||
err = mongo.Find(ctx, d.NoCache, bson.M{"provider": provider, "status": "active"}, &list, "ad_sources",
|
||||
err = mongo.DB().Find(ctx, bson.M{"provider": provider, "status": "active"}, &list, "ad_sources",
|
||||
options.Find().SetSort(bson.M{"priority": -1}))
|
||||
return
|
||||
}
|
||||
|
||||
// Create 创建广告源
|
||||
func (d *adSourceDao) Create(ctx context.Context, adSource *entity.AdSource) (id string, err error) {
|
||||
ids, err := mongo.Insert(ctx, []interface{}{adSource}, "ad_sources")
|
||||
ids, err := mongo.DB().Insert(ctx, []interface{}{adSource}, "ad_sources")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
@@ -55,7 +49,7 @@ func (d *adSourceDao) Create(ctx context.Context, adSource *entity.AdSource) (id
|
||||
|
||||
// Update 更新广告源
|
||||
func (d *adSourceDao) Update(ctx context.Context, adSource *entity.AdSource) (affected int64, err error) {
|
||||
result, err := mongo.Update(ctx, bson.M{"_id": adSource.Id}, bson.M{"$set": adSource}, "ad_sources")
|
||||
result, err := mongo.DB().Update(ctx, bson.M{"_id": adSource.Id}, bson.M{"$set": adSource}, "ad_sources")
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
@@ -64,7 +58,7 @@ func (d *adSourceDao) Update(ctx context.Context, adSource *entity.AdSource) (af
|
||||
|
||||
// Delete 删除广告源
|
||||
func (d *adSourceDao) Delete(ctx context.Context, id string) (affected int64, err error) {
|
||||
count, err := mongo.Delete(ctx, bson.M{"_id": id}, "ad_sources")
|
||||
count, err := mongo.DB().Delete(ctx, bson.M{"_id": id}, "ad_sources")
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
@@ -73,13 +67,13 @@ func (d *adSourceDao) Delete(ctx context.Context, id string) (affected int64, er
|
||||
|
||||
// GetByID 根据ID获取广告源
|
||||
func (d *adSourceDao) GetByID(ctx context.Context, id string) (adSource *entity.AdSource, err error) {
|
||||
err = mongo.FindOne(ctx, d.NoCache, bson.M{"_id": id}, &adSource, "ad_sources")
|
||||
err = mongo.DB().FindOne(ctx, bson.M{"_id": id}, &adSource, "ad_sources")
|
||||
return
|
||||
}
|
||||
|
||||
// UpdateFields 更新广告源部分字段
|
||||
func (d *adSourceDao) UpdateFields(ctx context.Context, id string, data *entity.AdSource) (affected int64, err error) {
|
||||
result, err := mongo.Update(ctx, bson.M{"_id": id}, bson.M{"$set": data}, "ad_sources")
|
||||
result, err := mongo.DB().Update(ctx, bson.M{"_id": id}, bson.M{"$set": data}, "ad_sources")
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
@@ -14,16 +14,9 @@ import (
|
||||
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||
)
|
||||
|
||||
var Advertisement = &advertisement{
|
||||
NoCache: true,
|
||||
}
|
||||
var Advertisement = &advertisement{}
|
||||
|
||||
type advertisement struct {
|
||||
NoCache bool
|
||||
}
|
||||
|
||||
func (d *advertisement) SetNoCache() {
|
||||
Advertisement.NoCache = true
|
||||
}
|
||||
|
||||
// Insert 插入广告
|
||||
@@ -37,7 +30,7 @@ func (d *advertisement) Insert(ctx context.Context, advertisement *entity.Advert
|
||||
g.Log().Infof(ctx, "获取到stream消息: %v", streamMsg)
|
||||
}
|
||||
|
||||
_, err = mongo.Insert(ctx, []interface{}{advertisement}, entity.AdvertisementCollection)
|
||||
_, err = mongo.DB().Insert(ctx, []interface{}{advertisement}, entity.AdvertisementCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -116,7 +109,7 @@ func (d *advertisement) Update(ctx context.Context, req *dto.UpdateAdvertisement
|
||||
|
||||
if len(updateFields) > 0 {
|
||||
update := bson.M{"$set": updateFields}
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
}
|
||||
return
|
||||
}
|
||||
@@ -130,7 +123,7 @@ func (d *advertisement) UpdateStatus(ctx context.Context, id, status string) (er
|
||||
filter := bson.M{"_id": objectId}
|
||||
update := bson.M{"$set": bson.M{"status": status}}
|
||||
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -155,7 +148,7 @@ func (d *advertisement) Audit(ctx context.Context, id, auditStatus, auditReason
|
||||
},
|
||||
}
|
||||
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -168,7 +161,7 @@ func (d *advertisement) UpdateStatistics(ctx context.Context, id string, stats m
|
||||
filter := bson.M{"_id": objectId}
|
||||
update := bson.M{"$set": stats}
|
||||
|
||||
_, err = mongo.Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
_, err = mongo.DB().Update(ctx, filter, update, entity.AdvertisementCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -181,7 +174,7 @@ func (d *advertisement) GetOne(ctx context.Context, id string) (advertisement *e
|
||||
filter := bson.M{"_id": objectId}
|
||||
|
||||
advertisement = &entity.Advertisement{}
|
||||
err = mongo.FindOne(ctx, d.NoCache, filter, advertisement, entity.AdvertisementCollection)
|
||||
err = mongo.DB().FindOne(ctx, filter, advertisement, entity.AdvertisementCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -223,7 +216,7 @@ func (d *advertisement) buildListFilter(req *dto.ListAdvertisementReq) bson.M {
|
||||
|
||||
// checkTotalCount 检查总数
|
||||
func (d *advertisement) checkTotalCount(ctx context.Context, filter bson.M) (total int64, err error) {
|
||||
total, err = mongo.Count(ctx, d.NoCache, filter, entity.AdvertisementCollection)
|
||||
total, err = mongo.DB().Count(ctx, filter, entity.AdvertisementCollection)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -256,6 +249,6 @@ func (d *advertisement) List(ctx context.Context, req *dto.ListAdvertisementReq)
|
||||
|
||||
opts := options.Find().SetLimit(limit).SetSkip(skip).SetSort(sort)
|
||||
|
||||
err = mongo.Find(ctx, d.NoCache, filter, &list, entity.AdvertisementCollection, opts)
|
||||
err = mongo.DB().Find(ctx, filter, &list, entity.AdvertisementCollection, opts)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ import (
|
||||
"context"
|
||||
|
||||
"cid/model/entity"
|
||||
|
||||
"gitee.com/red-future---jilin-g/common/mongo"
|
||||
"go.mongodb.org/mongo-driver/v2/bson"
|
||||
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||
@@ -11,20 +12,13 @@ import (
|
||||
|
||||
// applicationDao 应用DAO
|
||||
type applicationDao struct {
|
||||
NoCache bool
|
||||
}
|
||||
|
||||
var Application = &applicationDao{
|
||||
NoCache: true,
|
||||
}
|
||||
|
||||
func (d *applicationDao) SetNoCache() {
|
||||
Application.NoCache = true
|
||||
}
|
||||
var Application = &applicationDao{}
|
||||
|
||||
// Create 创建应用
|
||||
func (d *applicationDao) Create(ctx context.Context, app *entity.Application) (string, error) {
|
||||
ids, err := mongo.Insert(ctx, []interface{}{app}, "application")
|
||||
ids, err := mongo.DB().Insert(ctx, []interface{}{app}, "application")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
@@ -37,14 +31,14 @@ func (d *applicationDao) Create(ctx context.Context, app *entity.Application) (s
|
||||
// GetByID 根据ID获取应用
|
||||
func (d *applicationDao) GetByID(ctx context.Context, id string) (*entity.Application, error) {
|
||||
var app *entity.Application
|
||||
err := mongo.FindOne(ctx, d.NoCache, bson.M{"_id": id}, &app, "application")
|
||||
err := mongo.DB().FindOne(ctx, bson.M{"_id": id}, &app, "application")
|
||||
return app, err
|
||||
}
|
||||
|
||||
// GetByTenantID 根据租户ID获取应用列表
|
||||
func (d *applicationDao) GetByTenantID(ctx context.Context, tenantID string) ([]*entity.Application, error) {
|
||||
var apps []*entity.Application
|
||||
err := mongo.Find(ctx, d.NoCache,
|
||||
err := mongo.DB().Find(ctx,
|
||||
bson.M{"tenantId": tenantID}, &apps, "application")
|
||||
return apps, err
|
||||
}
|
||||
@@ -52,20 +46,20 @@ func (d *applicationDao) GetByTenantID(ctx context.Context, tenantID string) ([]
|
||||
// GetByAPIKey 根据API密钥获取应用
|
||||
func (d *applicationDao) GetByAPIKey(ctx context.Context, apiKey string) (*entity.Application, error) {
|
||||
var app *entity.Application
|
||||
err := mongo.FindOne(ctx, d.NoCache,
|
||||
err := mongo.DB().FindOne(ctx,
|
||||
bson.M{"appKey": apiKey}, &app, "application")
|
||||
return app, err
|
||||
}
|
||||
|
||||
// Update 更新应用
|
||||
func (d *applicationDao) Update(ctx context.Context, app *entity.Application) error {
|
||||
_, err := mongo.Update(ctx, bson.M{"_id": app.Id}, bson.M{"$set": app}, "application")
|
||||
_, err := mongo.DB().Update(ctx, bson.M{"_id": app.Id}, bson.M{"$set": app}, "application")
|
||||
return err
|
||||
}
|
||||
|
||||
// Delete 删除应用
|
||||
func (d *applicationDao) Delete(ctx context.Context, id string) error {
|
||||
_, err := mongo.Delete(ctx, bson.M{"_id": id}, "application")
|
||||
_, err := mongo.DB().Delete(ctx, bson.M{"_id": id}, "application")
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -77,13 +71,13 @@ func (d *applicationDao) List(ctx context.Context, tenantID string, page, pageSi
|
||||
}
|
||||
|
||||
var apps []*entity.Application
|
||||
total, err := mongo.Count(ctx, d.NoCache, filter, "application")
|
||||
total, err := mongo.DB().Count(ctx, filter, "application")
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
offset := (page - 1) * pageSize
|
||||
err = mongo.Find(ctx, d.NoCache, filter, &apps, "application",
|
||||
err = mongo.DB().Find(ctx, filter, &apps, "application",
|
||||
options.Find().SetSort(bson.M{"createdAt": -1}).
|
||||
SetSkip(int64(offset)).
|
||||
SetLimit(int64(pageSize)))
|
||||
@@ -97,12 +91,12 @@ func (d *applicationDao) List(ctx context.Context, tenantID string, page, pageSi
|
||||
// GetByName 根据名称获取应用
|
||||
func (d *applicationDao) GetByName(ctx context.Context, name string) (*entity.Application, error) {
|
||||
var app *entity.Application
|
||||
err := mongo.FindOne(ctx, d.NoCache, bson.M{"name": name}, &app, "application")
|
||||
err := mongo.DB().FindOne(ctx, bson.M{"name": name}, &app, "application")
|
||||
return app, err
|
||||
}
|
||||
|
||||
// UpdateFields 更新应用部分字段
|
||||
func (d *applicationDao) UpdateFields(ctx context.Context, id string, data *entity.Application) error {
|
||||
_, err := mongo.Update(ctx, bson.M{"_id": id}, bson.M{"$set": data}, "application")
|
||||
_, err := mongo.DB().Update(ctx, bson.M{"_id": id}, bson.M{"$set": data}, "application")
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -4,26 +4,20 @@ import (
|
||||
"context"
|
||||
|
||||
"cid/model/entity"
|
||||
|
||||
"gitee.com/red-future---jilin-g/common/mongo"
|
||||
"go.mongodb.org/mongo-driver/v2/bson"
|
||||
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||
)
|
||||
|
||||
var CIDRequest = &cidRequestDao{
|
||||
NoCache: true,
|
||||
}
|
||||
var CIDRequest = &cidRequestDao{}
|
||||
|
||||
type cidRequestDao struct {
|
||||
NoCache bool
|
||||
}
|
||||
|
||||
func (d *cidRequestDao) SetNoCache() {
|
||||
CIDRequest.NoCache = true
|
||||
}
|
||||
|
||||
// Create 创建CID请求记录
|
||||
func (d *cidRequestDao) Create(ctx context.Context, request *entity.CidRequest) (id string, err error) {
|
||||
ids, err := mongo.Insert(ctx, []interface{}{request}, entity.CidRequestCollection)
|
||||
ids, err := mongo.DB().Insert(ctx, []interface{}{request}, entity.CidRequestCollection)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
@@ -38,14 +32,14 @@ func (d *cidRequestDao) GetHistory(ctx context.Context, userId string, page, siz
|
||||
filter := bson.M{"userId": userId}
|
||||
|
||||
// 获取总数
|
||||
total, err = mongo.Count(ctx, d.NoCache, filter, entity.CidRequestCollection)
|
||||
total, err = mongo.DB().Count(ctx, filter, entity.CidRequestCollection)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// 分页查询
|
||||
offset := (page - 1) * size
|
||||
err = mongo.Find(ctx, d.NoCache, filter, &list, entity.CidRequestCollection,
|
||||
err = mongo.DB().Find(ctx, filter, &list, entity.CidRequestCollection,
|
||||
options.Find().SetSort(bson.M{"createdAt": -1}).
|
||||
SetSkip(int64(offset)).
|
||||
SetLimit(int64(size)))
|
||||
@@ -58,14 +52,14 @@ func (d *cidRequestDao) GetStatistics(ctx context.Context, userId string) (stats
|
||||
stats = make(map[string]interface{})
|
||||
|
||||
// 总请求数
|
||||
totalRequests, err := mongo.Count(ctx, d.NoCache, bson.M{"userId": userId}, entity.CidRequestCollection)
|
||||
totalRequests, err := mongo.DB().Count(ctx, bson.M{"userId": userId}, entity.CidRequestCollection)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
stats["total_requests"] = totalRequests
|
||||
|
||||
// 成功请求数
|
||||
successfulRequests, err := mongo.Count(ctx, d.NoCache, bson.M{"userId": userId, "status": "completed"}, entity.CidRequestCollection)
|
||||
successfulRequests, err := mongo.DB().Count(ctx, bson.M{"userId": userId, "status": "completed"}, entity.CidRequestCollection)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@@ -4,45 +4,39 @@ import (
|
||||
"context"
|
||||
|
||||
"cid/model/entity"
|
||||
|
||||
"gitee.com/red-future---jilin-g/common/mongo"
|
||||
"go.mongodb.org/mongo-driver/v2/bson"
|
||||
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||
)
|
||||
|
||||
var Strategy = &strategyDao{
|
||||
NoCache: true,
|
||||
}
|
||||
var Strategy = &strategyDao{}
|
||||
|
||||
type strategyDao struct {
|
||||
NoCache bool
|
||||
}
|
||||
|
||||
func (d *strategyDao) SetNoCache() {
|
||||
Strategy.NoCache = true
|
||||
}
|
||||
|
||||
// GetByName 根据名称获取策略
|
||||
func (d *strategyDao) GetByName(ctx context.Context, name string) (strategy *entity.Strategy, err error) {
|
||||
err = mongo.FindOne(ctx, d.NoCache, bson.M{"name": name}, &strategy, "strategies")
|
||||
err = mongo.DB().FindOne(ctx, bson.M{"name": name}, &strategy, "strategies")
|
||||
return
|
||||
}
|
||||
|
||||
// GetByID 根据ID获取策略
|
||||
func (d *strategyDao) GetByID(ctx context.Context, id string) (strategy *entity.Strategy, err error) {
|
||||
err = mongo.FindOne(ctx, d.NoCache, bson.M{"_id": id}, &strategy, "strategies")
|
||||
err = mongo.DB().FindOne(ctx, bson.M{"_id": id}, &strategy, "strategies")
|
||||
return
|
||||
}
|
||||
|
||||
// GetByTenantLevel 根据租户级别获取策略
|
||||
func (d *strategyDao) GetByTenantLevel(ctx context.Context, tenantLevel string) (strategy *entity.Strategy, err error) {
|
||||
err = mongo.FindOne(ctx, d.NoCache, bson.M{"tenantLevel": tenantLevel, "status": "active"}, &strategy, "strategies",
|
||||
err = mongo.DB().FindOne(ctx, bson.M{"tenantLevel": tenantLevel, "status": "active"}, &strategy, "strategies",
|
||||
options.FindOne().SetSort(bson.M{"priority": -1, "createdAt": 1}))
|
||||
return
|
||||
}
|
||||
|
||||
// Create 创建策略
|
||||
func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id string, err error) {
|
||||
ids, err := mongo.Insert(ctx, []interface{}{strategy}, "strategies")
|
||||
ids, err := mongo.DB().Insert(ctx, []interface{}{strategy}, "strategies")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
@@ -54,7 +48,7 @@ func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id
|
||||
|
||||
// Update 更新策略
|
||||
func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (affected int64, err error) {
|
||||
result, err := mongo.Update(ctx, bson.M{"_id": strategy.Id}, bson.M{"$set": strategy}, "strategies")
|
||||
result, err := mongo.DB().Update(ctx, bson.M{"_id": strategy.Id}, bson.M{"$set": strategy}, "strategies")
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
@@ -63,7 +57,7 @@ func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (af
|
||||
|
||||
// Delete 删除策略
|
||||
func (d *strategyDao) Delete(ctx context.Context, id string) (affected int64, err error) {
|
||||
count, err := mongo.Delete(ctx, bson.M{"_id": id}, "strategies")
|
||||
count, err := mongo.DB().Delete(ctx, bson.M{"_id": id}, "strategies")
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
@@ -83,14 +77,14 @@ func (d *strategyDao) GetList(ctx context.Context, page, size int, tenantLevel,
|
||||
}
|
||||
|
||||
// 获取总数
|
||||
total, err = mongo.Count(ctx, d.NoCache, filter, "strategies")
|
||||
total, err = mongo.DB().Count(ctx, filter, "strategies")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// 分页查询
|
||||
offset := (page - 1) * size
|
||||
err = mongo.Find(ctx, d.NoCache, filter, &list, "strategies",
|
||||
err = mongo.DB().Find(ctx, filter, &list, "strategies",
|
||||
options.Find().SetSort(bson.M{"priority": -1, "createdAt": -1}).
|
||||
SetSkip(int64(offset)).
|
||||
SetLimit(int64(size)))
|
||||
|
||||
6
go.mod
6
go.mod
@@ -3,7 +3,7 @@ module cid
|
||||
go 1.25.3
|
||||
|
||||
require (
|
||||
gitee.com/red-future---jilin-g/common v0.2.6
|
||||
gitee.com/red-future---jilin-g/common v0.2.9
|
||||
github.com/gogf/gf/contrib/drivers/mysql/v2 v2.9.5
|
||||
github.com/gogf/gf/contrib/nosql/redis/v2 v2.9.5
|
||||
github.com/gogf/gf/v2 v2.9.5
|
||||
@@ -11,7 +11,7 @@ require (
|
||||
golang.org/x/net v0.47.0
|
||||
)
|
||||
|
||||
//replace gitee.com/red-future---jilin-g/common v0.2.6 => ../common
|
||||
//replace gitee.com/red-future---jilin-g/common v0.2.9 => ../common
|
||||
|
||||
require (
|
||||
github.com/BurntSushi/toml v1.5.0 // indirect
|
||||
@@ -89,5 +89,3 @@ require (
|
||||
google.golang.org/protobuf v1.36.8 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
//replace gitee.com/red-future---jilin-g/common v0.1.9 => ../common
|
||||
|
||||
6
go.sum
6
go.sum
@@ -1,6 +1,8 @@
|
||||
cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
|
||||
gitee.com/red-future---jilin-g/common v0.2.1/go.mod h1:TFCNnI7801DamWd0m/M2CT5T090jFXXGLNDY4iZ2WY8=
|
||||
gitee.com/red-future---jilin-g/common v0.2.6/go.mod h1:TFCNnI7801DamWd0m/M2CT5T090jFXXGLNDY4iZ2WY8=
|
||||
gitee.com/red-future---jilin-g/common v0.2.8 h1:mqe/qb6MejwV3VmnYxrO5MAHsFhvMejyfBA1993ejBg=
|
||||
gitee.com/red-future---jilin-g/common v0.2.8/go.mod h1:TFCNnI7801DamWd0m/M2CT5T090jFXXGLNDY4iZ2WY8=
|
||||
gitee.com/red-future---jilin-g/common v0.2.9 h1:lqYUnqTQLEcvxfFBqRMpetMudQze5cQlM4fKDouu2CU=
|
||||
gitee.com/red-future---jilin-g/common v0.2.9/go.mod h1:TFCNnI7801DamWd0m/M2CT5T090jFXXGLNDY4iZ2WY8=
|
||||
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||
github.com/BurntSushi/toml v1.5.0 h1:W5quZX/G/csjUnuI8SUYlsHs9M38FC7znL0lIO+DvMg=
|
||||
github.com/BurntSushi/toml v1.5.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
|
||||
|
||||
1
update.sql
Normal file
1
update.sql
Normal file
@@ -0,0 +1 @@
|
||||
-----------张斌2025-06-16 15:00:00--------------
|
||||
Reference in New Issue
Block a user