2019-08-25

请求处理

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> parameterMap;
    private final String CRLF = "\r\n";

    public Request(InputStream is) {
        parameterMap = new HashMap>();
        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());
            }
            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 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;
    }
}