com.nineya.tool.http.HttpRequest Maven / Gradle / Ivy

Go to download
package com.nineya.tool.http;

import java.net.MalformedURLException;
import java.net.URL;
import java.util.Map;
import java.util.stream.Collectors;

import com.nineya.tool.restful.Methods;

public class HttpRequest extends AbstractHttpRequest {
    private String method;
    private String url;
    private String body;

    private HttpRequest(String method, String url) {
        this.method = method;
        this.url = url;
    }

    /**
     * 创建post请求
     *
     * @param url
     * @return
     */
    public static HttpRequest sendPost(String url) {
        return new HttpRequest(Methods.POST, url);
    }

    /**
     * 创建get请求
     *
     * @param url
     * @return
     */
    public static HttpRequest sendGet(String url) {
        return new HttpRequest(Methods.GET, url);
    }

    public String getMethod() {
        return method;
    }

    public URL getUrl() {
        try {
            Map params = getParams();
            if (params == null || params.isEmpty()) {
                return new URL(url);
            }
            String paramsStr = getParams().entrySet().stream()
                .map(entry -> entry.getKey() + "=" + entry.getValue())
                .collect(Collectors.joining("&", url.contains("?") ? "&" : "?", ""));
            return new URL(url + paramsStr);
        } catch (MalformedURLException e) {
            e.printStackTrace();
        }
        return null;
    }

    public HttpRequest setBody(String body) {
        this.body = body;
        return this;
    }

    public HttpRequest setBody(HttpEntity entity) {
        this.body = entity.getContent();
        if (entity.getContentEncoding() != null) {
            setContentEncoding(entity.getContentEncoding());
        }
        if (entity.getContentType() != null) {
            setContentType(entity.getContentType());
        }
        return this;
    }

    public String getBody() {
        return body;
    }
}