美文网首页
2019-08-25

2019-08-25

作者: 勿陌 | 来源:发表于2019-08-25 21:01 被阅读0次

    请求处理

    
    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;
        }
    }

    相关文章

      网友评论

          本文标题:2019-08-25

          本文链接:https://www.haomeiwen.com/subject/ydbkectx.html