78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
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
|
|
}
|