126 lines
3.7 KiB
Go
126 lines
3.7 KiB
Go
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.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))
|
|
}
|
|
}
|