package com.qianwen.smartman.common.filter;
|
|
import java.io.IOException;
|
import javax.servlet.FilterChain;
|
import javax.servlet.ServletException;
|
import javax.servlet.ServletRequest;
|
import javax.servlet.ServletResponse;
|
import javax.servlet.http.HttpServletRequest;
|
import com.qianwen.smartman.common.request.HeaderHttpServletRequestWrapper;
|
import org.springframework.boot.web.servlet.filter.OrderedFilter;
|
import org.springframework.util.StringUtils;
|
|
public class WebSocketFilter implements OrderedFilter {
|
public void doFilter(ServletRequest request, ServletResponse response, FilterChain filterChain) throws IOException, ServletException {
|
String bladeToken = request.getParameter("access_token");
|
if (!StringUtils.isEmpty(bladeToken)) {
|
HeaderHttpServletRequestWrapper servletRequest = new HeaderHttpServletRequestWrapper((HttpServletRequest) request);
|
servletRequest.addHeader("Blade-Auth", String.format("%s %s", "bearer", bladeToken));
|
filterChain.doFilter(servletRequest, response);
|
return;
|
}
|
filterChain.doFilter(request, response);
|
}
|
|
public int getOrder() {
|
return -1000;
|
}
|
}
|
