Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,96 @@
+package activityVisit
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"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"
+)
+
+// 初始化model
+func Init() {
+	// 获取数据库表
+	db = mongos.New(mongos.MgoManager, CollActivityVisit)
+
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{{Key: "appId", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "activityId", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "createdAt", Value: -1}},
+		},
+		{
+			Keys: bson.D{{Key: "ch", Value: 1}},
+		},
+	}
+
+	if _, err := db.CreateIndex(context.Background(), indexes); err != nil {
+		panic(fmt.Sprintf("Model %s InitIndex error %+v", CollActivityVisit, err))
+	}
+}
+
+// 插入InsertActivityVisit 活动访问次数
+func InsertActivityVisit(ctx context.Context, model *ActivityVisit) (err error) {
+	_, err = db.InsertOne(ctx, model)
+	return
+}
+
+// 插入InsertManyActivityVisit 活动访问次数
+func InsertManyActivityVisit(ctx context.Context, docs []interface{}) (err error) {
+	_, err = db.GetCollection().InsertMany(ctx, docs)
+	return
+}
+
+// FindManyActivityList 查找一批
+func FindManyActivityList(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (ActivityVisitList, error) {
+	data := make(ActivityVisitList, 0)
+	if err := db.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("FindManyActivityList db.Find failed", log.E(err), log.Any("filter", filter))
+		return nil, err
+	}
+	return data, nil
+}
+
+// 根据appId 获取activityVisit
+func GetActivityVisitByTime(ctx context.Context, appid int32, start, end time.Time) (ActivityVisitList, error) {
+	filter := bson.M{"appId": appid, "createdAt": bson.M{"$gte": start, "$lt": end}}
+	results := make(ActivityVisitList, 0)
+	if err := db.Find(ctx, &results, filter); err != nil {
+		log.Error("GetActivityVisitByTime FindOne err", log.Any("start", start), log.Any("end", end), log.E(err))
+		return nil, err
+	}
+
+	return results, nil
+}
+
+// 根据ch 获取activityVisit
+func GetActivityVisitByCh(ctx context.Context, ch string, start, end time.Time) (ActivityVisitList, error) {
+	filter := bson.M{"ch": ch, "createdAt": bson.M{"$gte": start, "$lt": end}}
+	results := make(ActivityVisitList, 0)
+	if err := db.Find(ctx, &results, filter); err != nil {
+		log.Error("GetActivityVisitByTime FindOne err", log.Any("start", start), log.Any("end", end), log.E(err))
+		return nil, err
+	}
+
+	return results, nil
+}
+
+// CountActivityVisit 统计活动访问数量
+func CountActivityVisit(ctx context.Context, filter bson.M) (int64, error) {
+	count, err := db.Count(ctx, filter)
+	if err != nil {
+		log.Warn("CountActivityVisit db.Count failed",
+			log.E(err), log.Any("filter", filter))
+		return 0, err
+	}
+	return count, nil
+}
@@ -0,0 +1,28 @@
+package activityVisit
+
+import (
+	"stat/common/mongos"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson/primitive"
+)
+
+// 表名
+const CollActivityVisit = "activity_visit"
+
+var db *mongos.MongoTool
+
+type ActivityVisit struct {
+	AppId        int32              `json:"appId" bson:"appId"`               // appId
+	CH           string             `json:"ch" bson:"ch"`                     // 渠道号
+	IP           string             `json:"ip" bson:"ip"`                     // 用户IP
+	DevType      string             `json:"devType" bson:"devType"`           // 设备类型
+	SysType      string             `json:"sysType" bson:"sysType"`           // 系统类型
+	PeerIP       string             `json:"peerIP" bson:"peerIP"`             // 网站IP
+	ActivityId   primitive.ObjectID `json:"activityId" bson:"activityId"`     // 活动Id
+	ActivityName string             `json:"activityName" bson:"activityName"` // 活动名字
+	CreatedAt    time.Time          `bson:"createdAt"`                        // 创建时间
+	UpdatedAt    time.Time          `bson:"updatedAt"`                        // 更新时间
+}
+
+type ActivityVisitList []*ActivityVisit