package logic import ( "context" "lone-services/pkg/utils" "lone-services/pkg/validate" ad "lone-services/rpc/ad/pb" "lone-services/services/ad/internal/dao" "lone-services/services/ad/internal/model" "lone-services/services/ad/internal/svc" "lone-services/services/ad/validator" "github.com/zeromicro/go-zero/core/logx" ) type CreateLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger } func NewCreateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateLogic { return &CreateLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *CreateLogic) Create(in *ad.CreateReq) (*ad.Response, error) { var req validator.AdCreateValidator if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty { return outResponse(utils.ErrorParams, msg), nil } adminInfo := utils.GetUserFromCtx(l.ctx) if adminInfo.ID < utils.NumberOne { return failResponse(utils.ErrorNoLoginInfo), nil } data := dao.AdCreate{ Type: uint8(req.Type), Image: req.Image, Content: req.Content, Url: req.Url, AdminId: int(adminInfo.ID), AdminName: adminInfo.Name, } if err := (model.AdModel{}.Init().Create(&data)); err != nil { l.Errorf("create ad: %v", err) return failResponse(utils.Fail), nil } return okResponse(nil), nil }