package com.ruoyi.framework.security.filter;
|
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
import com.ruoyi.common.core.domain.model.LoginUser;
|
import com.ruoyi.common.core.domain.model.LoginUserApplet;
|
import com.ruoyi.common.core.redis.RedisCache;
|
import com.ruoyi.common.utils.SecurityUtils;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.framework.web.service.TokenService;
|
import com.ruoyi.system.service.ISysUserService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
|
import org.springframework.security.core.context.SecurityContextHolder;
|
import org.springframework.security.web.authentication.WebAuthenticationDetailsSource;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.filter.OncePerRequestFilter;
|
|
import javax.servlet.FilterChain;
|
import javax.servlet.ServletException;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
|
/**
|
* token过滤器 验证token有效性
|
*
|
* @author ruoyi
|
*/
|
@Component
|
public class JwtAuthenticationTokenFilter extends OncePerRequestFilter
|
{
|
@Autowired
|
private TokenService tokenService;
|
@Autowired
|
private RedisCache redisCache;
|
@Autowired
|
private ISysUserService sysUserService;
|
|
@Override
|
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
|
throws ServletException, IOException
|
{
|
LoginUser loginUser = tokenService.getLoginUser(request);
|
LoginUserApplet applet = tokenService.getLoginUserApplet(request);
|
if (StringUtils.isNotNull(loginUser) && StringUtils.isNull(SecurityUtils.getAuthentication())||
|
StringUtils.isNotNull(applet))
|
{
|
if (StringUtils.isNotNull(loginUser)){
|
Long userId = loginUser.getUserId();
|
SysUser sysUser = sysUserService.selectUserById(userId);
|
if (StringUtils.isNull(sysUser)){
|
redisCache.deleteObject(tokenService.getTokenKey(loginUser.getToken()));
|
return;
|
}
|
if("1".equals(sysUser.getStatus())){
|
redisCache.deleteObject(tokenService.getTokenKey(loginUser.getToken()));
|
return;
|
}
|
tokenService.verifyToken(loginUser);
|
UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(loginUser, null, loginUser.getAuthorities());
|
authenticationToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
|
SecurityContextHolder.getContext().setAuthentication(authenticationToken);
|
}
|
if (StringUtils.isNotNull(applet)){
|
tokenService.verifyTokenApplet(applet);
|
UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(applet, null, applet.getAuthorities());
|
authenticationToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
|
SecurityContextHolder.getContext().setAuthentication(authenticationToken);
|
}
|
}
|
chain.doFilter(request, response);
|
}
|
}
|
