tribally report logic

internal/job/tribally_report/cron.go

@@ -2,9 +2,11 @@ package tribally_report
 
 import (
 	"context"
+	"errors"
 	"github.com/robfig/cron/v3"
 	"github.com/zeromicro/go-zero/core/logx"
 	"nova_task/internal/consts"
+	"nova_task/internal/model"
 	"nova_task/internal/svc"
 	"time"
 )
@@ -23,19 +25,48 @@ func (c *Cron) Spec() string {
 }
 
 func (c *Cron) Run() {
-	tr, err := c.svcCtx.GlobalDataModel.GetValue(c.ctx, consts.TriballyReportTime)
-	if err != nil {
+	tr, err := c.svcCtx.GlobalDataModel.FindOneByKey(c.ctx, consts.TriballyReportTime)
+	if err != nil && !errors.Is(err, model.ErrNotFound) {
 		logx.Errorw("get tribally report time failed", logx.Field("err", err))
 		return
 	}
-	var updateTime time.Time
-	if tr != "" {
-		updateTime, err = time.Parse(time.DateTime, tr)
+	if tr == nil {
+		tr = &model.NhGlobalData{
+			Key: consts.TriballyReportTime,
+		}
+	}
+	var start time.Time
+	if tr.Value != "" {
+		start, err = time.Parse(time.DateTime, tr.Value)
 		if err != nil {
 			logx.Errorw("parse tribally report time failed", logx.Field("err", err))
 			return
 		}
 	}
-	
-	c.svcCtx.TriballyUserModel.FindUpdateTriballyUsers(c.ctx, updateTime, 100)
+	end := time.Now()
+	triballyUsers, err := c.svcCtx.TriballyUserModel.FindUpdateTriballyUsers(c.ctx, start, end, 100)
+	if err != nil {
+		logx.Errorw("find update tribally users failed", logx.Field("err", err))
+		return
+	}
+
+	tr.Value = end.Format(time.DateTime)
+	if tr.Id > 0 {
+		err = c.svcCtx.GlobalDataModel.Update(c.ctx, tr)
+	} else {
+		_, err = c.svcCtx.GlobalDataModel.Insert(c.ctx, tr)
+	}
+	if err != nil {
+		logx.Errorw("insert or update tribally report time failed", logx.Field("err", err))
+		return
+	}
+
+	if len(triballyUsers) == 0 {
+		logx.Infow("no tribally users", logx.Field("start", start), logx.Field("end", end))
+		return
+	}
+
+	//for _, u := range triballyUsers {
+	//
+	//}
 }