管道基础大数据平台系统开发-【后端】-Server
1
13693261870
2023-08-14 c8e3797083c44beaeb909a2c2d7a6095cef7aac3
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
@@ -38,13 +35,20 @@
    @Autowired
    LoginService loginService;
    @Autowired
    RedisService redisService;
    @Override
    public Integer selectCount(String name, Integer type) {
        name = StringHelper.getLikeUpperStr(name);
        return tokenMapper.selectCount(name, type);
    }
    @Override
    public List<TokenEntity> selectByPage(String name, Integer type, Integer limit, Integer offset) {
        name = StringHelper.getLikeUpperStr(name);
        return tokenMapper.selectByPage(name, type, limit, offset);
    }
@@ -96,14 +100,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;
    }
@@ -142,8 +147,8 @@
            return false;
        }
        // 清除Cookie WebHelper.saveCookie(StaticData.TOKEN_COOKIE_KEY, "", 60, res)
        WebHelper.deleteAll(req, res);
        // 清除Cookie
        WebHelper.deleteCookies(req, res);
        // 获取当前用户
        UserEntity ue = getCurrentUser(req);
@@ -165,7 +170,7 @@
        }
        // 写日志
        LoginEntity le = loginService.getNewLogin(ue.getId(), 3, req);
        LoginEntity le = loginService.getNewLogin(ue.getId(), 1, 3, 1, req);
        rows = loginService.insertLogin(le);
        return rows > 0;
@@ -183,7 +188,7 @@
        // redis
        Object obj = redisService.get(tokenKey);
        if (obj != null && obj instanceof TokenEntity) {
        if (obj instanceof TokenEntity) {
            return (TokenEntity) obj;
        }
@@ -214,9 +219,6 @@
    /**
     * 获取当前用户
     *
     * @param req
     * @return
     */
    public UserEntity getCurrentUser(HttpServletRequest req) {
        String token = WebHelper.getToken(req);
@@ -228,7 +230,7 @@
        // redis
        Object obj = redisService.get(userKey);
        if (obj != null && obj instanceof UserEntity) {
        if (obj instanceof UserEntity) {
            return (UserEntity) obj;
        }
@@ -243,4 +245,37 @@
        return ue;
    }
    /**
     * 设置密码错误缓存
     */
    public void setPwdErrCache(UserEntity ue) {
        String key = RedisCacheKey.signPwdError(ue.getUid());
        Object objCount = redisService.get(key);
        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);
        }
    }
    /**
     * 用户ID是/否禁用
     */
    public boolean isUidDisable(UserEntity ue) {
        String key = RedisCacheKey.signPwdError(ue.getUid());
        Object objCount = redisService.get(key);
        return objCount != null && (int) objCount >= SettingData.PWD_ERR_COUNT;
    }
}