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 RoleRuleLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewRoleRuleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RoleRuleLogic { return &RoleRuleLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *RoleRuleLogic) RoleRule(in *admin.RoleRuleRequest) (*admin.Response, error) { var v validator.RoleRuleValidator 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) } info := dao.RoleRule{} w := modelbase.Params{Eq: map[string]string{"id": strconv.FormatInt(in.Id, utils.NumberTen)}} modelObj := model.RoleModel{}.Init() 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) } info.Rules = in.Rule info.AdminId = adminInfo.ID info.AdminName = adminInfo.Name _, editErr := modelObj.Edit(w, info) if editErr != nil { l.Logger.Error(editErr) return l.fail(utils.Fail) } return l.ok(utils.NumberOne) }