EntityBuilder.java

package com.litesuits.http.impl.apache;

import com.litesuits.http.data.Consts;
import com.litesuits.http.exception.HttpClientException;
import com.litesuits.http.impl.apache.entity.FileEntity;
import com.litesuits.http.impl.apache.entity.InputStreamEntity;
import com.litesuits.http.impl.apache.entity.MultipartEntity;
import com.litesuits.http.request.AbstractRequest;
import com.litesuits.http.request.content.*;
import com.litesuits.http.request.content.multi.MultipartBody;
import org.apache.http.HttpEntity;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.StringEntity;

/**
 * help us to build {@link org.apache.http.HttpEntity}
 *
 * @author MaTianyu
 *         2014-1-18上午1:41:41
 */
public class EntityBuilder {

    public static HttpEntity build(AbstractRequest req) throws HttpClientException {
        try {
            HttpBody body = req.getHttpBody();
            if (body != null) {
                req.addHeader(Consts.CONTENT_TYPE, body.getContentType());
                if (body instanceof StringBody) {
                    // StringBody JsonBody UrlEncodedFormBody
                    StringBody b = (StringBody) body;
                    return new StringEntity(b.getString(), b.getCharset());
                } else if (body instanceof ByteArrayBody) {
                    // ByteArrayBody SerializableBody
                    ByteArrayBody b = (ByteArrayBody) body;
                    return new ByteArrayEntity(b.getBytes());
                } else if (body instanceof InputStreamBody) {
                    InputStreamBody b = (InputStreamBody) body;
                    return new InputStreamEntity(b.getInstream(), b.getInstream().available(), req);
                } else if (body instanceof FileBody) {
                    FileBody b = (FileBody) body;
                    return new FileEntity(b.getFile(), b.getContentType(), req);
                } else if (body instanceof MultipartBody) {
                    return new MultipartEntity((MultipartBody) body);
                } else {
                    throw new RuntimeException("Unpredictable Entity Body(非法实体)");
                }
            }
        } catch (Exception e) {
            throw new HttpClientException(e);
        }
        return null;
    }
}