《完善用户信息》修改用户登录错误提示

ruoyi-system/src/main/java/com/ruoyi/system/service/SysLoginService.java

@@ -75,7 +75,7 @@ public class SysLoginService {
             validateCaptcha(username, code, uuid, request);
             validateCaptcha(username, code, uuid, request);
         }
         }
         SysUser user = loadUserByUsername(username);
         SysUser user = loadUserByUsername(username);
-        checkLogin(LoginType.PASSWORD, username, () -> !BCrypt.checkpw(password, user.getPassword()));
+        checkLoginNew(LoginType.PASSWORD, username, () -> !BCrypt.checkpw(password, user.getPassword()));
         // 此处可根据登录用户的数据不同 自行创建 loginUser
         // 此处可根据登录用户的数据不同 自行创建 loginUser
         LoginUser loginUser = buildLoginUser(user);
         LoginUser loginUser = buildLoginUser(user);
         // 生成token
         // 生成token
@@ -292,4 +292,39 @@ public class SysLoginService {
         // 登录成功 清空错误次数
         // 登录成功 清空错误次数
         RedisUtils.deleteObject(errorKey);
         RedisUtils.deleteObject(errorKey);
     }
     }
+
+    /**
+     * 登录校验
+     */
+    private void checkLoginNew(LoginType loginType, String username, Supplier<Boolean> supplier) {
+        String errorKey = CacheConstants.PWD_ERR_CNT_KEY + username;
+        String loginFail = Constants.LOGIN_FAIL;
+
+        // 获取用户登录错误次数(可自定义限制策略 例如: key + username + ip)
+        Integer errorNumber = RedisUtils.getCacheObject(errorKey);
+        // 锁定时间内登录 则踢出
+        if (ObjectUtil.isNotNull(errorNumber) && errorNumber.equals(maxRetryCount)) {
+            recordLogininfor(username, loginFail, MessageUtils.message(loginType.getRetryLimitExceed(), maxRetryCount, lockTime));
+            throw new ServiceException("账号密码错误"+maxRetryCount +"次");
+        }
+
+        if (supplier.get()) {
+            // 是否第一次
+            errorNumber = ObjectUtil.isNull(errorNumber) ? 1 : errorNumber + 1;
+            // 达到规定错误次数 则锁定登录
+            if (errorNumber.equals(maxRetryCount)) {
+                RedisUtils.setCacheObject(errorKey, errorNumber, Duration.ofMinutes(lockTime));
+                recordLogininfor(username, loginFail, MessageUtils.message(loginType.getRetryLimitExceed(), maxRetryCount, lockTime));
+                throw new ServiceException("账号密码错误"+maxRetryCount +"次");
+            } else {
+                // 未达到规定错误次数 则递增
+                RedisUtils.setCacheObject(errorKey, errorNumber);
+                recordLogininfor(username, loginFail, MessageUtils.message(loginType.getRetryLimitCount(), errorNumber));
+                throw new ServiceException("账号密码错误"+errorNumber +"次");
+            }
+        }
+
+        // 登录成功 清空错误次数
+        RedisUtils.deleteObject(errorKey);
+    }
 }
 }