package foundation.handler;
|
|
import java.io.BufferedReader;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.io.InputStreamReader;
|
import java.util.Enumeration;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
|
import foundation.json.JObjectReader;
|
import foundation.route.Operation;
|
import foundation.token.IOnlineUser;
|
import foundation.util.MapList;
|
import foundation.util.Util;
|
|
public class DataPool implements IDataPool {
|
|
protected static Logger logger;
|
public static String ProviderName = "dataPool";
|
private IOnlineUser user;
|
private HttpServletRequest request;
|
private MapList<String, String> params;
|
private DataPoolType type;
|
private Operation operation;
|
private JObjectReader jObjectReader;
|
private String token;
|
private String body;
|
|
static {
|
logger = LogManager.getLogger();
|
}
|
|
public DataPool(HttpServletRequest request) throws Exception {
|
params = new MapList<String, String>();
|
this.request = request;
|
|
loadParams();
|
parseType(request);
|
}
|
|
private void parseType(HttpServletRequest request) {
|
String method = request.getMethod();
|
method = method.toLowerCase();
|
|
String contentCount = request.getContentType();
|
|
if (contentCount == null) {
|
contentCount = "";
|
}
|
contentCount = contentCount.toLowerCase();
|
|
if (contentCount.indexOf("multipart") >= 0) {
|
type = DataPoolType.MultiPart;
|
}
|
else if ("get".equals(method)) {
|
type = DataPoolType.Get;
|
}
|
else if ("options".equals(method)) {
|
type = DataPoolType.Options;
|
}
|
else {
|
type = DataPoolType.Post;
|
}
|
}
|
|
public String getToken() throws IOException {
|
//1. read from URL parameters
|
token = params.get("token");
|
|
if (!Util.isEmpty(token)) {
|
return token;
|
}
|
|
//2. read from body
|
getJObjectReader();
|
|
if (jObjectReader == null) {
|
return null;
|
}
|
|
token = jObjectReader.getString("token");
|
return token;
|
}
|
|
private void loadParams() throws Exception {
|
Enumeration<String> paramNames = request.getParameterNames();
|
|
while (paramNames.hasMoreElements()) {
|
String paramName = (String) paramNames.nextElement();
|
String value = request.getParameter(paramName);
|
loadOneParam(paramName, value);
|
}
|
}
|
|
public void loadOneParam(String name, String value) {
|
if (Util.isEmpty(name)) {
|
return;
|
}
|
|
params.add(name, value);
|
}
|
|
private void loadBody() throws IOException {
|
if (body != null) {
|
return;
|
}
|
|
InputStream inputStream = request.getInputStream();
|
|
//1. read stream
|
StringBuilder result = new StringBuilder();
|
String line;
|
|
BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, "utf-8"));
|
|
while ((line = reader.readLine()) != null) {
|
result.append(line);
|
}
|
|
body = result.toString();
|
logger.debug("request: {}", body);
|
|
jObjectReader = new JObjectReader(body);
|
}
|
|
public String getParam(String name) {
|
return params.get(name);
|
}
|
|
public HttpServletRequest getRequest() {
|
return request;
|
}
|
|
public String getRemoteIP() {
|
return request.getRemoteAddr();
|
}
|
|
public Operation getOperation() {
|
return operation;
|
}
|
|
public void setOperation(Operation operation) {
|
this.operation = operation;
|
}
|
|
public String getBody() throws IOException {
|
if (body == null) {
|
loadBody();
|
}
|
|
return body;
|
}
|
|
@Override
|
public synchronized JObjectReader getJObjectReader() throws IOException {
|
if (jObjectReader == null && (DataPoolType.MultiPart != type)) {
|
loadBody();
|
}
|
|
return jObjectReader;
|
}
|
|
public MapList<String, String> getParams() {
|
return params;
|
}
|
|
public DataPoolType getType() {
|
return type;
|
}
|
|
public IOnlineUser getUser() {
|
return user;
|
}
|
|
public void setUser(IOnlineUser user) {
|
this.user = user;
|
}
|
}
|
