com.qiniu.api.net.Client Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdk Show documentation
Qiniu Resource (Cloud) Storage SDK demo for Java
The newest version!
package com.qiniu.api.net;

import java.util.List;

import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.StatusLine;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.AbstractHttpEntity;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.StringEntity;
import org.apache.http.entity.mime.MultipartEntity;
import org.apache.http.util.EntityUtils;

import com.qiniu.api.auth.AuthException;
import com.qiniu.api.config.*;

/**
 * The class {@code Client} is a typical wrapper of RPC. Also see
 * {@code com.qiniu.api.auth.DigestAuthClient}
 */
public class Client {

	private final static  String HEADER_AGENT="User-Agent";

	/**
	 *
	 * @param post
	 * @throws AuthException
	 */
	public void setAuth(HttpPost post) throws AuthException {

	}

	public static HttpPost newPost(String url){
		HttpPost postMethod = new HttpPost(url);
		postMethod.setHeader(HEADER_AGENT, getUserAgent());
		return postMethod;
	}

	private static String getUserAgent(){
		String javaVersion = "Java/" + System.getProperty("java.version");
		String os = System.getProperty("os.name") + " "
		+  System.getProperty("os.arch") + " " + System.getProperty("os.version");
		String sdk = "QiniuJava/" + Config.VERSION;
		return sdk + " (" + os +") " + javaVersion;
	}

	/**
	 * Sends a http post request to the specified url.
	 *
	 * @param url
	 *            the request url
	 * @return A general response
	 */
	public CallRet call(String url) {
		HttpClient client = Http.getClient();
		HttpPost postMethod = newPost(url);
		try {
			setAuth(postMethod);
			HttpResponse response = client.execute(postMethod);
			return handleResult(response);
		} catch (Exception e) {
			e.printStackTrace();
			return new CallRet(Config.ERROR_CODE, e);
		}
	}

	/**
	 * Sends a http post request to the specified url with a list of
	 * NameValuePair.
	 *
	 * @param url
	 *            the request url
	 * @param nvps
	 * @return A general response
	 */
	public CallRet call(String url, List nvps) {
		HttpClient client = Http.getClient();
		HttpPost postMethod = newPost(url);
		try {
			StringEntity entity = new UrlEncodedFormEntity(nvps, "UTF-8");
			entity.setContentType("application/x-www-form-urlencoded");
			postMethod.setEntity(entity);
			setAuth(postMethod);
			HttpResponse response = client.execute(postMethod);

			return handleResult(response);
		} catch (Exception e) {
			e.printStackTrace();
			return new CallRet(Config.ERROR_CODE, e);
		}
	}

	/**
	 * Sends a http request to the specified url with the specified entity.
	 * @param url
	 * @param entity
	 * @return A general response
	 */
	public CallRet callWithBinary(String url, AbstractHttpEntity entity) {
		HttpClient client = Http.getClient();
		HttpPost postMethod = newPost(url);
		postMethod.setEntity(entity);

		try {
			setAuth(postMethod);
			HttpResponse response = client.execute(postMethod);
			return handleResult(response);
		} catch (Exception e) {
			e.printStackTrace();
			return new CallRet(Config.ERROR_CODE, e);
		}
	}

	/**
	 *
	 * @param url
	 *            the request url
	 * @param contentType
	 *            the request content type
	 * @param body
	 *            the request body
	 * @param bodyLength
	 *            the length of the request body
	 * @return A general response
	 */
	public CallRet callWithBinary(String url, String contentType, byte[] body) {
		ByteArrayEntity entity = new ByteArrayEntity(body);

		if (contentType == null || contentType.isEmpty()) {
			contentType = "application/octet-stream";
		}
		entity.setContentType(contentType);
		return callWithBinary(url, entity);
	}

	/**
	 *
	 * @param url
	 * @param requestEntity
	 * @return A general response format
	 */
	public CallRet callWithMultiPart(String url, MultipartEntity requestEntity) {
		HttpPost postMethod = newPost(url);
		postMethod.setEntity(requestEntity);
		HttpClient client = Http.getClient();

		try {
			HttpResponse response = client.execute(postMethod);
			return handleResult(response);
		} catch (Exception e) {
			e.printStackTrace();
			return new CallRet(Config.ERROR_CODE, e);
		}
	}

	/**
	 * Transforms a httpresponse to user expected format.
	 *
	 * @param response
	 *            http response body
	 * @return a formated general response structure
	 */
	private CallRet handleResult(HttpResponse response) {
		if (response == null || response.getStatusLine() == null) {
			return new CallRet(Config.ERROR_CODE, "No response");
		}

		String responseBody;
		try {
			responseBody = EntityUtils.toString(response.getEntity(),"UTF-8");
		} catch (Exception e) {
			e.printStackTrace();
			return new CallRet(Config.ERROR_CODE, e);
		}

		StatusLine status = response.getStatusLine();
		int statusCode = (status == null) ? Config.ERROR_CODE : status.getStatusCode();
		return new CallRet(statusCode, responseBody);
	}

}