93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/spf13/cast"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
"nova_task/internal/model"
|
|
"nova_task/internal/pkg/errs"
|
|
"time"
|
|
|
|
"nova_task/internal/svc"
|
|
"nova_task/internal/types"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
// GetTaskRewardLogic v领取任务奖励
|
|
type GetTaskRewardLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGetTaskRewardLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetTaskRewardLogic {
|
|
return &GetTaskRewardLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetTaskRewardLogic) GetTaskReward(req *types.TaskIdPath) (*types.GetTaskRewardResp, error) {
|
|
uid := cast.ToStringMapInt(l.ctx.Value("data"))["id"]
|
|
task, err := l.svcCtx.TaskModel.FindOne(l.ctx, req.ID)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
return nil, errs.BadRequest(errs.ErrTaskNotFound, "task not found")
|
|
}
|
|
return nil, errs.InternalServer(errs.ErrDatabaseOperate, err)
|
|
}
|
|
var taskSeq int
|
|
if task.Type == model.TASKTYPE_DAILY_PAY {
|
|
taskSeq = cast.ToInt(time.Now().Format("20060102"))
|
|
}
|
|
tp, err := l.svcCtx.TaskProgressModel.FindOneByUidTaskIdTaskSeq(l.ctx, uid, task.Id, taskSeq)
|
|
if err != nil {
|
|
if !errors.Is(err, model.ErrNotFound) {
|
|
l.Errorw("find task progress error", logx.Field("err", err))
|
|
return nil, errs.InternalServer(errs.ErrDatabaseOperate, err)
|
|
}
|
|
return nil, errs.BadRequest(errs.ErrTaskNotFinished, "task not finished")
|
|
}
|
|
switch {
|
|
case tp.Stage < model.TASK_PROGRESS_WAIT_REWARD:
|
|
return nil, errs.BadRequest(errs.ErrTaskNotFinished, "task not finished")
|
|
case tp.Stage > model.TASK_PROGRESS_WAIT_REWARD:
|
|
return nil, errs.BadRequest(errs.ErrTaskAlreadyReward, "task already reward")
|
|
}
|
|
|
|
// 修改状态,增加积分和记录都在事物中执行
|
|
err = l.svcCtx.DBConn.TransactCtx(l.ctx, func(ctx context.Context, session sqlx.Session) error {
|
|
// 修改状态
|
|
tp.Stage = model.TASK_PROGRESS_REWARDED
|
|
err = l.svcCtx.TaskProgressModel.WithSession(session).Update(l.ctx, tp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 给予用户奖励
|
|
err = l.svcCtx.TaskAssetModel.WithSession(session).AddUserPoint(l.ctx, uid, task.Points)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 记录用户获奖记录
|
|
_, err = l.svcCtx.TaskAssetRecordModel.Insert(l.ctx, &model.NhTaskAssetRecord{
|
|
Uid: uid,
|
|
EventId: uint64(task.Id),
|
|
AssetField: "points",
|
|
Count: float64(task.Points),
|
|
Remark: "完成任务：" + task.Title,
|
|
CreateTime: int(time.Now().Unix()),
|
|
})
|
|
return err
|
|
})
|
|
|
|
if err != nil {
|
|
l.Errorw("给予用户奖励实物执行失败", logx.Field("err", err), logx.Field("task", task.Id), logx.Field("uid", uid))
|
|
return nil, errs.InternalServer(errs.ErrDatabaseOperate, err)
|
|
}
|
|
|
|
return &types.GetTaskRewardResp{Points: task.Points}, nil
|
|
}
|