add

wangpengfei

2023-07-14 5f00d720ea52ababeb10954da4d33ab6c46e93b7

 api/v1/user.go

@@ -6,9 +6,7 @@
   "aps_crm/model"
   "aps_crm/model/request"
   "aps_crm/model/response"
   _ "aps_crm/model/response"
   "aps_crm/pkg/contextx"
   "aps_crm/pkg/convertx"
   "aps_crm/pkg/ecode"
   "aps_crm/pkg/encrypt"
   "aps_crm/pkg/logx"
@@ -18,7 +16,6 @@
   "github.com/gin-gonic/gin"
   "github.com/go-redis/redis/v8"
   "github.com/mojocn/base64Captcha"
   "time"
)

// 当开启多服务器部署时，替换下面的配置，使用redis共享存储验证码
@@ -26,6 +23,7 @@
var store = base64Captcha.DefaultMemStore

// Login
//
//   @Tags      Base
//   @Summary   用户登录
//   @Produce   application/json
@@ -41,41 +39,41 @@

   // 判断验证码是否开启
   key := c.ClientIP()
   openCaptcha := conf.Conf.Captcha.OpenCaptcha               // 是否开启防爆次数
   openCaptchaTimeOut := conf.Conf.Captcha.OpenCaptchaTimeOut // 缓存超时时间
   v, ok := constvar.BlackCache.Get(key)
   if !ok {
      constvar.BlackCache.Set(key, 1, time.Second*time.Duration(openCaptchaTimeOut))
   }
   //openCaptcha := conf.Conf.Captcha.OpenCaptcha               // 是否开启防暴次数
   //openCaptchaTimeOut := conf.Conf.Captcha.OpenCaptchaTimeOut // 缓存超时时间
   //v, ok := constvar.BlackCache.Get(key)
   //if !ok {
   //   constvar.BlackCache.Set(key, 1, time.Second*time.Duration(openCaptchaTimeOut))
   //}

   var oc bool = openCaptcha == 0 || convertx.InterfaceToInt(v) > openCaptcha // 0 表示每次登录都需要验证码 或者当前次数已超过防爆次数
   //var oc = openCaptcha == 0 || convertx.InterfaceToInt(v) > openCaptcha // 0 表示每次登录都需要验证码 或者当前次数已超过防暴次数

   if !oc || store.Verify(params.CaptchaId, params.Captcha, true) {
      u := &model.User{Username: params.Username, Password: params.Password}
      user, errCode := userService.Login(u)
      if errCode != ecode.OK {
         logx.Errorf("登陆失败! 用户名不存在或者密码错误! errCode:%v", errCode)
         // 验证码次数+1
         _ = constvar.BlackCache.Increment(key, 1)
         ctx.Fail(errCode)
         return
      }
      if !user.Enable {
         logx.Errorf("登陆失败! 用户被禁止登录!")
         // 验证码次数+1
         _ = constvar.BlackCache.Increment(key, 1)
         ctx.Fail(ecode.UserForbidden)
         return
      }
      // 赋值菜单ID列表
      //user.MenuIds, _ = menuService.GetUserMenuIds(user.ID, user.UserType)
      slf.TokenNext(ctx, *user)
   //if !oc || store.Verify(params.CaptchaId, params.Captcha, true) {
   u := &model.User{Username: params.Username, Password: params.Password}
   user, errCode := userService.Login(u)
   if errCode != ecode.OK {
      logx.Errorf("登陆失败! 用户名不存在或者密码错误! errCode:%v", errCode)
      // 验证码次数+1
      _ = constvar.BlackCache.Increment(key, 1)
      ctx.Fail(errCode)
      return
   }
   if !user.Enable {
      logx.Errorf("登陆失败! 用户被禁止登录!")
      // 验证码次数+1
      _ = constvar.BlackCache.Increment(key, 1)
      ctx.Fail(ecode.UserForbidden)
      return
   }
   // 赋值菜单ID列表
   //user.MenuIds, _ = menuService.GetUserMenuIds(user.ID, user.UserType)
   slf.TokenNext(ctx, *user)
   return
   //}

   // 验证码次数+1
   _ = constvar.BlackCache.Increment(key, 1)
   ctx.Fail(ecode.CaptchaErr)
   //_ = constvar.BlackCache.Increment(key, 1)
   //ctx.Fail(ecode.CaptchaErr)
}

// TokenNext 登录以后签发jwt
@@ -137,6 +135,7 @@
}

// Register
//
//   @Tags      User
//   @Summary   注册账号
//   @Produce   application/json
@@ -204,6 +203,7 @@
}

// ChangePassword
//
//   @Tags      User
//   @Summary   用户修改密码
//   @Produce   application/json
@@ -227,6 +227,7 @@
}

// GetUserList
//
//   @Tags      User
//   @Summary   分页获取用户列表(不传分页参数，获取全部)
//   @Produce   application/json
@@ -253,6 +254,7 @@
}

// DeleteUser
//
//   @Tags      User
//   @Summary   删除用户
//   @Produce   application/json
@@ -287,6 +289,7 @@
}

// SetUserInfo
//
//   @Tags      User
//   @Summary   设置用户信息
//   @Produce   application/json
@@ -317,6 +320,7 @@
}

// SetSelfInfo
//
//   @Tags      User
//   @Summary   设置用户信息
//   @Produce   application/json
@@ -348,6 +352,7 @@
}

// GetUserInfo
//
//   @Tags      User
//   @Summary   获取自身信息
//   @Produce   application/json
@@ -372,6 +377,7 @@
}

// ResetPassword
//
//   @Tags      User
//   @Summary   重置用户密码
//   @Produce   application/json