jager/novatask
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

nft质押任务逻辑

Browse Source
This commit is contained in:
lianghuanjie
2024-12-27 18:06:13 +08:00
parent a22f73df20
commit 31a674080d
39 changed files with 1532 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
internal/job/earn/earn.go → internal/job/earn/cron.go
View File

@@ -3,65 +3,66 @@ package earn
import (
"context"
ea "github.com/earn-alliance/earnalliance-go"
"github.com/robfig/cron/v3"
"github.com/spf13/cast"
"github.com/zeromicro/go-zero/core/logx"
"nova_task/internal/consts"
"nova_task/internal/svc"
)
// Earn 用户数据上报earn平台定时任务
type Earn struct {
// Cron 用户数据上报earn平台定时任务
type Cron struct {
ctx context.Context
svcCtx *svc.ServiceContext
}
func NewEarn(ctx context.Context, svcCtx *svc.ServiceContext) *Earn {
e := &Earn{ctx: ctx, svcCtx: svcCtx}
func NewCron(ctx context.Context, svcCtx *svc.ServiceContext) cron.Job {
e := &Cron{ctx: ctx, svcCtx: svcCtx}
if e.svcCtx.Config.EarnCorn.RunOnStart {
e.Run()
}
return e
}
func (e *Earn) Spec() string {
return e.svcCtx.Config.EarnCorn.Spec
func (c *Cron) Spec() string {
return c.svcCtx.Config.EarnCorn.Spec
}
func (e *Earn) Run() {
func (c *Cron) Run() {
logx.Debugw("run earn cron task")
c, err := e.svcCtx.ConfigModel.FindOneByName(e.ctx, consts.EarnAllianceInviterId)
conf, err := c.svcCtx.ConfigModel.FindOneByName(c.ctx, consts.EarnAllianceInviterId)
if err != nil {
logx.Errorw("find earn alliance inviter id failed", logx.Field("err", err))
return
}
e.pushUserInfo(cast.ToUint(c.Value))
e.pushUserBind(cast.ToUint(c.Value))
c.pushUserInfo(cast.ToUint(conf.Value))
c.pushUserBind(cast.ToUint(conf.Value))
}
func (e *Earn) pushUserInfo(shareId uint) {
us, err := e.svcCtx.PromoteBindModel.FindRequirePushUser(e.ctx, shareId)
func (c *Cron) pushUserInfo(shareId uint) {
us, err := c.svcCtx.PromoteBindModel.FindRequirePushUser(c.ctx, shareId)
if err != nil {
logx.Errorw("find require push user failed", logx.Field("err", err), logx.Field("share_id", shareId))
return
}
logx.Debugw("find require push user", logx.Field("count", len(us)))
for _, u := range us {
ui, err := e.svcCtx.UserModel.FindOne(e.ctx, u.InvitedUid)
ui, err := c.svcCtx.UserModel.FindOne(c.ctx, u.InvitedUid)
if err != nil {
logx.Errorw("find user failed", logx.Field("err", err), logx.Field("uid", u.InvitedUid))
continue
}
var twitterId string
ut, err := e.svcCtx.TwitterModel.FindOne(e.ctx, u.InvitedUid)
ut, err := c.svcCtx.TwitterModel.FindOne(c.ctx, u.InvitedUid)
if err == nil {
twitterId = ut.TwitterId
}
err = e.svcCtx.PromoteBindModel.UpdatePushUser(e.ctx, u.Id)
err = c.svcCtx.PromoteBindModel.UpdatePushUser(c.ctx, u.Id)
if err != nil {
logx.Errorw("update push user failed", logx.Field("err", err), logx.Field("uid", u.InvitedUid), logx.Field("twitter_id", twitterId), logx.Field("email", ui.Email))
} else {
e.svcCtx.Earn.SetIdentifiers(cast.ToString(ui.Id), &ea.Identifiers{
c.svcCtx.Earn.SetIdentifiers(cast.ToString(ui.Id), &ea.Identifiers{
Email: ea.IdentifierFrom(ui.Email),
TwitterId: ea.IdentifierFrom(twitterId),
})
@@ -70,19 +71,19 @@ func (e *Earn) pushUserInfo(shareId uint) {
}
}
func (e *Earn) pushUserBind(shareId uint) {
us, err := e.svcCtx.TouristBindModel.FindRequirePushUser(e.ctx, shareId)
func (c *Cron) pushUserBind(shareId uint) {
us, err := c.svcCtx.TouristBindModel.FindRequirePushUser(c.ctx, shareId)
if err != nil {
logx.Errorw("find require push bind role user failed", logx.Field("err", err), logx.Field("share_id", shareId))
return
}
logx.Debugw("find require push bind role user", logx.Field("count", len(us)))
for _, u := range us {
err = e.svcCtx.PromoteBindModel.UpdatePushRole(e.ctx, u.Id)
err = c.svcCtx.PromoteBindModel.UpdatePushRole(c.ctx, u.Id)
if err != nil {
logx.Errorw("update push user failed", logx.Field("err", err), logx.Field("uid", u.Uid))
} else {
e.svcCtx.Earn.Track(cast.ToString(u.Uid), "BIND_ROLE", nil, nil)
c.svcCtx.Earn.Track(cast.ToString(u.Uid), "BIND_ROLE", nil, nil)
logx.Infow("push user info success", logx.Field("uid", u.Uid))
}
}