121 lines
3.7 KiB
Go
121 lines
3.7 KiB
Go
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/svc"
|
|
"time"
|
|
)
|
|
|
|
type Cron struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewCron(ctx context.Context, svcCtx *svc.ServiceContext) cron.Job {
|
|
return &Cron{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (c *Cron) Spec() string {
|
|
return c.svcCtx.Config.NftTaskCron.SettleSpec
|
|
}
|
|
|
|
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))
|
|
}
|
|
}
|
|
}
|