package com.ruoyi.extend.datasecurity.aesencrypt.wrapper;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletRequestWrapper;
|
import java.util.Enumeration;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.Vector;
|
|
/**
|
* 重写 HttpServletRequestWrapper 处理 Get
|
*
|
* @author Tellsea
|
* @date 2022/9/3
|
*/
|
public class GetParamsRequestWrapper extends HttpServletRequestWrapper {
|
|
private Map<String, String[]> params = new HashMap<>();
|
|
public GetParamsRequestWrapper(HttpServletRequest request) {
|
super(request);
|
//将参数表,赋予给当前的Map以便于持有request中的参数
|
this.params.putAll(request.getParameterMap());
|
}
|
|
public GetParamsRequestWrapper(HttpServletRequest request, Map<String, Object> extendParams) {
|
this(request);
|
//这里将扩展参数写入参数表
|
addAllParameters(extendParams);
|
}
|
|
/**
|
* 在获取所有的参数名,必须重写此方法,否则对象中参数值映射不上
|
*
|
* @return
|
*/
|
@Override
|
public Enumeration<String> getParameterNames() {
|
return new Vector(params.keySet()).elements();
|
}
|
|
/**
|
* 重写getParameter方法
|
*
|
* @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];
|
}
|
|
@Override
|
public String[] getParameterValues(String name) {
|
String[] values = params.get(name);
|
if (values == null || values.length == 0) {
|
return null;
|
}
|
return values;
|
}
|
|
/**
|
* 增加多个参数
|
*
|
* @param otherParams 增加的多个参数
|
*/
|
public void addAllParameters(Map<String, Object> otherParams) {
|
for (Map.Entry<String, Object> entry : otherParams.entrySet()) {
|
addParameter(entry.getKey(), entry.getValue());
|
}
|
}
|
|
/**
|
* 增加参数
|
*
|
* @param name 参数名
|
* @param value 参数值
|
*/
|
public void addParameter(String name, Object value) {
|
if (value != null) {
|
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)});
|
}
|
}
|
}
|
}
|