package com.ruoyi.framework.security.filter;
|
|
import java.io.IOException;
|
import javax.servlet.FilterChain;
|
import javax.servlet.ServletException;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
|
import com.ruoyi.common.constant.Constants;
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
import com.ruoyi.common.core.domain.model.LoginUserApplet;
|
import com.ruoyi.system.service.ScreenAccountService;
|
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 com.ruoyi.common.core.domain.model.LoginUser;
|
import com.ruoyi.common.utils.SecurityUtils;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.framework.web.service.TokenService;
|
|
/**
|
* token过滤器 验证token有效性
|
*
|
* @author ruoyi
|
*/
|
@Component
|
public class JwtAuthenticationTokenFilter extends OncePerRequestFilter
|
{
|
@Autowired
|
private TokenService tokenService;
|
|
@Autowired
|
private ScreenAccountService screenAccountService;
|
|
@Override
|
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
|
throws ServletException, IOException
|
{
|
LoginUser loginUser = tokenService.getLoginUser(request);
|
String key = request.getHeader("Authorization");
|
Long id = screenAccountService.getToken(key);
|
if (StringUtils.isNotNull(loginUser) && StringUtils.isNull(SecurityUtils.getAuthentication())
|
|| id!= null)
|
|
{
|
if (StringUtils.isNotNull(loginUser)){
|
tokenService.verifyToken(loginUser);
|
UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(loginUser, null, loginUser.getAuthorities());
|
authenticationToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
|
SecurityContextHolder.getContext().setAuthentication(authenticationToken);
|
}
|
if(id != null){
|
// accessToken验证通过，创建认证对象
|
LoginUser loginUserByKey = new LoginUser();
|
loginUserByKey.setUserId(id);
|
SysUser sysUser = new SysUser();
|
sysUser.setUserId(1L);
|
sysUser.setUserName("admin");
|
loginUserByKey.setUser(sysUser);
|
|
// 创建临时认证对象，不包含具体权限信息
|
UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(loginUserByKey, null, null);
|
authenticationToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
|
SecurityContextHolder.getContext().setAuthentication(authenticationToken);
|
}
|
// accessToken验证失败时不抛出异常，允许请求继续
|
}
|
chain.doFilter(request, response);
|
}
|
|
|
|
|
}
|
