package logic import ( "admin/internal/dao" "admin/internal/model" "admin/validator" "context" "strconv" "admin/admin" "admin/internal/svc" "github.com/zeromicro/go-zero/core/logx" "pkg.local/modelbase" "pkg.local/utils" ) type ServiceEditLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewServiceEditLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ServiceEditLogic { return &ServiceEditLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *ServiceEditLogic) ServiceEdit(in *admin.ServiceEditRequest) (*admin.Response, error) { var v validator.ServiceEditValidator 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) } edit := dao.Services{ Name: in.Name, Description: in.Description, AdminId: adminInfo.ID, AdminName: adminInfo.Name, } modelObj := model.ServicesModel{}.Init() var err error if in.Id > utils.NumberZero { w := modelbase.Params{Eq: map[string]string{"id": strconv.Itoa(int(in.Id))}} info := dao.Services{} err = modelObj.GetOne(w, &info) if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } if info.Id < utils.NumberOne { return l.fail(utils.ErrorNotFund) } edit.Id = info.Id _, err = modelObj.Edit(w, &edit) } else { err = modelObj.Create(&edit) } if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } return l.ok(edit.Id) }