package logic import ( "context" "lone-services/pkg/modelbase" "lone-services/pkg/utils" admin "lone-services/rpc/admin/pb" "lone-services/services/admin/internal/dao" "lone-services/services/admin/internal/model" "lone-services/services/admin/internal/svc" "lone-services/services/admin/validator" "strconv" "github.com/zeromicro/go-zero/core/logx" ) 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 var row int64 var action utils.ActionAdd 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) } action.OldContent = info edit.Id = info.Id row, err = modelObj.Edit(w, &edit) action.Type = utils.LogActionTypeEdit } else { err = modelObj.Create(&edit) row = edit.Id action.Type = utils.LogActionTypeAdd } if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } if row > utils.NumberZero { action.NewContent = edit action.ModuleName = utils.LogActionModuleService utils.SetActionLog(adminInfo, action) } return l.ok(edit.Id) }