增加软质押手动结算测试接口

internal/job/stake_settle/cron.go

@@ -3,13 +3,9 @@ package stake_settle
 import (
 	"context"
 	"github.com/robfig/cron/v3"
-	"github.com/shopspring/decimal"
-	"github.com/spf13/cast"
 	"github.com/zeromicro/go-zero/core/logx"
-	"nova_task/internal/model"
-	"nova_task/internal/pkg/utils"
+	"nova_task/internal/logic/nft"
 	"nova_task/internal/svc"
-	"time"
 )
 
 type Cron struct {
@@ -30,91 +26,6 @@ func (c *Cron) Spec() string {
 
 func (c *Cron) Run() {
 	logx.Debugw("run settle cron task")
-	start, end, err := c.svcCtx.ConfigModel.GetNftStakeTaskOpenDate(c.ctx)
-	if err != nil {
-		logx.Errorw("get nft stake task open date failed", logx.Field("err", err))
-		return
-	}
-	end = end.AddDate(0, 0, 1).Add(-time.Second)
-	now := time.Now()
-	if now.Before(start) || now.After(end) {
-		logx.Debugw("now is not in the date range", logx.Field("now", now), logx.Field("start", start), logx.Field("end", end))
-		return
-	}
-
-	taskConf, err := c.svcCtx.ConfigModel.GetNftStakeTaskConf(c.ctx)
-	if err != nil {
-		logx.Errorw("get nft stake task conf failed", logx.Field("err", err))
-		return
-	}
-
-	stakes, err := c.svcCtx.StakeNftModel.AllStakeNft(c.ctx)
-	if err != nil {
-		logx.Errorw("get all stake nft failed", logx.Field("err", err))
-		return
-	}
-
-	uid2tokens := map[uint]float64{}
-	for _, s := range stakes {
-		if s.State != 1 {
-			continue
-		}
-		nftHolder, err := c.svcCtx.NftHolderModel.FindOneByTokenId(c.ctx, s.TokenId)
-		if err != nil {
-			logx.Errorw("find nft holder failed", logx.Field("err", err), logx.Field("tokenId", s.TokenId))
-			continue
-		}
-
-		uid, err := c.svcCtx.WalletModel.FindUidByAddress(c.ctx, nftHolder.Address)
-		if err != nil {
-			logx.Errorw("find uid by address failed", logx.Field("err", err), logx.Field("address", nftHolder.Address))
-			continue
-		}
-
-		if uid != s.Uid {
-			logx.Errorw("uid not match", logx.Field("uid", s.Uid), logx.Field("nftHolderUid", uid))
-			continue
-		}
-
-		if utils.IsBigTarot(s.TokenId) {
-			uid2tokens[s.Uid] += float64(taskConf.GreatTarot)
-		} else {
-			uid2tokens[s.Uid] += float64(taskConf.LittleTarot)
-		}
-	}
-	awardSeq := cast.ToInt(time.Now().AddDate(0, 0, -1).Format("20060102"))
-	for uid, tokens := range uid2tokens {
-		coefficient, err := c.svcCtx.StakeRewardModel.GetRandomCoefficientByUid(c.ctx, uid, awardSeq, taskConf.MinCoefficient, taskConf.MaxCoefficient)
-		if err != nil {
-			logx.Errorw("get random coefficient failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq))
-			continue
-		}
-		tokensDecimal := decimal.NewFromFloat(tokens)
-		reward := tokensDecimal.Mul(decimal.NewFromFloat(coefficient))
-		err = c.svcCtx.StakeRewardModel.SetReward(c.ctx, uid, awardSeq, taskConf.OccupyPercent, tokensDecimal, reward)
-		if err != nil {
-			logx.Errorw("set reward failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq), logx.Field("coefficient", coefficient), logx.Field("tokens", tokens), logx.Field("reward", reward))
-			continue
-		}
-
-		// 加资产
-		err = c.svcCtx.TaskAssetModel.AddCastile(c.ctx, uid, reward)
-		if err != nil {
-			logx.Errorw("add castile failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq), logx.Field("reward", reward))
-			continue
-		}
-		// 加资产记录表
-		_, err = c.svcCtx.TaskAssetRecordModel.Insert(c.ctx, &model.NhTaskAssetRecord{
-			Uid:        int(uid),
-			EventId:    0,
-			AssetField: "castile",
-			Count:      reward.InexactFloat64(),
-			Remark:     "nft软质押奖励",
-			ProvideUid: 0,
-			CreateTime: int(now.Unix()),
-		})
-		if err != nil {
-			logx.Errorw("insert task asset record failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq), logx.Field("reward", reward))
-		}
-	}
+	lg := nft.NewStakeSettleLogic(c.ctx, c.svcCtx)
+	lg.StakeSettle()
 }