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<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;
    }
}
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 前言: 最近,有一个小伙伴成功获得好几份不错的offer,和他交谈的过程中得知面试官问他关于Spring的问题比较...
    阿秀老师阅读 3,240评论 0 0
  • 我劝你做一个“懂事”的人 我们这一生,大概会遇到800多万人。 其中4万人会跟你打过招呼,3600人会跟你有过交集...
    子沄阅读 2,701评论 0 2
  • 今天听到FM中《墨菲定律》里说“先处理好情绪再去处理事情”,有很多感慨。首先自己性格直率,往往很多事情在情...
    南方有高人阅读 1,841评论 0 1
  • 听说过格桑花吗?那种只生长在海拔5000米以上的地区,被称为顽强的象征,也是幸福象征的花。一个是藏族贵胄女儿,一...
    保护发际线濒危少女阅读 6,094评论 2 8
  • PPT排版指南 只有平面排版,并不存在PPT排版。所以,这些排版知识完全可以用在任何排版地方,不管你是H5制作还是...
    马腾Madao阅读 5,074评论 1 7

友情链接更多精彩内容