1

13693261870

2022-09-30 fe25c74329239f48314e8c3f7f0d756f66f0768c

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

@@ -1,5 +1,7 @@
package com.lf.server.service.data;

import com.lf.server.entity.all.StaticData;
import com.lf.server.entity.data.LoginEntity;
import com.lf.server.entity.data.TokenEntity;
import com.lf.server.entity.data.UsersEntity;
import com.lf.server.entity.sys.RedisCacheKey;
@@ -29,6 +31,12 @@
    @Autowired
    TokenMapper tokenMapper;

    @Autowired
    UsersService usersService;

    @Autowired
    LoginService loginService;

    @Override
    public Integer selectCount(String token) {
        return tokenMapper.selectCount(token);
@@ -37,6 +45,21 @@
    @Override
    public List<TokenEntity> selectByPage(String token, Integer limit, Integer offset) {
        return tokenMapper.selectByPage(token, limit, offset);
    }

    @Override
    public TokenEntity selectToken(int id) {
        return tokenMapper.selectToken(id);
    }

    @Override
    public TokenEntity selectOneByToken(String token) {
        return tokenMapper.selectOneByToken(token);
    }

    @Override
    public List<TokenEntity> selectTokenAll() {
        return tokenMapper.selectTokenAll();
    }

    @Override
@@ -65,13 +88,8 @@
    }

    @Override
    public TokenEntity selectToken(int id) {
        return tokenMapper.selectToken(id);
    }

    @Override
    public List<TokenEntity> selectTokenAll() {
        return tokenMapper.selectTokenAll();
    public Integer updateTokenExpire(TokenEntity tokenEntity) {
        return tokenMapper.updateTokenExpire(tokenEntity);
    }

    /**
@@ -97,7 +115,7 @@
    }

    /**
     * 是否登录
     * 是/否登录
     *
     * @param req
     * @param res
@@ -110,11 +128,61 @@
        }

        // redis
        if (redisService.hasKey(RedisCacheKey.signTokenKey(token))) {
        String tokenKey = RedisCacheKey.signTokenKey(token);
        if (redisService.hasKey(tokenKey)) {
            return true;
        }

        // db
        TokenEntity te = selectOneByToken(token);
        if (te != null) {
            redisService.put(tokenKey, te, te.getDuration(), TimeUnit.MINUTES);
            return true;
        }

        return false;
    }

    /**
     * 登出
     *
     * @param token
     * @param req
     * @return
     */
    public Boolean logout(String token, HttpServletRequest req) {
        TokenEntity te = getEntityByToken(token);
        if (te == null) {
            return false;
        }

        // 清除Cookie
        WebHelper.deleteCookie(StaticData.TOKEN_COOKIE_KEY, req);

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

        // 清除缓存
        String tokenKey = RedisCacheKey.signTokenKey(token);
        redisService.delete(tokenKey);
        String userKey = RedisCacheKey.signUserKey(te.getToken());
        redisService.delete(userKey);

        // db，设置令牌过期
        te.setUpdateUser(ue.getId());
        Integer rows = updateTokenExpire(te);
        if (rows == 0) {
            return false;
        }

        // 写日志
        LoginEntity le = loginService.getNewLogin(ue.getId(), 3, req);
        rows = loginService.insertLogin(le);

        return rows > 0;
    }

    /**
@@ -128,13 +196,21 @@
            return null;
        }

        String key = RedisCacheKey.signTokenKey(token);
        Object obj = redisService.get(key);
        String tokenKey = RedisCacheKey.signTokenKey(token);

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

        return null;
        // db
        TokenEntity te = selectOneByToken(token);
        if (te != null) {
            redisService.put(tokenKey, te, te.getDuration(), TimeUnit.MINUTES);
        }

        return te;
    }

    /**
@@ -158,6 +234,7 @@

    /**
     * 获取当前用户
     *
     * @param req
     * @return
     */
@@ -168,11 +245,22 @@
        }

        String userKey = RedisCacheKey.signUserKey(token);

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

        return null;
        // db
        UsersEntity ue = usersService.selectByToken(token);
        if (ue != null) {
            TokenEntity te = getEntityByToken(token);
            if (te != null) {
                redisService.put(userKey, ue, te.getDuration(), TimeUnit.MINUTES);
            }
        }

        return ue;
    }
}