增加软质押手动结算测试接口
This commit is contained in:
lianghuanjie
158
internal/logic/nft/holder_update_logic.go
Normal file
158
internal/logic/nft/holder_update_logic.go
Normal file
@@ -0,0 +1,158 @@
|
||||
package nft
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/spf13/cast"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
"net/http"
|
||||
"nova_task/internal/consts"
|
||||
"nova_task/internal/model"
|
||||
"nova_task/internal/svc"
|
||||
"time"
|
||||
)
|
||||
|
||||
type HolderUpdateLogic struct {
|
||||
ctx context.Context
|
||||
svcCtx *svc.ServiceContext
|
||||
}
|
||||
|
||||
func NewHolderUpdateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *HolderUpdateLogic {
|
||||
return &HolderUpdateLogic{
|
||||
ctx: ctx,
|
||||
svcCtx: svcCtx,
|
||||
}
|
||||
}
|
||||
|
||||
func (l *HolderUpdateLogic) HolderUpdate() {
|
||||
conf := struct {
|
||||
Network string `json:"network"`
|
||||
Contract string `json:"contract"`
|
||||
}{}
|
||||
cf, err := l.svcCtx.ConfigModel.FindOneByName(l.ctx, consts.NftHolderApiConf)
|
||||
if err != nil {
|
||||
logx.Errorw("find nft holder api conf error", logx.Field("error", err))
|
||||
return
|
||||
}
|
||||
|
||||
err = json.Unmarshal([]byte(cf.Value), &conf)
|
||||
if err != nil {
|
||||
logx.Errorw("unmarshal nft holder api conf error", logx.Field("error", err), logx.Field("value", cf.Value))
|
||||
return
|
||||
}
|
||||
|
||||
ols, err := l.GetOwnerList(conf.Network, conf.Contract)
|
||||
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 := l.svcCtx.NftHolderModel.FindOneByTokenId(l.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 = l.svcCtx.NftHolderModel.Insert(l.ctx, nft)
|
||||
value = balance
|
||||
} else {
|
||||
// 持有数量变化
|
||||
value = balance - nft.Balance
|
||||
if value != 0 {
|
||||
nft.Balance = balance
|
||||
nft.UpdateSeq = updateSeq
|
||||
err = l.svcCtx.NftHolderModel.Update(l.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 = l.svcCtx.NftHolderChangeLogModel.Insert(l.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 := l.svcCtx.NftHolderModel.FindOtherUpdateSeq(l.ctx, updateSeq)
|
||||
if err != nil {
|
||||
logx.Errorw("find other update seq error", logx.Field("error", err))
|
||||
return
|
||||
}
|
||||
for _, nft := range nfts {
|
||||
_, err = l.svcCtx.NftHolderChangeLogModel.Insert(l.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 := l.svcCtx.WalletModel.FindUidByAddress(l.ctx, nft.Address)
|
||||
if err != nil {
|
||||
logx.Errorw("find uid by address error", logx.Field("error", err), logx.Field("address", nft.Address))
|
||||
continue
|
||||
}
|
||||
err = l.svcCtx.StakeNftModel.UnStakeNft(l.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 = l.svcCtx.NftHolderModel.DeleteOtherUpdateSeq(l.ctx, updateSeq)
|
||||
if err != nil {
|
||||
logx.Errorw("delete other update seq error", logx.Field("error", err), logx.Field("update_seq", updateSeq))
|
||||
}
|
||||
}
|
||||
|
||||
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 (l *HolderUpdateLogic) GetOwnerList(network, contractAddress string) (*OwnerList, error) {
|
||||
url := fmt.Sprintf("https://eth-%s.g.alchemy.com/nft/v3/alcht_1a183fAsPqF9upfTfp3AC1l0iedGLo/getOwnersForContract?contractAddress=%s&withTokenBalances=true", network, 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
|
||||
}
|
||||
114
internal/logic/nft/stake_settle_logic.go
Normal file
114
internal/logic/nft/stake_settle_logic.go
Normal file
@@ -0,0 +1,114 @@
|
||||
package nft
|
||||
|
||||
import (
|
||||
"context"
|
||||
"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/svc"
|
||||
"time"
|
||||
)
|
||||
|
||||
type StakeSettleLogic struct {
|
||||
ctx context.Context
|
||||
svcCtx *svc.ServiceContext
|
||||
}
|
||||
|
||||
func NewStakeSettleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StakeSettleLogic {
|
||||
return &StakeSettleLogic{
|
||||
ctx: ctx,
|
||||
svcCtx: svcCtx,
|
||||
}
|
||||
}
|
||||
|
||||
func (l *StakeSettleLogic) StakeSettle() {
|
||||
start, end, err := l.svcCtx.ConfigModel.GetNftStakeTaskOpenDate(l.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.Infow("now is not in the date range", logx.Field("now", now), logx.Field("start", start), logx.Field("end", end))
|
||||
return
|
||||
}
|
||||
|
||||
taskConf, err := l.svcCtx.ConfigModel.GetNftStakeTaskConf(l.ctx)
|
||||
if err != nil {
|
||||
logx.Errorw("get nft stake task conf failed", logx.Field("err", err))
|
||||
return
|
||||
}
|
||||
|
||||
stakes, err := l.svcCtx.StakeNftModel.AllStakeNft(l.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 := l.svcCtx.NftHolderModel.FindOneByTokenId(l.ctx, s.TokenId)
|
||||
if err != nil {
|
||||
logx.Errorw("find nft holder failed", logx.Field("err", err), logx.Field("tokenId", s.TokenId))
|
||||
continue
|
||||
}
|
||||
|
||||
uid, err := l.svcCtx.WalletModel.FindUidByAddress(l.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 := l.svcCtx.StakeRewardModel.GetRandomCoefficientByUid(l.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 = l.svcCtx.StakeRewardModel.SetReward(l.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 = l.svcCtx.TaskAssetModel.AddCastile(l.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 = l.svcCtx.TaskAssetRecordModel.Insert(l.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))
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user