增加双端下载

models/cpcDownloadCfg/cpcDownloadCfg.go

@@ -0,0 +1,81 @@
+package cpcDownloadCfg
+
+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.CpcDownloadCfg)
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{
+				{Key: FN.CpcDownloadCfg.AppId, Value: 1},
+				{Key: FN.CpcDownloadCfg.Channel, Value: 1},
+			},
+		},
+		{
+			Keys: bson.D{{Key: FN.CpcDownloadCfg.Status, Value: 1}},
+		},
+	}
+	if _, err := db.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(cpcDownloadCfg:Init) CreateIndex failed: %+v", err))
+	}
+}
+
+func InsertOneCpcDownloadCfg(ctx context.Context, doc *CpcDownloadCfg) error {
+	if _, err := db.InsertOne(ctx, doc); err != nil {
+		log.Warn("cpcDownloadCfg InsertOne failed", log.E(err))
+		return err
+	}
+	return nil
+}
+
+func FindOneCpcDownloadCfg(ctx context.Context, filter bson.M) (*CpcDownloadCfg, error) {
+	var doc CpcDownloadCfg
+	if err := db.FindOne(ctx, &doc, filter); err != nil {
+		return nil, err
+	}
+	return &doc, nil
+}
+
+func DelOneCpcDownloadCfg(ctx context.Context, filter bson.M) (int64, error) {
+	count, err := db.DeleteOne(ctx, filter)
+	if err != nil {
+		log.Warn("cpcDownloadCfg DeleteOne failed", log.E(err), log.Any("filter", filter))
+		return 0, err
+	}
+	return count, nil
+}
+
+func UpdateOneCpcDownloadCfg(ctx context.Context, filter bson.M, update bson.M) error {
+	if _, err := db.UpdateOne(ctx, filter, update); err != nil {
+		log.Warn("cpcDownloadCfg UpdateOne failed", log.E(err), log.Any("filter", filter))
+		return err
+	}
+	return nil
+}
+
+func CountCpcDownloadCfgs(ctx context.Context, filter bson.M) (int64, error) {
+	count, err := db.Count(ctx, filter)
+	if err != nil {
+		log.Warn("cpcDownloadCfg Count failed", log.E(err), log.Any("filter", filter))
+		return 0, err
+	}
+	return count, nil
+}
+
+func FindManyCpcDownloadCfg(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (CpcDownloadCfgList, error) {
+	list := make(CpcDownloadCfgList, 0)
+	if err := db.Find(ctx, &list, filter, opts...); err != nil {
+		log.Warn("cpcDownloadCfg Find failed", log.E(err), log.Any("filter", filter))
+		return nil, err
+	}
+	return list, nil
+}