rootandClaude Opus 5
3c8cefbb2a
advitise_info 增加 sort 字段,列表默认按 sort 从小到大排序。 info/list 增加 orderBy 参数: sortAsc(默认) / sortDesc / newest, 三种排序末尾都追加 _id 作为最后一级,避免同值文档在分页时重复或漏数据。 启动时回填历史数据缺失的 sort 字段: 缺字段的文档在升序里会排在 0 之前, 且按 sort=0 精确查询查不到。用 UpdateMMany 避免刷新全表 updatedAt。 Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package advitiseInfoMod
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"stat/common/log"
|
|
"stat/common/mongos"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
func Init(ctx context.Context) {
|
|
db = mongos.New(mongos.MgoManager, TN.AdvitiseInfo)
|
|
indexes := []mongo.IndexModel{
|
|
{Keys: bson.D{
|
|
{Key: FN.AdvitiseInfo.StartTime, Value: 1},
|
|
{Key: FN.AdvitiseInfo.EndTime, Value: 1},
|
|
{Key: FN.AdvitiseInfo.Status, Value: 1},
|
|
}},
|
|
{Keys: bson.D{{Key: FN.AdvitiseInfo.CreatedAt, Value: -1}}},
|
|
{Keys: bson.D{
|
|
{Key: FN.AdvitiseInfo.Sort, Value: 1},
|
|
{Key: FN.AdvitiseInfo.CreatedAt, Value: -1},
|
|
}},
|
|
}
|
|
if _, err := db.CreateIndex(ctx, indexes); err != nil {
|
|
panic(fmt.Sprintf("(advitiseInfoMod:Init) CreateIndex failed: %+v", err))
|
|
}
|
|
backfillSort(ctx)
|
|
}
|
|
|
|
// backfillSort 给历史数据补 sort 字段。缺字段的文档在升序排序里会排在 0 之前,
|
|
// 且按 sort=0 精确查询查不到,所以启动时统一补 0。幂等,补完后不再命中。
|
|
func backfillSort(ctx context.Context) {
|
|
ret, err := db.UpdateMMany(ctx,
|
|
bson.M{FN.AdvitiseInfo.Sort: bson.M{"$exists": false}},
|
|
bson.M{"$set": bson.M{FN.AdvitiseInfo.Sort: int64(0)}},
|
|
)
|
|
if err != nil {
|
|
log.Warn("advitiseInfoMod backfill sort failed", log.E(err))
|
|
return
|
|
}
|
|
if ret != nil && ret.ModifiedCount > 0 {
|
|
log.Info("advitiseInfoMod backfill sort done", log.Any("modified", ret.ModifiedCount))
|
|
}
|
|
}
|
|
|
|
func InsertOneAdvitiseInfo(ctx context.Context, doc *AdvitiseInfo) error {
|
|
if _, err := db.InsertOne(ctx, doc); err != nil {
|
|
log.Warn("advitiseInfoMod InsertOne failed", log.E(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func FindOneAdvitiseInfo(ctx context.Context, filter bson.M) (*AdvitiseInfo, error) {
|
|
var doc AdvitiseInfo
|
|
if err := db.FindOne(ctx, &doc, filter); err != nil {
|
|
return nil, err
|
|
}
|
|
return &doc, nil
|
|
}
|
|
|
|
func DelOneAdvitiseInfo(ctx context.Context, filter bson.M) (int64, error) {
|
|
count, err := db.DeleteOne(ctx, filter)
|
|
if err != nil {
|
|
log.Warn("advitiseInfoMod DeleteOne failed", log.E(err), log.Any("filter", filter))
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func UpdateOneAdvitiseInfo(ctx context.Context, filter bson.M, update bson.M) error {
|
|
if _, err := db.UpdateOne(ctx, filter, update); err != nil {
|
|
log.Warn("advitiseInfoMod UpdateOne failed", log.E(err), log.Any("filter", filter))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func CountAdvitiseInfos(ctx context.Context, filter bson.M) (int64, error) {
|
|
count, err := db.Count(ctx, filter)
|
|
if err != nil {
|
|
log.Warn("advitiseInfoMod Count failed", log.E(err), log.Any("filter", filter))
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func FindManyAdvitiseInfos(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (AdvitiseInfoList, error) {
|
|
list := make(AdvitiseInfoList, 0)
|
|
if err := db.Find(ctx, &list, filter, opts...); err != nil {
|
|
log.Warn("advitiseInfoMod Find failed", log.E(err), log.Any("filter", filter))
|
|
return nil, err
|
|
}
|
|
return list, nil
|
|
}
|