nft质押任务逻辑

internal/job/holder/cron.go

@@ -0,0 +1,116 @@
+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/svc"
+	"time"
+)
+
+type Cron struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewCron(ctx context.Context, svcCtx *svc.ServiceContext) cron.Job {
+	pg := &Cron{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+	if svcCtx.Config.NftTaskCron.HolderCheckRunOnStart {
+		pg.Run()
+	}
+	return pg
+}
+
+func (c *Cron) Spec() string {
+	return c.svcCtx.Config.NftTaskCron.HolderSpec
+}
+
+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.FindOneByAddressTokenId(c.ctx, o.OwnerAddress, 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))
+		}
+	}
+	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))
+	}
+}

internal/job/holder/nft.go

@@ -0,0 +1,31 @@
+package holder
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+)
+
+type OwnerList struct {
+	Owners []struct {
+		OwnerAddress  string `json:"ownerAddress"`
+		TokenBalances []struct {
+			TokenID string `json:"tokenId"`
+			Balance string `json:"balance"`
+		} `json:"tokenBalances"`
+	} `json:"owners"`
+	PageKey string `json:"pageKey"`
+}
+
+func GetOwnerList() (*OwnerList, error) {
+	contractAddress := "0x89B8D549feA2eBd2aA0b375ce0DCaBba79e7e636"
+	url := fmt.Sprintf("https://eth-mainnet.g.alchemy.com/nft/v3/alcht_1a183fAsPqF9upfTfp3AC1l0iedGLo/getOwnersForContract?contractAddress=%s&withTokenBalances=true", contractAddress)
+	resp, err := http.Get(url)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+	result := new(OwnerList)
+	err = json.NewDecoder(resp.Body).Decode(result)
+	return result, err
+}