init: web_hgdjc_server

2026-11-07 10:46:18 +08:00
commit b2ff6ac8ff
707 changed files with 147645 additions and 0 deletions
@@ -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
+}
@@ -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