|
|
@@ -7,6 +7,7 @@ import cn.dev33.satoken.stp.StpLogic;
|
|
|
import cn.dev33.satoken.stp.StpUtil;
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
import cn.tr.core.utils.ServletUtils;
|
|
|
+import cn.tr.plugin.security.bo.UserLoginInfoBO;
|
|
|
import cn.tr.plugin.security.constant.SecurityConstant;
|
|
|
import cn.tr.plugin.security.context.LoginUserContextHolder;
|
|
|
import org.springframework.web.util.NestedServletException;
|
|
|
@@ -24,8 +25,8 @@ public class TrSaTokenFilter implements Filter {
|
|
|
@Override
|
|
|
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
|
|
|
if(StrUtil.isEmpty(LoginUserContextHolder.getStpType())){
|
|
|
- String stpType = ServletUtils.getHeader(SecurityConstant.STP_TYPE);
|
|
|
- LoginUserContextHolder.setStpType( StrUtil.isEmpty(stpType)? StpUtil.TYPE:stpType);
|
|
|
+ UserLoginInfoBO user = LoginUserContextHolder.getUser();
|
|
|
+ LoginUserContextHolder.setStpType( StrUtil.isEmpty(user.getLoginType())?StpUtil.TYPE:user.getLoginType());
|
|
|
}
|
|
|
try {
|
|
|
filterChain.doFilter(servletRequest,servletResponse);
|