springboot整合shiro登录失败次数限制功能的实现代码

package com.springboot.test.shiro.config.shiro;
import java.util.concurrent.atomic.AtomicInteger;
import com.springboot.test.shiro.modules.user.dao.UserMapper;
import com.springboot.test.shiro.modules.user.dao.entity.User;
import org.apache.log4j.Logger;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.LockedAccountException;
import org.apache.shiro.authc.credential.SimpleCredentialsMatcher;
import org.apache.shiro.cache.Cache;
import org.apache.shiro.cache.CacheManager;
import org.springframework.beans.factory.annotation.Autowired;
/**
 * @author: WangSaiChao
 * @date: 2018/5/25
 * @description: 登陆次数限制
 */
public class RetryLimitHashedCredentialsMatcher extends SimpleCredentialsMatcher {
 private static final Logger logger = Logger.getLogger(RetryLimitHashedCredentialsMatcher.class);
 @Autowired
 private UserMapper userMapper;
 private Cache<String, AtomicInteger> passwordRetryCache;
 public RetryLimitHashedCredentialsMatcher(CacheManager cacheManager) {
  passwordRetryCache = cacheManager.getCache("passwordRetryCache");
 }
 @Override
 public boolean doCredentialsMatch(AuthenticationToken token, AuthenticationInfo info) {
  //获取用户名
  String username = (String)token.getPrincipal();
  //获取用户登录次数
  AtomicInteger retryCount = passwordRetryCache.get(username);
  if (retryCount == null) {
   //如果用户没有登陆过,登陆次数加1 并放入缓存
   retryCount = new AtomicInteger(0);
   passwordRetryCache.put(username, retryCount);
  }
  if (retryCount.incrementAndGet() > 5) {
   //如果用户登陆失败次数大于5次 抛出锁定用户异常 并修改数据库字段
   User user = userMapper.findByUserName(username);
   if (user != null && "0".equals(user.getState())){
    //数据库字段 默认为 0 就是正常状态 所以 要改为1
    //修改数据库的状态字段为锁定
    user.setState("1");
    userMapper.update(user);
   }
   logger.info("锁定用户" + user.getUsername());
   //抛出用户锁定异常
   throw new LockedAccountException();
  }
  //判断用户账号和密码是否正确
  boolean matches = super.doCredentialsMatch(token, info);
  if (matches) {
   //如果正确,从缓存中将用户登录计数 清除
   passwordRetryCache.remove(username);
  }
  return matches;
 }
 /**
  * 根据用户名 解锁用户
  * @param username
  * @return
  */
 public void unlockAccount(String username){
  User user = userMapper.findByUserName(username);
  if (user != null){
   //修改数据库的状态字段为锁定
   user.setState("0");
   userMapper.update(user);
   passwordRetryCache.remove(username);
  }
 }
}

/**
 * 配置密码比较器
 * @return
 */
@Bean("credentialsMatcher")
public RetryLimitHashedCredentialsMatcher retryLimitHashedCredentialsMatcher(){
 RetryLimitHashedCredentialsMatcher retryLimitHashedCredentialsMatcher = new RetryLimitHashedCredentialsMatcher(ehCacheManager());

 //如果密码加密,可以打开下面配置
 //加密算法的名称
 //retryLimitHashedCredentialsMatcher.setHashAlgorithmName("MD5");
 //配置加密的次数
 //retryLimitHashedCredentialsMatcher.setHashIterations(1024);
 //是否存储为16进制
 //retryLimitHashedCredentialsMatcher.setStoredCredentialsHexEncoded(true);

 return retryLimitHashedCredentialsMatcher;
}

/**
 * 身份认证realm; (这个需要自己写，账号密码校验；权限等)
 * @return
 */
@Bean
public ShiroRealm shiroRealm(){
 ShiroRealm shiroRealm = new ShiroRealm();
 ......
 //配置自定义密码比较器
 shiroRealm.setCredentialsMatcher(retryLimitHashedCredentialsMatcher());
 return shiroRealm;
}

<!-- 登录失败次数缓存
  注意 timeToLiveSeconds 设置为300秒 也就是5分钟
  可以根据自己的需求更改
 -->
<cache name="passwordRetryCache"
  maxEntriesLocalHeap="2000"
  eternal="false"
  timeToIdleSeconds="0"
  timeToLiveSeconds="300"
  overflowToDisk="false"
  statistics="true">
</cache>

/**
 * 解除admin 用户的限制登录 
 * 写死的 方便测试
 * @return
 */
@RequestMapping("/unlockAccount")
public String unlockAccount(Model model){
 model.addAttribute("msg","用户解锁成功");

 retryLimitHashedCredentialsMatcher.unlockAccount("admin");

 return "login";
}

<a href="/unlockAccount" rel="external nofollow" >解锁admin用户</a></button>