init: 初始化 web_hgdjb_server 项目
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,153 @@
|
||||
package skdrecord
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"videoweb/pub/common/log"
|
||||
"videoweb/pub/common/srvdef"
|
||||
|
||||
"videoweb/pub/mongos"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
var db *mongos.Collection
|
||||
|
||||
func Init(ctx context.Context) {
|
||||
// 获取数据库表
|
||||
db = mongos.Shop.Coll(TN.SkdRecord)
|
||||
// 初始化索引
|
||||
indexes := []mongo.IndexModel{
|
||||
{
|
||||
Keys: bson.D{
|
||||
{Key: FN.SkdRecord.Job, Value: 1},
|
||||
},
|
||||
},
|
||||
{
|
||||
Keys: bson.D{
|
||||
{Key: FN.SkdRecord.CreatedAt, Value: 1},
|
||||
},
|
||||
},
|
||||
}
|
||||
if _, err := db.CreateIndex(ctx, indexes); err != nil {
|
||||
panic(fmt.Sprintf("(SkdRecord:initSkdRecordIndex) Create SkdRecord Index failed! err:%+v:", err))
|
||||
}
|
||||
}
|
||||
|
||||
// InsertOneSkdRecord 插入定时任务
|
||||
func InsertOneSkdRecord(ctx context.Context, SkdRecord *SkdRecord) *srvdef.SrvErr {
|
||||
if _, err := db.InsertOne(ctx, SkdRecord); err != nil {
|
||||
log.Warn("SkdRecord:InsertOneSkdRecord db.InsertOne failed", log.E(err))
|
||||
return srvdef.New(srvdef.ErrDbInsertError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// FindOneSkdRecord 插入定时任务
|
||||
func FindOneSkdRecord(ctx context.Context, filter bson.M) (
|
||||
*SkdRecord, *srvdef.SrvErr) {
|
||||
var SkdRecord SkdRecord
|
||||
if err := db.FindOneEx(ctx, &SkdRecord, filter); err != nil {
|
||||
if err != mongo.ErrNoDocuments {
|
||||
log.Warn("SkdRecord:FindOneSkdRecord db.FindOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil, srvdef.New(srvdef.ErrSkdRecordNotExsit, "", "")
|
||||
}
|
||||
return &SkdRecord, nil
|
||||
}
|
||||
|
||||
// DelOneSkdRecord 插入定时任务
|
||||
func DelOneSkdRecord(ctx context.Context, filter bson.M) (
|
||||
int64, *srvdef.SrvErr) {
|
||||
count, err := db.DeleteOne(ctx, filter)
|
||||
if err != nil {
|
||||
log.Warn("SkdRecord:DelOneSkdRecord db.DeleteOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "")
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
// UpdateOneSkdRecord 插入定时任务
|
||||
func UpdateOneSkdRecord(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
|
||||
result, err := db.UpdateOne(ctx, filter, update)
|
||||
if err != nil {
|
||||
log.Warn("SkdRecord:UpdateOneSkdRecord db.UpdateOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
if result.MatchedCount != 1 {
|
||||
return srvdef.New(srvdef.ErrSkdRecordNotExsit, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpsertOneSkdRecord 更新插入一个记录
|
||||
func UpsertOneSkdRecord(ctx context.Context, filter bson.M, upsertDoc bson.M) *srvdef.SrvErr {
|
||||
_, err := db.UpsertOne(ctx, filter, upsertDoc)
|
||||
if err != nil {
|
||||
log.Info("SkdRecord:UpsertOneSkdRecord db.UpsertOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CountSkdRecords 统计用户数量
|
||||
func CountSkdRecords(ctx context.Context, filter bson.M) (
|
||||
int64, *srvdef.SrvErr) {
|
||||
count, err := db.Count(ctx, filter)
|
||||
if err != nil {
|
||||
log.Warn("SkdRecord:CountSkdRecords db.Count failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return 0, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
// UpdateManySkdRecord 更新满足条件的用户
|
||||
func UpdateManySkdRecord(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
|
||||
_, err := db.UpdateMany(ctx, filter, update)
|
||||
if err != nil {
|
||||
log.Warn("SkdRecord:UpdateManySkdRecord db.UpdateMany failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// FindManySkdRecord 查找一批用户数据
|
||||
func FindManySkdRecord(ctx context.Context, filter bson.M,
|
||||
opts ...*options.FindOptions) (SkdRecordList, *srvdef.SrvErr) {
|
||||
data := make(SkdRecordList, 0)
|
||||
if err := db.Find(ctx, &data, filter, opts...); err != nil {
|
||||
log.Warn("SkdRecord:FindManySkdRecord db.Find failed", log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return data, nil
|
||||
}
|
||||
|
||||
// FindManySkdRecordWithBind 用自己定义结构查找一批用户数据
|
||||
func FindManySkdRecordWithBind(ctx context.Context, bind interface{}, filter bson.M,
|
||||
opts ...*options.FindOptions) *srvdef.SrvErr {
|
||||
if err := db.Find(ctx, bind, filter, opts...); err != nil {
|
||||
log.Warn("SkdRecord:FindManySkdRecordWithBind db.Find failed", log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// AggregateSkdRecords 聚合查找一批用户信息
|
||||
func AggregateSkdRecords(ctx context.Context,
|
||||
bind interface{}, pipeline []bson.M) *srvdef.SrvErr {
|
||||
if err := db.Aggregate(ctx, bind, pipeline); err != nil {
|
||||
log.Warn("feedback:AggregateSkdRecords db.Aggregate failed",
|
||||
log.E(err), log.Any("pipeline", pipeline))
|
||||
return srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Executable
+30
@@ -0,0 +1,30 @@
|
||||
package skdrecord
|
||||
|
||||
type TagSkdRecord struct {
|
||||
CreatedAt string
|
||||
FirstRecordTime string
|
||||
ID string
|
||||
Item string
|
||||
Job string
|
||||
RecordTime string
|
||||
UpdatedAt string
|
||||
}
|
||||
|
||||
var FN struct {
|
||||
SkdRecord TagSkdRecord
|
||||
}
|
||||
|
||||
var TN struct {
|
||||
SkdRecord string
|
||||
}
|
||||
|
||||
func init() {
|
||||
FN.SkdRecord.CreatedAt = "createdAt"
|
||||
FN.SkdRecord.FirstRecordTime = "firstRecordTime"
|
||||
FN.SkdRecord.ID = "_id"
|
||||
FN.SkdRecord.Item = "item"
|
||||
FN.SkdRecord.Job = "job"
|
||||
FN.SkdRecord.RecordTime = "recordTime"
|
||||
FN.SkdRecord.UpdatedAt = "updatedAt"
|
||||
TN.SkdRecord = "skd_record"
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
package skdrecord
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
// 表名
|
||||
// SkdRecord 获取充值记录
|
||||
// @table:skd_record
|
||||
type SkdRecord struct {
|
||||
ID primitive.ObjectID `bson:"_id,omitempty"`
|
||||
Job string `bson:"job"` // 工作
|
||||
Item string `bson:"item"` // 统计项
|
||||
RecordTime time.Time `bson:"recordTime"` // 提交时间
|
||||
FirstRecordTime time.Time `bson:"firstRecordTime"` // 第一次提交时间
|
||||
CreatedAt time.Time `bson:"createdAt"` // 创建时间
|
||||
UpdatedAt time.Time `bson:"updatedAt"` // 更新时间
|
||||
}
|
||||
|
||||
const (
|
||||
GameRecordJob = "GameRecordJob" // 按天统计访问信息
|
||||
)
|
||||
|
||||
type SkdRecordList []*SkdRecord
|
||||
Reference in New Issue
Block a user