package logic import ( "context" "lone-services/pkg/utils" "lone-services/services/task/internal/dao" "lone-services/services/task/internal/model" "lone-services/services/task/validator" "lone-services/rpc/task/pb" "lone-services/services/task/internal/svc" "github.com/zeromicro/go-zero/core/logx" ) type CreateLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewCreateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateLogic { return &CreateLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *CreateLogic) Create(in *task.CreateRequest) (*task.Response, error) { var v validator.TaskCreateValidator if fail := l.checkParams(in, &v); fail != nil { return fail, nil } adminInfo := utils.GetUserFromCtx(l.ctx) if adminInfo.ID < utils.NumberOne { return l.fail(utils.ErrorNoLoginInfo) } data := dao.TaskCreate{ Type: in.Type, Time: in.Time, ProvinceName: in.ProvinceName, ProductId: in.ProductId, Task: in.Task, Award: utils.RoundFloat(in.Award), Province: in.Province, AdminId: adminInfo.ID, AdminName: adminInfo.Name, } err := model.TaskModel{}.Init().Create(&data) if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } action := utils.ActionAdd{ NewContent: data, Type: utils.LogActionTypeAdd, ModuleName: utils.LogActionModuleTask, } utils.SetActionLog(adminInfo, action) return l.ok(data.Id) }