top.jfunc.http.component.apache.DefaultApacheBodyContentCallbackCreator Maven / Gradle / Ivy

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

import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpRequest;
import top.jfunc.http.base.ContentCallback;
import top.jfunc.http.component.AbstractBodyContentCallbackCreator;
import top.jfunc.http.request.StringBodyRequest;
import top.jfunc.http.util.ApacheUtil;

import java.io.IOException;

/**
 * @author xiongshiyan at 2020/1/7 , contact me with email [email protected] or phone 15208384257
 */
public class DefaultApacheBodyContentCallbackCreator extends AbstractBodyContentCallbackCreator {
    @Override
    protected ContentCallback doCreate(StringBodyRequest stringBodyRequest) throws IOException {
        return request -> setRequestBody(request, stringBodyRequest.getBody(), stringBodyRequest.calculateBodyCharset());
    }

    protected void setRequestBody(HttpRequest request, String body, String bodyCharset){
        if(body == null || !(request instanceof HttpEntityEnclosingRequest)){return;}

        ApacheUtil.setRequestBody((HttpEntityEnclosingRequest)request, body, bodyCharset);
    }
}