Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
package com.sparkpost.transport;
import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.ProtocolException;
import java.net.URL;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.lang3.StringUtils;
import org.apache.log4j.Logger;
import com.sparkpost.Build;
import com.sparkpost.Client;
import com.sparkpost.exception.SparkPostException;
import com.sparkpost.exception.SparkPostIllegalServerResponseException;
import com.sparkpost.model.responses.Response;
/**
* The REST connection class wraps HTTP requests to the SparkPost API.
*
* @author grava
*/
public class RestConnection {
private static final Logger logger = Logger.getLogger(RestConnection.class);
// TODO: set this up to be set by build machine.
private static final String VERSION = Build.VERSION + " (" + Build.GIT_SHORT_HASH + ")";
private static final Base64 BASE64 = new Base64();
private static final String DEFAULT_CHARSET = "UTF-8";
/**
* Default endpoint to use for connections :
* https://api.sparkpost.com/api/v1/
*/
public final static String defaultApiEndpoint = "https://api.sparkpost.com/api/v1/";
private final Client client;
private final String endpoint;
/**
* Supported HTTP methods
*/
public enum Method {
GET,
POST,
PUT,
DELETE
}
/**
* Create a REST connection object. The default endpoint defaultApiEndpoint
* will be used for connections.
*
* @param client
* Client object to use (in particular for authentication info)
* @throws SparkPostException
*/
public RestConnection(Client client) throws SparkPostException {
this(client, null /* means:set to default endpoint */);
}
/**
* Create a REST connection object. The given {@link endpoint} will be used
* for connections
*
* @param client
* Client object to use (in particular for authentication info)
* @param endpoint
* Endpoint to use instead of the default defaultApiEndpoint
* @throws SparkPostException
*/
public RestConnection(Client client, String endpoint) throws SparkPostException {
this.client = client;
if (StringUtils.isAnyEmpty(endpoint)) {
this.endpoint = defaultApiEndpoint;
} else {
if (endpoint.endsWith("/")) {
this.endpoint = endpoint;
} else {
this.endpoint = endpoint + '/';
}
}
}
/**
* Create a connection object but doesn't actually connect to the server.
*
* @param path
* URL suffix to be appended to the object's endpoint
* @param method
* HTTP method for the connection
* @return the connection object
* @throws SparkPostException
*/
private HttpURLConnection createConnectionObject(String path, Method method) throws SparkPostException {
HttpURLConnection conn;
try {
URL url;
url = new URL(this.endpoint + path);
// Retrieve the URLConnection object (but doesn't actually connect):
// (HttpUrlConnection doesn't connect to the server until we've
// got one of its streams)
conn = (HttpURLConnection) url.openConnection();
if (StringUtils.isNotEmpty(this.client.getAuthKey())) {
conn.setRequestProperty("Authorization", this.client.getAuthKey());
} else if (StringUtils.isNotEmpty(this.client.getUsername()) && StringUtils.isNotEmpty(this.client.getPassword())) {
String encoding = BASE64.encodeAsString((this.client.getUsername() + ":" + this.client.getPassword()).getBytes(DEFAULT_CHARSET));
conn.setRequestProperty("Authorization", "Basic " + encoding);
}
conn.setRequestProperty("User-Agent", "java-sparkpost/" + VERSION);
conn.setRequestProperty("Content-Type", "application/json");
switch (method) {
case GET:
conn.setRequestMethod("GET");
if (logger.isDebugEnabled()) {
logger.debug("GET " + url);
}
break;
case POST:
conn.setRequestMethod("POST");
// we write the POST data to the "output" stream:
conn.setDoOutput(true);
if (logger.isDebugEnabled()) {
logger.debug("POST " + path);
}
break;
case PUT:
conn.setRequestMethod("PUT");
// we write the POST data to the "output" stream:
conn.setDoOutput(true);
if (logger.isDebugEnabled()) {
logger.debug("PUT " + path);
}
break;
case DELETE:
conn.setRequestMethod("DELETE");
if (logger.isDebugEnabled()) {
logger.debug("DELETE " + path);
}
break;
default:
throw new SparkPostException("Invalid Method");
}
} catch (MalformedURLException ex) {
throw new SparkPostException("Invalid path: " + path + ex.toString());
} catch (ProtocolException ex) {
throw new SparkPostException("Invalid method:" + ex.toString());
} catch (IOException ex) {
throw new SparkPostException("Error with connection to " + path + ex.toString());
}
return conn;
}
// Send HTTP data (payload) to server
private void sendData(HttpURLConnection conn, String data) throws SparkPostException {
byte[] bytes = null;
try
{
bytes = data.getBytes(DEFAULT_CHARSET);
}
catch (UnsupportedEncodingException e)
{
// This should never happen. UTF-8 should always be available but we
// have to catch it so pass it on if it fails.
throw new SparkPostException(e);
}
String lenStr = Integer.toString(bytes.length);
conn.setRequestProperty("Content-Length", lenStr);
conn.setRequestProperty("Content-Type", "application/json");
if (logger.isDebugEnabled()) {
logger.debug("Sending data (" + lenStr + " bytes): " + data);
}
// Send data. At this point connection to server may not be established,
// but writing data to it will trigger the connection.
try (DataOutputStream wr = new DataOutputStream(conn.getOutputStream())) {
wr.write(bytes);
wr.flush();
} catch (IOException ex) {
throw new SparkPostException("Error sending request data:" + ex.toString());
}
}
// Send HTTP request to server
private void sendRequest(HttpURLConnection conn, String data, Response response) throws SparkPostException {
if (data != null) {
sendData(conn, data);
}
try {
// If no data was sent (right above), then connection to server
// is not made yet ; however asking for the response code (below)
// makes conn connect to the server,
// send the request, and start reading the response.
// getResponseCode() blocks until the response code is read from the
// stream from the server
int code = conn.getResponseCode();
response.setResponseCode(code);
response.setContentType(conn.getHeaderField("Content-Type"));
String msg = conn.getResponseMessage();
response.setResponseMessage(msg);
} catch (IOException ex) {
throw new SparkPostException("Connection error:" + ex.toString());
}
}
// Read response body from server
private Response receiveResponse(HttpURLConnection conn, Response response) throws SparkPostException {
if (!conn.getContentType().toLowerCase().startsWith("application/json")) {
throw new SparkPostIllegalServerResponseException("Unexpected content type (" + conn.getContentType() + ") from " + conn.getURL());
}
StringBuilder sb = new StringBuilder();
try (BufferedReader rd = new BufferedReader(new InputStreamReader(conn.getInputStream(), DEFAULT_CHARSET))) {
// Buffer the result into a string:
String line;
while ((line = rd.readLine()) != null) {
sb.append(line);
}
response.setResponseBody(sb.toString());
response.setRequestId(conn.getHeaderField("X-SparkPost-Request-Id"));
} catch (FileNotFoundException ex) {
// We get here if the connection was closed:
// There are cases in REST where the server won't return a response
// body but only a response status. So if we get here , it is not
// an error.
response.setResponseBody("");
} catch (IOException ex) {
throw new SparkPostException("Error reading server response: " + ex.toString() + ": " + sb.toString() + "(" + response.getResponseMessage() + ")");
}
return response;
}
// This method actually performs an HTTP request.
// It is called by get(), put(), post() and delete() below
private Response doHttpMethod(String path, Method method, String data, Response response) throws SparkPostException {
HttpURLConnection conn = null;
try {
response.setRequest(path);
conn = createConnectionObject(path, method);
sendRequest(conn, data, response);
receiveResponse(conn, response);
if (logger.isDebugEnabled()) {
logger.debug("Server Response:" + response);
}
return response;
} finally {
if (conn != null) {
conn.disconnect();
}
}
}
/**
* Perform an HTTP GET request. This method throws an exception if the
* server returns anything else than a 200.
*
* @param path
* API endpoint to send the request to.
* @return Server response to the request.
* @throws SparkPostException
* if something goes wrong
*/
public Response get(String path) throws SparkPostException {
Response response = new Response();
return doHttpMethod(path, Method.GET, null, response);
}
/**
* Perform an HTTP POST request. This method throws an exception if the
* server returns anything else than a 200.
*
* @param path
* API endpoint to send the request to.
* @param json
* POST data block to send with the request. May be null.
* @return Server response to the request.
* @throws SparkPostException
* if something goes wrong
*/
public Response post(String path, String json) throws SparkPostException {
Response response = new Response();
return doHttpMethod(path, Method.POST, json, response);
}
/**
* Perform an HTTP PUT request. This method throws an exception if the
* server returns anything else than a 200.
*
* @param path
* API endpoint to send the request to.
* @param json
* PUT data block to send with the request. May be null.
* @return Server response to the request.
* @throws SparkPostException
* if something goes wrong
*/
public Response put(String path, String json) throws SparkPostException {
Response response = new Response();
return doHttpMethod(path, Method.PUT, json, response);
}
/**
* Perform an HTTP DELETE request. This method throws an exception if the
* server returns anything else than a 200.
*
* @param path
* API endpoint to send the request to.
* @return Server response to the request.
* @throws SparkPostException
* if something goes wrong
*/
public Response delete(String path) throws SparkPostException {
Response response = new Response();
return doHttpMethod(path, Method.DELETE, null, response);
}
}
