Initial commit

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-11-07 10:47:18 +08:00
commit 79ac89b458
708 changed files with 151634 additions and 0 deletions
@@ -0,0 +1,77 @@
+package uservisitlog
+
+import (
+	"context"
+	"fmt"
+	"time"
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/mongos"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+// Init 初始化访问日志表
+func Init(ctx context.Context) {
+	visitColl = mongos.Shop.Coll(TN.UserVisitLog)
+	initIndex(ctx)
+}
+
+func initIndex(ctx context.Context) {
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{
+				{Key: FN.UserVisitLog.UserId, Value: -1},
+			},
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.UserVisitLog.VisitAt, Value: -1},
+			},
+			Options: options.Index().SetExpireAfterSeconds(5 * 24 * 60 * 60),
+		},
+	}
+	if _, err := visitColl.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(uservisitlog:initIndex) Create uservisitlog Index failed! err:%+v:", err))
+	}
+}
+
+// InsertOneVisitLog 插入一条访问记录
+func InsertOneVisitLog(ctx context.Context, data *UserVisitLog) *srvdef.SrvErr {
+	if _, err := visitColl.InsertOne(ctx, data); err != nil {
+		log.Info("visitColl:InsertOneVisitLog db.InsertOne failed",
+			log.E(err), log.Any("data", data))
+		return srvdef.New(srvdef.ErrDbInsertError, "", "")
+	}
+	return nil
+}
+
+// FindManyVisitLogs 查找一批访问记录
+func FindManyVisitLogs(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (VisitLogList, *srvdef.SrvErr) {
+	data := make(VisitLogList, 0)
+	if err := visitColl.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("visitColl:FindManyVisitLogs db.Find failed",
+			log.E(err), log.Any("filter", filter))
+		return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return data, nil
+}
+
+// AccessSyncById 按 ObjectId 增量拉取访问数据（统计中心同步）
+func AccessSyncById(ctx context.Context, id string, size int64) (data VisitLogList, err error) {
+	oid, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		return
+	}
+	endTime := time.Now().Add(-time.Minute * 1)
+	visitFilter := bson.M{
+		FN.UserVisitLog.Id:      bson.M{"$gt": oid},
+		FN.UserVisitLog.VisitAt: bson.M{"$lt": endTime},
+	}
+	visitOpt := (&options.FindOptions{}).SetLimit(size).SetSort(bson.M{FN.UserVisitLog.Id: 1})
+	err = visitColl.Find(ctx, &data, visitFilter, visitOpt)
+	return data, err
+}