com.seejoke.wechat.util.http.MediaUploadRequestExecutor Maven / Gradle / Ivy

Go to download
package com.seejoke.wechat.util.http;

import com.seejoke.wechat.entity.WxVideoIntroduction;
import com.seejoke.wechat.entity.result.WxError;
import com.seejoke.wechat.entity.result.WxMediaUploadResult;
import com.seejoke.wechat.exception.WxErrorException;
import org.apache.http.HttpEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.mime.HttpMultipartMode;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.StringBody;
import org.apache.http.impl.client.CloseableHttpClient;
import org.codehaus.jackson.JsonNode;
import org.codehaus.jackson.map.ObjectMapper;

import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;

public class MediaUploadRequestExecutor implements RequestExecutor {
	private WxVideoIntroduction introduction;
	
	public MediaUploadRequestExecutor() {
	}
	
	public MediaUploadRequestExecutor(WxVideoIntroduction introduction){
		this.introduction = introduction;
	}
	
	@Override
	public WxMediaUploadResult execute(CloseableHttpClient httpclient, String uri, File file)
			throws WxErrorException, IOException {
		HttpPost httpPost = new HttpPost(uri);

		if (file != null) {
			HttpEntity entity = null;
			if(this.introduction !=null){
				StringBody stringBody = new StringBody(introduction.toJson(),ContentType.TEXT_PLAIN.withCharset(StandardCharsets.UTF_8));
				entity = MultipartEntityBuilder.create().setCharset(StandardCharsets.UTF_8).addBinaryBody("media", file).addPart("description", stringBody).setMode(HttpMultipartMode.RFC6532).build();
			}else{
				entity = MultipartEntityBuilder.create().addBinaryBody("media", file).setMode(HttpMultipartMode.RFC6532).build();
			}
			httpPost.setEntity(entity);
			httpPost.setHeader("Content-Type", ContentType.MULTIPART_FORM_DATA.toString());
		}
		try (CloseableHttpResponse response = httpclient.execute(httpPost)) {
			String responseContent = Utf8ResponseHandler.INSTANCE.handleResponse(response);
			ObjectMapper mapper = new ObjectMapper();
			JsonNode node = mapper.readTree(responseContent);
			if(node.get("errcode")!=null && !(node.get("errcode").asInt()==0)){
				WxError error = WxError.fromJson(responseContent);
				throw new WxErrorException(error);
			}
			return WxMediaUploadResult.fromJson(responseContent);
		}
	}

}