1

13693261870

2022-11-17 15375a1d0d07c61c132906ff8da45f67bcd711f6

 src/main/java/com/lf/server/service/sys/TokenService.java

@@ -27,9 +27,6 @@
@Service
public class TokenService implements TokenMapper {
    @Autowired
    public RedisService redisService;

    @Autowired
    TokenMapper tokenMapper;

    @Autowired
@@ -37,6 +34,9 @@

    @Autowired
    LoginService loginService;

    @Autowired
    RedisService redisService;

    @Override
    public Integer selectCount(String name, Integer type) {
@@ -96,14 +96,15 @@
    /**
     * 获取新的令牌实体类
     */
    public TokenEntity getNewToken(int userid, HttpServletRequest req) {
    public TokenEntity getNewToken(UserEntity ue, HttpServletRequest req) {
        TokenEntity te = new TokenEntity();
        te.setToken(WebHelper.getGuid());
        te.setDuration(SettingData.TOKEN_EXPIRE);
        te.setExpire(WebHelper.getTimestamp(SettingData.TOKEN_EXPIRE));
        te.setType(0);
        te.setIp(WebHelper.getIpAddress(req));
        te.setCreateUser(userid);
        te.setCreateUser(ue.getId());
        te.setUname(ue.getUname());

        return te;
    }
@@ -145,6 +146,12 @@
        // 清除Cookie
        WebHelper.deleteCookies(req, res);

        // 获取当前用户
        UserEntity ue = getCurrentUser(req);
        if (ue == null) {
            return false;
        }

        // 清除缓存
        String tokenKey = RedisCacheKey.signTokenKey(token);
        if (redisService.hasKey(tokenKey)) {
@@ -153,12 +160,6 @@
        String userKey = RedisCacheKey.signUserKey(te.getToken());
        if (redisService.hasKey(userKey)) {
            redisService.delete(userKey);
        }

        // 获取当前用户
        UserEntity ue = getCurrentUser(req);
        if (ue == null) {
            return false;
        }

        // db，设置令牌过期
@@ -187,7 +188,7 @@

        // redis
        Object obj = redisService.get(tokenKey);
        if (obj != null && obj instanceof TokenEntity) {
        if (obj instanceof TokenEntity) {
            return (TokenEntity) obj;
        }

@@ -229,7 +230,7 @@

        // redis
        Object obj = redisService.get(userKey);
        if (obj != null && obj instanceof UserEntity) {
        if (obj instanceof UserEntity) {
            return (UserEntity) obj;
        }

@@ -254,6 +255,18 @@

        int count = objCount == null ? 1 : (int) objCount + 1;
        redisService.put(key, count, SettingData.PWD_ERR_TIME, TimeUnit.MINUTES);

        // 记录日志
        HttpServletRequest req = WebHelper.getRequest();
        LoginEntity le = loginService.getNewLogin(ue.getId(), 1, 1, 0, req);
        le.setDescr("密码不正确");
        loginService.insertLogin(le);

        if (count >= SettingData.PWD_ERR_COUNT) {
            String token = WebHelper.getToken(req);
            HttpServletResponse res = WebHelper.getResponse();
            logout(token, req, res);
        }
    }

    /**