Files
novatask/internal/logic/admin/send_email_reward_logic.go
lianghuanjie b098e50eb0 email reward
2025-01-03 20:44:03 +08:00

85 lines
2.4 KiB
Go

package admin
import (
"context"
"github.com/zeromicro/go-zero/core/threading"
"nova_task/internal/model"
"nova_task/internal/pkg/errs"
"strings"
"time"
"github.com/zeromicro/go-zero/core/logx"
"nova_task/internal/svc"
)
type SendEmailRewardLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
// 执行发放奖励操作
func NewSendEmailRewardLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendEmailRewardLogic {
return &SendEmailRewardLogic{
Logger: logx.WithContext(ctx),
ctx: ctx,
svcCtx: svcCtx,
}
}
func (l *SendEmailRewardLogic) SendEmailReward() error {
rewards, err := l.svcCtx.EmailRewardModel.AllRequireSend(l.ctx)
if err != nil {
l.Errorw("get email reward failed", logx.Field("err", err))
return errs.New(errs.ErrDatabaseOperate, err)
}
if len(rewards) <= 0 {
return errs.Success()
}
threading.GoSafe(func() {
ctx := context.Background()
for _, rw := range rewards {
u, err := l.svcCtx.UserModel.FindOneByEmail(ctx, rw.Email)
if err != nil {
l.Errorw("find user failed", logx.Field("err", err), logx.Field("email", rw.Email))
continue
}
err = l.svcCtx.EmailRewardModel.SetSent(ctx, rw.Id, u.Id)
if err != nil {
l.Errorw("set sent failed", logx.Field("err", err), logx.Field("id", rw.Id), logx.Field("uid", u.Id))
continue
}
// points,elite_points,castile,keys
switch strings.ToLower(rw.RewardType) {
case "points":
err = l.svcCtx.TaskAssetModel.AddPoint(ctx, u.Id, rw.Value)
case "elite_points":
err = l.svcCtx.TaskAssetModel.AddElitePoints(ctx, u.Id, rw.Value)
case "castile":
err = l.svcCtx.TaskAssetModel.AddCastile(ctx, u.Id, rw.Value)
case "keys":
err = l.svcCtx.TaskAssetModel.AddKeys(ctx, u.Id, int(rw.Value.IntPart()))
}
if err != nil {
l.Errorw("add asset failed", logx.Field("err", err), logx.Field("uid", u.Id), logx.Field("rewardType", rw.RewardType), logx.Field("value", rw.Value))
}
_, err = l.svcCtx.TaskAssetRecordModel.Insert(ctx, &model.NhTaskAssetRecord{
Uid: int(u.Id),
AssetField: rw.RewardType,
Count: rw.Value.InexactFloat64(),
Remark: rw.Remark,
CreateTime: int(time.Now().Unix()),
})
if err != nil {
l.Errorw("insert task asset record failed", logx.Field("err", err), logx.Field("uid", u.Id), logx.Field("rewardType", rw.RewardType), logx.Field("value", rw.Value))
}
}
})
return nil
}