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

internal/job/holder/cron.go

@@ -2,13 +2,10 @@ package holder
 
 import (
 	"context"
-	"errors"
 	"github.com/robfig/cron/v3"
-	"github.com/spf13/cast"
 	"github.com/zeromicro/go-zero/core/logx"
-	"nova_task/internal/model"
+	"nova_task/internal/logic/nft"
 	"nova_task/internal/svc"
-	"time"
 )
 
 type Cron struct {
@@ -33,93 +30,6 @@ func (c *Cron) Spec() string {
 
 func (c *Cron) Run() {
 	logx.Debugw("run Cron cron task")
-	ols, err := GetOwnerList()
-	if err != nil {
-		logx.Errorw("get owner list error", logx.Field("error", err))
-		return
-	}
-	updateSeq := int(time.Now().Unix())
-	for _, o := range ols.Owners {
-		for _, tk := range o.TokenBalances {
-			balance := cast.ToInt(tk.Balance)
-			logx.Debugw("owner token", logx.Field("address", o.OwnerAddress), logx.Field("token", tk.TokenID), logx.Field("balance", tk.Balance))
-			nft, err := c.svcCtx.NftHolderModel.FindOneByTokenId(c.ctx, tk.TokenID)
-			if err != nil {
-				if errors.Is(err, model.ErrNotFound) {
-					nft = &model.NhNftHolder{
-						Address:   o.OwnerAddress,
-						TokenId:   tk.TokenID,
-						Balance:   balance,
-						UpdateSeq: updateSeq,
-					}
-				} else {
-					logx.Errorw("find nft holder error", logx.Field("error", err), logx.Field("address", o.OwnerAddress), logx.Field("token", tk.TokenID))
-					continue
-				}
-			}
-
-			var value int
-			if nft.Id == 0 {
-				// 新增
-				_, err = c.svcCtx.NftHolderModel.Insert(c.ctx, nft)
-				value = balance
-			} else {
-				// 持有数量变化
-				value = balance - nft.Balance
-				if value != 0 {
-					nft.Balance = balance
-					nft.UpdateSeq = updateSeq
-					err = c.svcCtx.NftHolderModel.Update(c.ctx, nft)
-				}
-			}
-
-			if err != nil {
-				logx.Errorw("insert or update nft holder error", logx.Field("error", err), logx.Field("address", o.OwnerAddress), logx.Field("token", tk.TokenID), logx.Field("balance", balance))
-			}
-
-			// 余额有变化，记录变化日志
-			if value != 0 {
-				_, err = c.svcCtx.NftHolderChangeLogModel.Insert(c.ctx, &model.NhNftHolderChangeLog{
-					Address: o.OwnerAddress,
-					TokenId: tk.TokenID,
-					Value:   balance,
-					Balance: balance,
-				})
-				if err != nil {
-					logx.Errorw("insert nft holder change log error", logx.Field("error", err), logx.Field("address", o.OwnerAddress), logx.Field("token", tk.TokenID))
-				}
-			}
-		}
-	}
-
-	// 删除已经不持有的地址，且添加变化日志
-	nfts, err := c.svcCtx.NftHolderModel.FindOtherUpdateSeq(c.ctx, updateSeq)
-	if err != nil {
-		logx.Errorw("find other update seq error", logx.Field("error", err))
-		return
-	}
-	for _, nft := range nfts {
-		_, err = c.svcCtx.NftHolderChangeLogModel.Insert(c.ctx, &model.NhNftHolderChangeLog{
-			Address: nft.Address,
-			TokenId: nft.TokenId,
-			Value:   -nft.Balance,
-			Balance: 0,
-		})
-		if err != nil {
-			logx.Errorw("delete nft holder error", logx.Field("error", err), logx.Field("address", nft.Address), logx.Field("token", nft.TokenId))
-		}
-		uid, err := c.svcCtx.WalletModel.FindUidByAddress(c.ctx, nft.Address)
-		if err != nil {
-			logx.Errorw("find uid by address error", logx.Field("error", err), logx.Field("address", nft.Address))
-			continue
-		}
-		err = c.svcCtx.StakeNftModel.UnStakeNft(c.ctx, uid, nft.TokenId, true)
-		if err != nil {
-			logx.Errorw("un stake nft error", logx.Field("error", err), logx.Field("address", nft.Address), logx.Field("token", nft.TokenId))
-		}
-	}
-	err = c.svcCtx.NftHolderModel.DeleteOtherUpdateSeq(c.ctx, updateSeq)
-	if err != nil {
-		logx.Errorw("delete other update seq error", logx.Field("error", err), logx.Field("update_seq", updateSeq))
-	}
+	lg := nft.NewHolderUpdateLogic(c.ctx, c.svcCtx)
+	lg.HolderUpdate()
 }