90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package syncrecord
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"stat/common/log"
|
|
"stat/common/mongos"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
// 初始化model
|
|
func Init() {
|
|
// 获取数据库表
|
|
db = mongos.New(mongos.MgoManager, CollSyncRecord)
|
|
|
|
indexes := []mongo.IndexModel{
|
|
{
|
|
Keys: bson.D{{Key: "job", Value: 1}, {Key: "appid", Value: 1}},
|
|
Options: options.Index().SetUnique(true),
|
|
},
|
|
}
|
|
|
|
if _, err := db.CreateIndex(context.Background(), indexes); err != nil {
|
|
panic(fmt.Sprintf("Model %s InitIndex error %+v", CollSyncRecord, err))
|
|
}
|
|
}
|
|
|
|
// LastRecordTimeMap 查询统计记录
|
|
func LastRecordTimeMap(ctx context.Context) (list []SyncRecord, err error) {
|
|
list = make([]SyncRecord, 0)
|
|
err = db.Find(ctx, &list, bson.M{})
|
|
if err != nil {
|
|
log.Error("SyncRecord LastRecordTimeMap err", log.E(err))
|
|
return nil, err
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// UpsertItemRecord 插入统计成功项记录
|
|
func UpsertItemRecord(ctx context.Context, job string, appid int32, userid uint64, platformId string, successAt time.Time) (err error) {
|
|
filter := bson.M{
|
|
"job": job,
|
|
"appid": appid,
|
|
}
|
|
exists, err := db.Exists(ctx, filter)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if exists {
|
|
_, err = db.UpdateOne(ctx, filter,
|
|
bson.M{
|
|
"$set": bson.M{
|
|
"userId": userid,
|
|
"successAt": successAt,
|
|
"platformId": platformId,
|
|
"updatedAt": time.Now(),
|
|
}})
|
|
} else {
|
|
insertDoc := bson.M{
|
|
"job": job,
|
|
"appid": appid,
|
|
"userId": userid,
|
|
"platformId": platformId,
|
|
"successAt": successAt,
|
|
"createdAt": time.Now(),
|
|
}
|
|
|
|
_, err = db.InsertOne(ctx, insertDoc)
|
|
}
|
|
return
|
|
}
|
|
|
|
// LastOneRecordTime 查询上一次统计时间
|
|
func LastOneRecordTime(ctx context.Context, job string, appid int32) (record SyncRecord, err error) {
|
|
|
|
err = db.FindOne(ctx, &record, bson.M{"job": job, "appid": appid})
|
|
if err != nil {
|
|
log.Error("SyncRecord LastRecordTimeMap err", log.E(err))
|
|
return
|
|
}
|
|
return
|
|
}
|