top.jfunc.http.component.DefaultSimpleAssemblingFactory Maven / Gradle / Ivy

Go to download
package top.jfunc.http.component;

import top.jfunc.http.base.FormFile;
import top.jfunc.http.request.*;
import top.jfunc.common.utils.ArrayUtil;
import top.jfunc.common.utils.MapUtil;
import top.jfunc.common.utils.MultiValueMap;

/**
 * @author xiongshiyan at 2020/1/15 , contact me with email [email protected] or phone 15208384257
 */
public class DefaultSimpleAssemblingFactory implements AssemblingFactory {
    @Override
    public HttpRequest create(String url,
                              MultiValueMap queryParams,
                              MultiValueMap headers,
                              int connectTimeout,
                              int readTimeout,
                              String resultCharset) {
        HttpRequest httpRequest = DefaultRequest.of(url);
        set(httpRequest, queryParams, headers, connectTimeout, readTimeout, resultCharset);
        return httpRequest;
    }

    @Override
    public StringBodyRequest create(String url,
                                    MultiValueMap queryParams,
                                    String body,
                                    String contentType,
                                    MultiValueMap headers,
                                    int connectTimeout,
                                    int readTimeout,
                                    String bodyCharset,
                                    String resultCharset) {
        MutableStringBodyRequest stringBodyRequest = DefaultBodyRequest.of(url);
        stringBodyRequest.setBody(body , contentType);
        if(null != bodyCharset){
            stringBodyRequest.setBodyCharset(bodyCharset);
        }
        set(stringBodyRequest, queryParams, headers, connectTimeout, readTimeout, resultCharset);
        return stringBodyRequest;
    }

    @Override
    public UploadRequest create(String url,
                                MultiValueMap formParams,
                                FormFile[] formFiles,
                                MultiValueMap headers,
                                int connectTimeout,
                                int readTimeout,
                                String resultCharset) {
        UploadRequest uploadRequest = DefaultUploadRequest.of(url);

        if(MapUtil.notEmpty(formParams)){
            uploadRequest.setFormParams(formParams);
        }
        if(ArrayUtil.isNotEmpty(formFiles)){
            uploadRequest.addFormFile(formFiles);
        }

        set(uploadRequest, null, headers, connectTimeout, readTimeout, resultCharset);

        return uploadRequest;
    }

    protected void set(HttpRequest httpRequest,
                       MultiValueMap queryParams,
                       MultiValueMap headers,
                       int connectTimeout,
                       int readTimeout,
                       String resultCharset){

        if(MapUtil.notEmpty(queryParams)){
            httpRequest.setQueryParams(queryParams);
        }

        if(MapUtil.notEmpty(headers)){
            httpRequest.setHeaders(headers);
        }

        httpRequest.setConnectionTimeout(connectTimeout).setReadTimeout(readTimeout);

        if(null != resultCharset){
            httpRequest.setResultCharset(resultCharset);
        }
    }
}