/*!
* AtlantBH Custom Jmeter Components v1.0.0
* http://www.atlantbh.com/jmeter-components/
*
* Copyright 2011, AtlantBH
*
* Licensed under the under the Apache License, Version 2.0.
*/
package com.atlantbh.jmeter.plugins.rest;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.zip.GZIPInputStream;
import org.apache.commons.httpclient.Header;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpMethodBase;
import org.apache.commons.httpclient.methods.DeleteMethod;
import org.apache.commons.httpclient.methods.EntityEnclosingMethod;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.HeadMethod;
import org.apache.commons.httpclient.methods.OptionsMethod;
import org.apache.commons.httpclient.methods.PostMethod;
import org.apache.commons.httpclient.methods.PutMethod;
import org.apache.commons.httpclient.methods.StringRequestEntity;
import org.apache.commons.httpclient.methods.TraceMethod;
import org.apache.jmeter.protocol.http.sampler.HTTPSampleResult;
import org.apache.jmeter.protocol.http.sampler.HTTPSampler2;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.jorphan.util.JOrphanUtils;
import org.apache.log.Logger;
import org.apache.jmeter.samplers.SampleResult;
public class RestSampler extends HTTPSampler2 {
private static final long serialVersionUID = -5877623539165274730L;
private static final Logger log = LoggingManager.getLoggerForClass();
public static final String REQUEST_BODY = "RestSampler.request_body";
public static final String PORT_NUMBER = "RestSampler.port_number";
public static final String RESOURCE = "RestSampler.resource";
public static final String BASE_HOST = "RestSampler.base_host";
public static final String REQUEST_HEADERS = "RestSampler.request_headers";
public RestSampler() {
}
public void setRequestBody(String data) {
setProperty(REQUEST_BODY, data);
}
public void setRequestHeaders(String headers) {
setProperty(REQUEST_HEADERS, headers);
}
public String getRequestBody() {
return getPropertyAsString(REQUEST_BODY);
}
public String getRequestHeaders() {
return getPropertyAsString(REQUEST_HEADERS);
}
public void setResource(String data) {
setProperty(RESOURCE, data);
}
public String getResource() {
return getPropertyAsString(RESOURCE);
}
public void setPortNumber(String data) {
setProperty(PORT_NUMBER, data);
}
public String getPortNumber()
{
return getPropertyAsString(PORT_NUMBER);
}
public void setHostBaseUrl(final String data) {
setProperty(BASE_HOST, data);
}
public String getHostBaseUrl() {
return getPropertyAsString(BASE_HOST);
}
public URL getUrl() throws MalformedURLException{
URL u = null;
try {
String validHost = toValidUrl(getHostBaseUrl());
if (validHost != null && getResource() != null) {
String fullUrl = validHost + ":" + getPortNumber() + "/" + getResource();
u = toURL(fullUrl);
}
} catch (MalformedURLException e) {
if(isProtocolDefaultPort()) {
u = new URL(getProtocol(), getDomain(), getPath() + "/" + getResource());
} else {
u = new URL(getProtocol(), getDomain(), getPort(), getPath() + "/" + getResource());
}
}
return u;
}
public String toString() {
return "Base host url: " + getHostBaseUrl() + ", resource: " + getResource() + ", Method: " + getMethod();
}
private String toValidUrl(String u) throws MalformedURLException{
URL url = new URL(u);
String urlStr = url.toString();
if (urlStr.endsWith("/")) {
url = toURL(urlStr.substring(0, urlStr.length() - 1));
urlStr = url.toString();
}
return urlStr;
}
private URL toURL(String u) throws MalformedURLException{
return new URL(u);
}
private void overrideHeaders(HttpMethodBase httpMethod) {
String headers = getRequestHeaders();
String[] header = headers.split(System.getProperty("line.separator"));
for (String kvp : header) {
int pos = kvp.indexOf(':');
if (pos < 0)
pos = kvp.indexOf('=');
if (pos > 0) {
String k = kvp.substring(0, pos).trim();
String v = "";
if (kvp.length() > pos + 1)
v = kvp.substring(pos + 1).trim();
httpMethod.addRequestHeader(k, v);
}
}
}
protected HttpClient setupConnection(URL u, HttpMethodBase httpMethod) throws IOException
{
HTTPSampleResult temp = new HTTPSampleResult();
return super.setupConnection(u, httpMethod, temp);
}
protected HTTPSampleResult sample(URL url, String method, boolean areFollowingRedirect, int frameDepth){
throw new RuntimeException("Not implemented - should not be called");
}
/**
* Method invoked by JMeter when a sample needs to happen. It's actually an
* indirect call from the main sampler interface. it's resolved in the base
* class.
*
* This is a copy and paste from the HTTPSampler2 - quick and dirty hack as
* that class is not very extensible. The reason to extend and slightly
* modify is that I needed to get the body content from a text field in the
* GUI rather than a file.
*/
public SampleResult sample() {
HttpMethodBase httpMethod = null;
HttpClient client = null;
InputStream instream = null;
SampleResult res = new SampleResult();
try {
res.setSuccessful(false);
res.setResponseCode("000");
res.setSampleLabel(getName());
res.setURL(getUrl());
res.setDataEncoding("UTF-8");
res.setDataType("text/xml");
res.setMonitor(isMonitor());
res.sampleStart();
String urlStr = getUrl().toString();
String request = getMethod().toString() + " " + urlStr + "\n";
request += getRequestBody().toString();
res.setSamplerData(request);
log.debug("Start : sample " + urlStr);
log.debug("method " + getMethod());
httpMethod = createHttpMethod(getMethod(), urlStr);
// Set any default request headers
setDefaultRequestHeaders(httpMethod);
// Setup connection
client = setupConnection(getUrl(), httpMethod);
// Handle the various methods
if (httpMethod instanceof EntityEnclosingMethod) {
((EntityEnclosingMethod)httpMethod).setRequestEntity(new StringRequestEntity(getRequestBody(), "text/xml", "UTF-8"));
//res.setQueryString(getRequestBody());
//String postBody = sendData((EntityEnclosingMethod) httpMethod);
//res.setResponseData(postBody.getBytes());
//String postBody = "";
//try { postBody = Base64Util.processStargateRequest(getRequestBody());}
//catch (Exception e) {postBody =getRequestBody(); e.printStackTrace();}
//
//res.setSamplerData(/*res.getSamplerData() + "\r\n*/"ORIGINAL CONTENT:\r\n\r\n" + getRequestBody() + "\r\n\r\nSENT CONTENT:\r\n\r\n" + postBody);
}
overrideHeaders(httpMethod);
res.setRequestHeaders(getConnectionHeaders(httpMethod));
int statusCode = -1;
try {
statusCode = client.executeMethod(httpMethod);
} catch (RuntimeException e) {
log.error("Exception when executing '" + httpMethod + "'", e);
throw e;
}
// Request sent. Now get the response:
instream = httpMethod.getResponseBodyAsStream();
if (instream != null) {// will be null for HEAD
Header responseHeader = httpMethod.getResponseHeader(HEADER_CONTENT_ENCODING);
if (responseHeader != null && ENCODING_GZIP.equals(responseHeader.getValue())) {
instream = new GZIPInputStream(instream);
}
res.setResponseData(readResponse(res, instream, (int) httpMethod.getResponseContentLength()));
}
//if (res.getResponseDataAsString().length() != 0)
// try{
// String response = Base64Util.processStargateResponse(res.getResponseDataAsString());
// res.setResponseData(response);
// } catch (Exception e)
// {
// log.error(e.getMessage());
// }
res.sampleEnd();
// Done with the sampling proper.
// Now collect the results into the HTTPSampleResult:
// res.setSampleLabel(httpMethod.getURI().toString());
// Pick up Actual path (after redirects)
res.setResponseCode(Integer.toString(statusCode));
res.setSuccessful(isSuccessCode(statusCode));
res.setResponseMessage(httpMethod.getStatusText());
String ct = null;
org.apache.commons.httpclient.Header h = httpMethod.getResponseHeader(HEADER_CONTENT_TYPE);
if (h != null)// Can be missing, e.g. on redirect
{
ct = h.getValue();
res.setContentType(ct);// e.g. text/html; charset=ISO-8859-1
res.setEncodingAndType(ct);
}
String responseHeaders = getResponseHeaders(httpMethod);
res.setResponseHeaders(responseHeaders);
/*if (res.isRedirect()) {
final Header headerLocation = httpMethod.getResponseHeader(HEADER_LOCATION);
if (headerLocation == null) { // HTTP protocol violation, but
// avoids NPE
throw new IllegalArgumentException("Missing location header");
}
res.setRedirectLocation(headerLocation.getValue());
}
// If we redirected automatically, the URL may have changed
if (getAutoRedirects()) {
res.setURL(new URL(httpMethod.getURI().toString()));
}
// Store any cookies received in the cookie manager:
saveConnectionCookies(httpMethod, res.getURL(), getCookieManager());
// Save cache information
final CacheManager cacheManager = getCacheManager();
if (cacheManager != null) {
cacheManager.saveDetails(httpMethod, res);
}
// Follow redirects and download page resources if appropriate:
res = resultProcessing(areFollowingRedirect, frameDepth, res);
*/
log.debug("End : sample");
httpMethod.releaseConnection();
return res;
}
catch (MalformedURLException e) {
res.sampleEnd();
log.warn(e.getMessage());
res.setResponseMessage(e.getMessage());
return res;
} catch (IllegalArgumentException e)// e.g. some kinds of invalid URL
{
res.sampleEnd();
//SampleResult err = errorResult(e, res);
//err.setSampleLabel("Error: " + url.toString());
log.warn(e.getMessage());
res.setResponseMessage(e.getMessage());
return res;
} catch (IOException e) {
res.sampleEnd();
log.warn(e.getMessage());
res.setResponseMessage(e.getMessage());
return res;
} finally {
JOrphanUtils.closeQuietly(instream);
if (httpMethod != null) {
httpMethod.releaseConnection();
return res;
}
}
}
private void setDefaultRequestHeaders(HttpMethodBase httpMethod) {
// TODO Auto-generated method stub
}
private HttpMethodBase createHttpMethod(String method, String urlStr) {
HttpMethodBase httpMethod;
// May generate IllegalArgumentException
if (method.equals(POST)) {
httpMethod = new PostMethod(urlStr);
} else if (method.equals(PUT)) {
httpMethod = new PutMethod(urlStr);
} else if (method.equals(HEAD)) {
httpMethod = new HeadMethod(urlStr);
} else if (method.equals(TRACE)) {
httpMethod = new TraceMethod(urlStr);
} else if (method.equals(OPTIONS)) {
httpMethod = new OptionsMethod(urlStr);
} else if (method.equals(DELETE)) {
httpMethod = new DeleteMethod(urlStr);
} else if (method.equals(GET)) {
httpMethod = new GetMethod(urlStr);
} else {
log.error("Unexpected method (converted to GET): " + method);
httpMethod = new GetMethod(urlStr);
}
return httpMethod;
}
}