packagecom.mobile.web.common;importorg.springframework.web.filter.OncePerRequestFilter;importjavax.servlet.FilterChain;importjavax.servlet.ServletException;importjavax.servlet.http.Cookie;importjavax.servlet.http.HttpServletRequest;importjavax.servlet.http.HttpServletRequestWrapper;importjavax.servlet.http.HttpServletResponse;importjava.io.IOException;import java.util.*;/*** 自定义的过滤器,
* Created by Administrator on 2017/6/19 0019.*/
public class ModifyParametersFilter extendsOncePerRequestFilter {
@Overrideprotected voiddoFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)throwsServletException, IOException {//修改cookie
ModifyHttpServletRequestWrapper mParametersWrapper = newModifyHttpServletRequestWrapper(request);
String token= request.getHeader("token");if (token != null && !"".equals(token)) {
mParametersWrapper.putCookie("JSESSIONID", token);
}//finish
filterChain.doFilter(mParametersWrapper, response);
}/*** 修改cookie信息*/
private class ModifyHttpServletRequestWrapper extendsHttpServletRequestWrapper {private MapmapCookies;
ModifyHttpServletRequestWrapper(HttpServletRequest request) {super(request);this.mapCookies = new HashMap<>();
}voidputCookie(String name, String value) {this.mapCookies.put(name, value);
}publicCookie[] getCookies() {
HttpServletRequest request=(HttpServletRequest) getRequest();
Cookie[] cookies=request.getCookies();if (mapCookies == null ||mapCookies.isEmpty()) {returncookies;
}if (cookies == null || cookies.length == 0) {
List cookieList = new LinkedList<>();for (Map.Entryentry : mapCookies.entrySet()) {
String key=entry.getKey();if (key != null && !"".equals(key)) {
cookieList.add(newCookie(key, entry.getValue()));
}
}if(cookieList.isEmpty()) {returncookies;
}return cookieList.toArray(newCookie[cookieList.size()]);
}else{
List cookieList = new ArrayList<>(Arrays.asList(cookies));for (Map.Entryentry : mapCookies.entrySet()) {
String key=entry.getKey();if (key != null && !"".equals(key)) {for (int i = 0; i < cookieList.size(); i++) {if(cookieList.get(i).getName().equals(key)){
cookieList.remove(i);
}
}
cookieList.add(newCookie(key, entry.getValue()));
}
}return cookieList.toArray(newCookie[cookieList.size()]);
}
}
}
}