增加社区列表接口,任务相关接口修改
This commit is contained in:
61
internal/job/earn/earn.go
Normal file
61
internal/job/earn/earn.go
Normal file
@@ -0,0 +1,61 @@
|
||||
package earn
|
||||
|
||||
import (
|
||||
"context"
|
||||
ea "github.com/earn-alliance/earnalliance-go"
|
||||
"github.com/spf13/cast"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
"nova_task/internal/svc"
|
||||
)
|
||||
|
||||
// Earn 用户数据上报earn平台定时任务
|
||||
type Earn struct {
|
||||
ctx context.Context
|
||||
svcCtx *svc.ServiceContext
|
||||
}
|
||||
|
||||
func NewEarn(ctx context.Context, svcCtx *svc.ServiceContext) *Earn {
|
||||
e := &Earn{ctx: ctx, svcCtx: svcCtx}
|
||||
e.Run()
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *Earn) Spec() string {
|
||||
return "@every 5m"
|
||||
}
|
||||
|
||||
func (e *Earn) Run() {
|
||||
e.pushUserInfo(666)
|
||||
e.pushUserBind()
|
||||
}
|
||||
|
||||
func (e *Earn) pushUserInfo(shareId uint) {
|
||||
us, err := e.svcCtx.PromoteBindModel.FindRequirePushUser(e.ctx, shareId)
|
||||
if err != nil {
|
||||
logx.Errorw("find require push user failed", logx.Field("err", err), logx.Field("share_id", shareId))
|
||||
return
|
||||
}
|
||||
for _, u := range us {
|
||||
ui, err := e.svcCtx.UserModel.FindOne(e.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)
|
||||
if err == nil {
|
||||
twitterId = ut.TwitterId
|
||||
}
|
||||
|
||||
e.svcCtx.Earn.SetIdentifiers(cast.ToString(ui.Id), &ea.Identifiers{
|
||||
Email: ea.IdentifierFrom(ui.Email),
|
||||
TwitterId: ea.IdentifierFrom(twitterId),
|
||||
})
|
||||
err = e.svcCtx.PromoteBindModel.UpdatePushUser(e.ctx, u.Id)
|
||||
if err != nil {
|
||||
logx.Errorw("update push user failed", logx.Field("err", err), logx.Field("uid", u.InvitedUid))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (e *Earn) pushUserBind() {}
|
||||
63
internal/job/job.go
Normal file
63
internal/job/job.go
Normal file
@@ -0,0 +1,63 @@
|
||||
package job
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"github.com/robfig/cron/v3"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
"nova_task/internal/job/earn"
|
||||
"nova_task/internal/job/pledge"
|
||||
"nova_task/internal/svc"
|
||||
)
|
||||
|
||||
type Job struct {
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
svcCtx *svc.ServiceContext
|
||||
c *cron.Cron
|
||||
}
|
||||
|
||||
type Spec interface {
|
||||
Spec() string
|
||||
}
|
||||
|
||||
func NewJob(svcCtx *svc.ServiceContext) *Job {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
var cronList = []cron.Job{
|
||||
earn.NewEarn(ctx, svcCtx),
|
||||
pledge.NewPledge(ctx, svcCtx),
|
||||
}
|
||||
var err error
|
||||
c := cron.New()
|
||||
for _, cr := range cronList {
|
||||
if cs, ok := cr.(Spec); ok {
|
||||
_, err = c.AddJob(cs.Spec(), cr)
|
||||
logx.Must(err)
|
||||
continue
|
||||
}
|
||||
|
||||
if cs, ok := cr.(cron.Schedule); ok {
|
||||
c.Schedule(cs, cr)
|
||||
continue
|
||||
}
|
||||
logx.Must(errors.New("cron job must implement either Spec or Schedule interface"))
|
||||
}
|
||||
return &Job{
|
||||
ctx: ctx,
|
||||
cancel: cancel,
|
||||
svcCtx: svcCtx,
|
||||
c: c,
|
||||
}
|
||||
}
|
||||
|
||||
func (j *Job) Start() {
|
||||
logx.Info("start cron job")
|
||||
j.c.Start()
|
||||
}
|
||||
|
||||
func (j *Job) Stop() {
|
||||
logx.Info("stop cron job")
|
||||
<-j.c.Stop().Done()
|
||||
logx.Info("cron job stopped")
|
||||
j.cancel()
|
||||
}
|
||||
26
internal/job/pledge/pledge.go
Normal file
26
internal/job/pledge/pledge.go
Normal file
@@ -0,0 +1,26 @@
|
||||
package pledge
|
||||
|
||||
import (
|
||||
"context"
|
||||
"nova_task/internal/svc"
|
||||
)
|
||||
|
||||
type Pledge struct {
|
||||
ctx context.Context
|
||||
svcCtx *svc.ServiceContext
|
||||
}
|
||||
|
||||
func NewPledge(ctx context.Context, svcCtx *svc.ServiceContext) *Pledge {
|
||||
return &Pledge{
|
||||
ctx: ctx,
|
||||
svcCtx: svcCtx,
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Pledge) Spec() string {
|
||||
return "@every 30m"
|
||||
}
|
||||
|
||||
func (p *Pledge) Run() {
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user