com.mars.iserver.par.HttpMarsRequestFactory Maven / Gradle / Ivy

Go to download
package com.mars.iserver.par;

import com.alibaba.fastjson.JSONObject;
import com.mars.common.constant.MarsConstant;
import com.mars.iserver.constant.ParamTypeConstant;
import com.mars.iserver.par.formdata.ParsingFormData;
import com.mars.server.server.request.HttpMarsRequest;
import com.mars.server.server.request.model.MarsFileUpLoad;
import com.sun.net.httpserver.HttpExchange;

import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * HttpMarsRequest工厂
 *
 * @author yuye
 */
public class HttpMarsRequestFactory {



    /**
     * 从httpExchange中提取出所有的参数，并放置到HttpMarsRequest中
     * @param marsRequest mars请求
     * @return 加工后的mars请求
     * @throws Exception 异常
     */
    public static HttpMarsRequest getHttpMarsRequest(HttpMarsRequest marsRequest) throws Exception {
        Map files = new HashMap<>();
        Map> marsParams = new HashMap<>();

        HttpExchange httpExchange = marsRequest.getHttpExchange();
        if (httpExchange.getRequestMethod().toUpperCase().equals("GET")) {
            /* 从get请求中获取参数 */
            String paramStr = httpExchange.getRequestURI().getQuery();
            marsParams = urlencoded(paramStr, marsParams, false);
        } else {
            /* 从非GET请求中获取参数 */
            InputStream inputStream = httpExchange.getRequestBody();
            if (inputStream == null) {
                return marsRequest;
            }

            /* 根据提交方式，分别处理参数 */
            String contentType = marsRequest.getContentType();
            if (contentType.startsWith(ParamTypeConstant.URL_ENCODED)) {
                /* 正常的表单提交 */
                String paramStr = getParamStr(inputStream);
                marsParams = urlencoded(paramStr, marsParams, true);
            } else if (contentType.startsWith(ParamTypeConstant.FORM_DATA)) {
                /* formData提交，可以用于文件上传 */
                Map result = formData(httpExchange, marsParams, files, contentType);
                files = (Map) result.get(ParsingFormData.FILES_KEY);
                marsParams = (Map>) result.get(ParsingFormData.PARAMS_KEY);
            } else if (contentType.startsWith(ParamTypeConstant.JSON)) {
                /* RAW提交(json) */
                JSONObject jsonParams = raw(inputStream);
                marsRequest.setJsonObject(jsonParams);
            }
        }
        /* 将提取出来的参数，放置到HttpMarsRequest中 */
        marsRequest.setFiles(files);
        marsRequest.setParams(marsParams);

        return marsRequest;
    }



    /**
     * 从输入流里面读取所有的数据
     * @param inputStream 输入流
     * @return 数据
     * @throws Exception 异常
     */
    private static String getParamStr(InputStream inputStream) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(inputStream, MarsConstant.ENCODING));
        String line = null;
        StringBuffer paramsStr = new StringBuffer();
        while ((line = br.readLine()) != null) {
            paramsStr.append(line);
        }
        return paramsStr.toString();
    }

    /**
     * 表单提交处理
     * @param paramStr 数据
     * @param marsParams httpMarsRequest的参数对象
     * @param hasDecode 是否需要解码 true是，false不是
     * @return httpMarsRequest的参数对象
     * @throws Exception 异常
     */
    private static Map> urlencoded(String paramStr,Map> marsParams, boolean hasDecode) throws Exception {
        if(paramStr != null){
            String[] paramsArray = paramStr.split("&");
            if(paramsArray == null || paramsArray.length < 1){
                return marsParams;
            }
            List values = null;
            for(String paramItem : paramsArray){
                String[] param = paramItem.split("=");
                if(param == null || param.length < 2){
                    continue;
                }
                String key = param[0];

                values = marsParams.get(key);
                if(values == null){
                    values = new ArrayList<>();
                }

                String value = param[1];
                if(hasDecode){
                    value = URLDecoder.decode(value, MarsConstant.ENCODING);
                }
                values.add(value);
                marsParams.put(key,values);
            }
        }
        return marsParams;
    }

    /**
     * RAW提交处理
     * @param inputStream 输入流
     * @return httpMarsRequest的参数对象
     * @throws Exception 异常
     */
    private static JSONObject raw(InputStream inputStream) throws Exception {
        String paramStr = getParamStr(inputStream);
        if(paramStr == null || paramStr.trim().equals("")){
            return null;
        }

        JSONObject jsonObject = JSONObject.parseObject(paramStr);
        return jsonObject;
    }

    /**
     * formData提交处理
     * @param exchange 请求对象
     * @param marsParams httpMarsRequest的参数对象
     * @param files httpMarsRequest的文件参数对象
     * @param contentType 内容类型
     * @return httpMarsRequest的参数对象 和 httpMarsRequest的文件参数对象
     * @throws Exception 异常
     */
    private static Map formData(HttpExchange exchange, Map> marsParams, Map files, String contentType) throws Exception {
        return ParsingFormData.parsing(exchange,marsParams,files,contentType);
    }
}