forked from all_user/stat_center
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package statrecordmod
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"stat/common/log"
|
|
"stat/common/mongos"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
// 初始化model
|
|
func Init() {
|
|
// 获取数据库表
|
|
db = mongos.New(mongos.MgoManager, CollStatRecord)
|
|
|
|
indexes := []mongo.IndexModel{
|
|
{
|
|
Keys: bson.D{{Key: "item", Value: 1}},
|
|
},
|
|
}
|
|
|
|
if _, err := db.CreateIndex(context.Background(), indexes); err != nil {
|
|
panic(fmt.Sprintf("Model %s InitIndex error %+v", CollStatRecord, err))
|
|
}
|
|
}
|
|
|
|
// LastRecordTimeMap 查询统计记录
|
|
func LastRecordTimeMap(ctx context.Context, job string) (list []StatRecord, err error) {
|
|
list = make([]StatRecord, 0)
|
|
err = db.Find(ctx, &list, bson.M{"job": job})
|
|
if err != nil {
|
|
log.Error("statRecord LastRecordTimeMap err", log.E(err))
|
|
return nil, err
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// UpsertItemRecord 插入统计成功项记录
|
|
func UpsertItemRecord(ctx context.Context, job string, item string, recordTime time.Time) (err error) {
|
|
filter := bson.M{
|
|
"job": job,
|
|
"item": item,
|
|
}
|
|
update := bson.M{
|
|
"$set": bson.M{
|
|
"recordTime": recordTime,
|
|
"updatedAt": time.Now(),
|
|
},
|
|
"$setOnInsert": bson.M{
|
|
"job": job,
|
|
"item": item,
|
|
"fristRecordTime": recordTime,
|
|
"createdAt": time.Now(),
|
|
},
|
|
}
|
|
_, err = db.UpsertOne(ctx, filter, update)
|
|
return
|
|
}
|
|
|
|
// LastOneRecordTime 查询上一次统计时间
|
|
func LastOneRecordTime(ctx context.Context, job string, item string) (record StatRecord, err error) {
|
|
|
|
err = db.FindOne(ctx, &record, bson.M{"job": job, "item": item})
|
|
if err != nil {
|
|
log.Error("statRecord LastRecordTimeMap err", log.E(err))
|
|
return
|
|
}
|
|
return
|
|
}
|