package com.lf.server.service.data;
|
|
import com.lf.server.entity.data.TokenEntity;
|
import com.lf.server.entity.data.UsersEntity;
|
import com.lf.server.entity.sys.RedisCacheKey;
|
import com.lf.server.helper.StringHelper;
|
import com.lf.server.helper.WebHelper;
|
import com.lf.server.mapper.data.TokenMapper;
|
import com.lf.server.service.sys.RedisService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.util.List;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* 令牌表
|
* @author sws
|
* @date 2022-09-28
|
*/
|
|
@Service
|
public class TokenService implements TokenMapper {
|
@Autowired
|
private RedisService redisService;
|
|
@Autowired
|
TokenMapper tokenMapper;
|
|
@Autowired
|
UsersService usersService;
|
|
@Override
|
public Integer selectCount(String token) {
|
return tokenMapper.selectCount(token);
|
}
|
|
@Override
|
public List<TokenEntity> selectByPage(String token, Integer limit, Integer offset) {
|
return tokenMapper.selectByPage(token, limit, offset);
|
}
|
|
@Override
|
public Integer insertToken(TokenEntity tokenEntity) {
|
return tokenMapper.insertToken(tokenEntity);
|
}
|
|
@Override
|
public Integer insertTokens(List<TokenEntity> tokenEntity) {
|
return tokenMapper.insertTokens(tokenEntity);
|
}
|
|
@Override
|
public Integer deleteToken(int id) {
|
return tokenMapper.deleteToken(id);
|
}
|
|
@Override
|
public Integer deleteTokens(List<Integer> ids) {
|
return tokenMapper.deleteTokens(ids);
|
}
|
|
@Override
|
public Integer updateToken(TokenEntity tokenEntity) {
|
return tokenMapper.updateToken(tokenEntity);
|
}
|
|
@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();
|
}
|
|
/**
|
* 获取新的令牌实体类
|
*
|
* @param userid
|
* @param req
|
* @return
|
*/
|
public TokenEntity getNewToken(int userid, HttpServletRequest req) {
|
int duration = 240;
|
|
TokenEntity te = new TokenEntity();
|
te.setToken(WebHelper.getGuid());
|
te.setDuration(duration);
|
te.setExpire(WebHelper.getTimestamp(duration));
|
te.setType(0);
|
te.setIp(WebHelper.getIpAddress(req));
|
te.setCreateUser(userid);
|
|
|
return te;
|
}
|
|
/**
|
* 是/否登录
|
*
|
* @param req
|
* @param res
|
* @return
|
*/
|
public Boolean isLogin(HttpServletRequest req, HttpServletResponse res) {
|
String token = WebHelper.getToken(req);
|
if (StringHelper.isNull(token)) {
|
return false;
|
}
|
|
// redis
|
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
|
* @return
|
*/
|
public TokenEntity getEntityByToken(String token) {
|
if (StringHelper.isNull(token)) {
|
return null;
|
}
|
|
String tokenKey = RedisCacheKey.signTokenKey(token);
|
|
// redis
|
Object obj = redisService.get(tokenKey);
|
if (obj != null && obj instanceof TokenEntity) {
|
return (TokenEntity) obj;
|
}
|
|
// db
|
TokenEntity te = selectOneByToken(token);
|
if (te != null) {
|
redisService.put(tokenKey, te, te.getDuration(), TimeUnit.MINUTES);
|
}
|
|
return te;
|
}
|
|
/**
|
* 保存token
|
*
|
* @param te
|
* @param req
|
* @param res
|
*/
|
public void saveToken(UsersEntity ue, TokenEntity te, HttpServletRequest req, HttpServletResponse res) {
|
// 保存至Cookie
|
WebHelper.saveToken2Cookie(te.getToken(), req, res);
|
|
// 令牌保存至Redis
|
String tokenKey = RedisCacheKey.signTokenKey(te.getToken());
|
redisService.put(tokenKey, te, te.getDuration(), TimeUnit.MINUTES);
|
|
String userKey = RedisCacheKey.signUserKey(te.getToken());
|
redisService.put(userKey, ue, te.getDuration(), TimeUnit.MINUTES);
|
}
|
|
/**
|
* 获取当前用户
|
*
|
* @param req
|
* @return
|
*/
|
public UsersEntity getCurrentUser(HttpServletRequest req) {
|
String token = WebHelper.getToken(req);
|
if (StringHelper.isEmpty(token)) {
|
return null;
|
}
|
|
String userKey = RedisCacheKey.signUserKey(token);
|
|
// redis
|
Object obj = redisService.get(userKey);
|
if (obj != null && obj instanceof UsersEntity) {
|
return (UsersEntity) obj;
|
}
|
|
// 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;
|
}
|
}
|
