Initial commit

models/statrecord/statrecord.go

@@ -0,0 +1,73 @@
+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
+}