请求处理
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class Request {
// 协议信息
private String requestInfo;
// 请求方式
private String method;
// 请求URL
private String url;
// 请求参数
private String queryStr;
// 存储参数
private Map<String, List<String>> parameterMap;
private final String CRLF = "\r\n";
public Request(InputStream is) {
parameterMap = new HashMap<String, List<String>>();
byte[] datas = new byte[1024 * 1024 * 10];
int len;
try {
len = is.read(datas);
if (len <= 0) {
return;
}
requestInfo = new String(datas, 0, len);
} catch (IOException e) {
e.printStackTrace();
return;
}
// 分解字符串
parseRequestInfo();
}
public Request(Socket client) throws IOException {
this(client.getInputStream());
}
private void parseRequestInfo() {
// 分解
// 1.获取请求方式:开头到第一个/
requestInfo = decode(requestInfo, "UTF-8");
method = requestInfo.substring(0, requestInfo.indexOf('/')).trim().toLowerCase();
// 2.----获取请求url 第一个 /到第二个 HTTP/
// 可能包含?前面为url
// 1.获取/的位置
int startIndex = requestInfo.indexOf('/') + 1;
// 2.获取HTTP/的位置
int endIndex = requestInfo.indexOf("HTTP/");
// 3.分割字符串
url = requestInfo.substring(startIndex, endIndex).trim();
// 4.获取?的位置
int queryIndex = url.indexOf('?');
if (queryStr == null) {
queryStr = "";
}
if (queryIndex >= 0) {
String[] urlArry = url.split("\\?");
if (urlArry.length >= 1) {
url = urlArry[0].trim();
}
if (urlArry.length >= 2) {
queryStr = urlArry[1].trim();
}
}
// 3.----获取请求参数: 如果GET已经获取,如果POST可能在请求体中
if (method.equals("post")) {
String qStr = requestInfo.substring(requestInfo.lastIndexOf(CRLF)).trim();
if (qStr != null && !(qStr.equals(""))) {
if ("" == queryStr || null == queryStr) {
queryStr += qStr;
} else {
queryStr += "&" + qStr;
}
}
}
System.out.println(method + "-->" + url + "-->" + queryStr);
// 转成Map ddss=1&fav=2&uname=shsxt&age=18&others=
convertMap();
}
// 处理请求参数为Map
private void convertMap() {
// 1、分割字符串 &
String[] keyValues = this.queryStr.split("&");
for (String queryStr : keyValues) {
// 2、再次分割字符串 =
String[] kv = queryStr.split("=");
kv = Arrays.copyOf(kv, 2);
// 获取key和value
String key = kv[0];
String value = kv[1];
// 存储到map中
if (!parameterMap.containsKey(key)) {// 第一次
parameterMap.put(key, new ArrayList<String>());
}
parameterMap.get(key).add(value);
}
}
/**
* 处理中文
*
* @return
*/
private String decode(String value, String enc) {
try {
return java.net.URLDecoder.decode(value, enc);
} catch (UnsupportedEncodingException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return null;
}
/**
* 通过name获取对应的多个值
*
* @param key
* @return
*/
public String[] getParameterValues(String key) {
List<String> values = this.parameterMap.get(key);
if (null == values || values.size() < 1) {
return null;
}
return values.toArray(new String[0]);
}
/**
* 通过name获取对应的一个值
*
* @param key
* @return
*/
public String getParameter(String key) {
String[] values = getParameterValues(key);
return values == null ? null : values[0];
}
public String getMethod() {
return method;
}
public String getUrl() {
return url;
}
public String getQueryStr() {
return queryStr;
}
}
网友评论