package logic import ( "context" "lone-services/pkg/modelbase" "lone-services/pkg/utils" "lone-services/services/water/internal/dao" "lone-services/services/water/model" "lone-services/services/water/validator" "lone-services/rpc/water/pb" "lone-services/services/water/internal/svc" "github.com/zeromicro/go-zero/core/logx" ) type ApplyUserLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewApplyUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApplyUserLogic { return &ApplyUserLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } // 申请 func (l *ApplyUserLogic) ApplyUser(in *water.ApplyUserRequest) (*water.Response, error) { var v validator.UserApplyValidator 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) } w := modelbase.Params{Eq: map[string]string{"user_id": string(in.UserId), "status": utils.StringStatusOk}} var info dao.UsersCreate err := model.UsersModel{}.Init().GetOne(w, &info) if err != nil { return l.fail(utils.Fail) } if info.Id > utils.NumberZero { return l.fail(utils.ErrorExist) } info.UserId = in.UserId info.HostCode = in.HostCode err = model.UsersModel{}.Init().Create(&info) if err != nil { return l.fail(utils.Fail) } return l.ok(utils.NumberOne) }