最近发现不能修改request的参数内容,假如想要在request的paramMap里面默认注入修改内容,采用下面这种方法可以实现向request中修改参数,具体分析如下:
1)继承javax.servlet.http.HttpServletRequestWrapper
2)重写方法
2.1) getParameter(String name); // 取值方法
2.2) getParameterValues(String name); // 取值方法
2.3) getParameterNames(); // 获取请求参数方法(需要时修改)
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Collections;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import org.apache.commons.io.IOUtils;
public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {
private ByteArrayOutputStream cacheBytes;
// 用于存储请求参数
private Map<String , String[]> params = new HashMap<String, String[]>();
public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
cacheBytes = new ByteArrayOutputStream();
IOUtils.copy(super.getInputStream(), cacheBytes);
}
@Override
public ServletInputStream getInputStream() throws IOException {
return new CacheServletInputStream();
}
class CacheServletInputStream extends ServletInputStream{
private ByteArrayInputStream bais;
public CacheServletInputStream(){
bais = new ByteArrayInputStream(cacheBytes.toByteArray());
}
@Override
public int read() throws IOException {
return bais.read();
}
@Override
public void close() throws IOException {
super.close();
bais.close();
cacheBytes.close();
}
}
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(getInputStream(),"UTF-8"));
}
/**
* 添加参数到map中
* @param name
* @param value
*/
public void setParameter(String name, Object value) {
if (value != null) {
System.out.println(value);
if (value instanceof String[]) {
params.put(name, (String[]) value);
} else if (value instanceof String) {
params.put(name, new String[]{(String) value});
} else {
params.put(name, new String[]{String.valueOf(value)});
}
}
}
/**
* 重写getParameter,代表参数从当前类中的map获取
* @param name
* @return
*/
@Override
public String getParameter(String name) {
String[]values = params.get(name);
if(values == null || values.length == 0) {
return null;
}
return values[0];
}
/**
* 重写getParameterValues方法,从当前类的 map中取值
* @param name
* @return
*/
@Override
public String[] getParameterValues(String name) {
return params.get(name);
}
/**
* 添加参数到map中
* @param extraParams
*/
public void extraParams(Map<String, Object> extraParams) {
if (extraParams == null) return;
for (Map.Entry<String, Object> entry : extraParams.entrySet()) {
setParameter(entry.getKey(), entry.getValue());
}
}
/**
* 获取请求参数
* @return
*/
public Enumeration<String> getParameterNames() {
return Collections.enumeration(this.params.keySet());
}
}
3) 写过滤器
import net.xiaoyuanxia.framework.core.wrapper.BodyReaderHttpServletRequestWrapper;
import net.xiaoyuanxia.framework.core.wrapper.BodyReaderHttpServletResponseWrapper;
import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
public class AppVeveryFilter implements Filter {
@Override
public void destroy() {
}
@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
HttpServletRequest httpRequest = (HttpServletRequest) request;
HttpServletResponse httpResponse = (HttpServletResponse) response;
BodyReaderHttpServletResponseWrapper mResp = new BodyReaderHttpServletResponseWrapper(httpResponse);
Map map = new HashMap();
map.put("a", "a");
BodyReaderHttpServletRequestWrapper bodyReaderHttpServletRequestWrapper = new BodyReaderHttpServletRequestWrapper(httpRequest);
bodyReaderHttpServletRequestWrapper.extraParams(map);
chain.doFilter(bodyReaderHttpServletRequestWrapper, mResp);
}
@Override
public void init(FilterConfig config){
}
}
4)配置web.xml
<filter>
<filter-name>appVeveryFilter</filter-name>
<filter-class>xx.xx.xx.core.filter.AppVeveryFilter</filter-class>
</filter>
<filter-mapping>
<filter-name>appVeveryFilter</filter-name>
<url-pattern>/app/*</url-pattern>
</filter-mapping>
5) 测试
@Controller
@RequestMapping("/app")
public class TestController extends BaseAction {
@RequestMapping("/test")
public void test(String a) {
System.out.println("a:" + a);
}
}