Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,121 @@
+package channelpackage
+
+import (
+	"context"
+	"fmt"
+	"stat/common/log"
+	"stat/common/mongos"
+	"stat/common/stdLog"
+
+	"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, CollChannelPackage)
+
+	// 初始化索引
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{
+				{Key: FN.ChannelPackage.AppId, Value: 1},
+				{Key: FN.ChannelPackage.Channel, Value: 1},
+				{Key: FN.ChannelPackage.Version, Value: 1},
+				{Key: FN.ChannelPackage.FileName, Value: 1},
+			},
+			Options: options.Index().SetUnique(true),
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.ChannelPackage.AppId, Value: 1},
+				{Key: FN.ChannelPackage.Channel, Value: 1},
+				{Key: FN.ChannelPackage.Version, Value: 1},
+				{Key: FN.ChannelPackage.FileName, Value: 1},
+				{Key: FN.ChannelPackage.Status, Value: 1},
+			},
+		},
+	}
+
+	if _, err := db.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("Model %s InitIndex error %+v", CollChannelPackage, err))
+	}
+}
+
+// InsertOneChannelPackage 插入一个应用
+func InsertOneChannelPackage(ctx context.Context, ChannelPackage *ChannelPackage) *stdLog.SrvErr {
+	if _, err := db.InsertOne(ctx, ChannelPackage); err != nil {
+		log.Warn("media:InsertOneChannelPackage db.InsertOne failed", log.E(err))
+		return stdLog.New(stdLog.InsertDataError, "", "")
+	}
+	return nil
+}
+
+// DelOneChannelPackage 删除一个应用tag
+func DelOneChannelPackage(ctx context.Context, filter bson.M) (
+	int64, *stdLog.SrvErr) {
+	count, err := db.DeleteOne(ctx, filter)
+	if err != nil {
+		log.Warn("application:DelOneChannelPackage db.DeleteOne failed",
+			log.E(err), log.Any("filter", filter))
+		return 0, stdLog.New(stdLog.DelDataError, "", "")
+	}
+	return count, nil
+}
+
+// UpdateOneChannelPackage 更新一个应用
+func UpdateOneChannelPackage(ctx context.Context, filter bson.M, update bson.M) *stdLog.SrvErr {
+	result, err := db.UpdateOne(ctx, filter, update)
+	if err != nil {
+		log.Warn("application:UpdateOneChannelPackage db.UpdateOne failed",
+			log.E(err), log.Any("filter", filter))
+		return stdLog.New(stdLog.UpdateDataError, "", "")
+	}
+	if result.MatchedCount != 1 {
+		log.Warn("application:UpdateOneChannelPackage db.UpdateOne failed",
+			log.E(err), log.Any("filter", filter))
+		return stdLog.New(stdLog.DataNotExist, "", "")
+	}
+	return nil
+}
+
+// CountChannelPackages 统计应用数量
+func CountChannelPackages(ctx context.Context, filter bson.M) (
+	int64, *stdLog.SrvErr) {
+	count, err := db.Count(ctx, filter)
+	if err != nil {
+		log.Warn("media:CountChannelPackages db.Count failed",
+			log.E(err), log.Any("filter", filter))
+		return 0, stdLog.New(stdLog.QueryDataError, "", "")
+	}
+	return count, nil
+}
+
+// FindOneChannelPackage 查找一个应用
+func FindOneChannelPackage(ctx context.Context, filter bson.M) (
+	*ChannelPackage, *stdLog.SrvErr) {
+	var ChannelPackage ChannelPackage
+	if err := db.FindOneEx(
+		ctx, &ChannelPackage, filter); err != nil {
+		if err != mongo.ErrNoDocuments {
+			log.Info("media:FindOneChannelPackage db.FindOne failed",
+				log.E(err), log.Any("filter", filter))
+			return nil, stdLog.New(stdLog.QueryDataError, "", "")
+		}
+		return nil, stdLog.New(stdLog.DataNotExist, "", "")
+	}
+	return &ChannelPackage, nil
+}
+
+// FindManyChannelPackage 查找一批应用
+func FindManyChannelPackage(ctx context.Context, filter bson.M,
+	opts ...*options.FindOptions) (ChannelPackageList, *stdLog.SrvErr) {
+	data := make(ChannelPackageList, 0)
+	if err := db.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("application:FindManyChannelPackage db.Find failed",
+			log.E(err), log.Any("filter", filter))
+		return nil, stdLog.New(stdLog.QueryDataError, "", "")
+	}
+	return data, nil
+}