email reward

internal/logic/admin/add_email_reward_logic.go

@@ -0,0 +1,42 @@
+package admin
+
+import (
+	"context"
+	"github.com/shopspring/decimal"
+	"nova_task/internal/model"
+	"nova_task/internal/pkg/errs"
+
+	"nova_task/internal/svc"
+	"nova_task/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type AddEmailRewardLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+// 每日钱包签到任务
+func NewAddEmailRewardLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AddEmailRewardLogic {
+	return &AddEmailRewardLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AddEmailRewardLogic) AddEmailReward(req *types.EmailReward) error {
+	_, err := l.svcCtx.EmailRewardModel.Insert(l.ctx, &model.NhEmailReward{
+		Email:      req.Email,
+		RewardType: req.RewardType,
+		Value:      decimal.NewFromFloat(req.Value),
+	})
+	if err != nil {
+		l.Errorw("add email reward failed", logx.Field("err", err), logx.Field("email", req.Email))
+		return errs.New(errs.ErrDatabaseOperate, err)
+	}
+
+	return errs.Success()
+}

internal/logic/admin/send_email_reward_logic.go

@@ -0,0 +1,84 @@
+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
+}