smartrics.rest.client.RestClientImpl Maven / Gradle / Ivy
/* Copyright 2008 Fabrizio Cannizzo
*
* This file is part of RestFixture.
*
* RestFixture (http://code.google.com/p/rest-fixture/) is free software:
* you can redistribute it and/or modify it under the terms of the
* GNU Lesser General Public License as published by the Free Software Foundation,
* either version 3 of the License, or (at your option) any later version.
*
* RestFixture is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with RestFixture. If not, see multipartFiles = request.getMultipartFileNames();
if ((multipartFiles != null) && (!multipartFiles.isEmpty())) {
requestEntity = configureMultipartFileUpload(m, request, requestEntity, multipartFiles);
} else {
requestEntity = new RequestEntity() {
public boolean isRepeatable() {
return true;
}
public void writeRequest(OutputStream out) throws IOException {
PrintWriter printer = new PrintWriter(out);
printer.print(request.getBody());
printer.flush();
}
public long getContentLength() {
return request.getBody().getBytes().length;
}
public String getContentType() {
List values = request.getHeader("Content-Type");
String v = "text/xml";
if (values.size() != 0)
v = values.get(0).getValue();
return v;
}
};
}
}
((EntityEnclosingMethod) m).setRequestEntity(requestEntity);
} else {
m.setFollowRedirects(request.isFollowRedirect());
}
}
private RequestEntity configureMultipartFileUpload(HttpMethod m, final RestRequest request, RequestEntity requestEntity, Map multipartFiles) {
MultipartRequestEntity multipartRequestEntity = null;
// Current File Name reading for tracking missing file
String fileName = null;
List fileParts = new ArrayList(multipartFiles.size());
// Read File Part
for (Map.Entry multipartFile : multipartFiles.entrySet()) {
Part filePart = createMultipart(multipartFile.getKey(), multipartFile.getValue());
fileParts.add(filePart);
}
Part[] parts = fileParts.toArray(new Part[fileParts.size()]);
multipartRequestEntity = new MultipartRequestEntity(parts, ((EntityEnclosingMethod) m).getParams());
return multipartRequestEntity;
}
private Part createMultipart(String fileParamName, RestMultipart restMultipart) {
RestMultipart.RestMultipartType type = restMultipart.getType();
switch (type) {
case FILE:
String fileName = null;
try {
fileName = restMultipart.getValue();
File file = new File(fileName);
FilePart filePart = new FilePart(fileParamName, file, restMultipart.getContentType(), restMultipart.getCharset());
LOG.info("Configure Multipart file upload paramName={} : ContentType={} for file={} ", new String[]{ fileParamName, restMultipart.getContentType(), fileName});
return filePart;
} catch (FileNotFoundException e) {
throw new IllegalArgumentException("File not found: " + fileName, e);
}
case STRING:
StringPart stringPart = new StringPart(fileParamName, restMultipart.getValue(), restMultipart.getCharset());
stringPart.setContentType(restMultipart.getContentType());
LOG.info("Configure Multipart String upload paramName={} : ContentType={} ", fileParamName, stringPart.getContentType());
return stringPart;
default:
throw new IllegalArgumentException("Unknonw Multipart Type : " + type);
}
}
private RequestEntity configureFileUpload(String fileName) {
final File file = new File(fileName);
if (!file.exists()) {
throw new IllegalArgumentException("File not found: " + fileName);
}
return new FileRequestEntity(file, "application/octet-stream");
}
public String getContentType(RestRequest request) {
List values = request.getHeader("Content-Type");
String v = "text/xml";
if (values.size() != 0)
v = values.get(0).getValue();
return v;
}
private void setUri(HttpMethod m, String hostAddr, RestRequest request) {
String host = hostAddr == null ? client.getHostConfiguration().getHost() : hostAddr;
if (host == null)
throw new IllegalStateException("hostAddress is null: please config httpClient host configuration or " + "pass a valid host address or config a baseUrl on this client");
String uriString = host + request.getResource();
boolean escaped = request.isResourceUriEscaped();
try {
m.setURI(createUri(uriString, escaped));
} catch (URIException e) {
throw new IllegalStateException("Problem when building URI: " + uriString, e);
} catch (NullPointerException e) {
throw new IllegalStateException("Building URI with null string", e);
}
}
protected URI createUri(String uriString, boolean escaped) throws URIException {
return new URI(uriString, escaped);
}
/**
* factory method that maps a string with a HTTP method name to an
* implementation class in Apache HttpClient. Currently the name is mapped
* to org.apache.commons.httpclient.methods.%sMethod where
* %s is the parameter mName.
*
* @param mName the method name
* @return the method class
*/
protected String getMethodClassnameFromMethodName(String mName) {
return String.format("org.apache.commons.httpclient.methods.%sMethod", mName);
}
/**
* Utility method that creates an instance of {@code
* org.apache.commons.httpclient.HttpMethod}.
*
* @param request the rest request
* @return the instance of {@code org.apache.commons.httpclient.HttpMethod}
* matching the method in RestRequest.
*/
@SuppressWarnings("unchecked")
protected HttpMethod createHttpClientMethod(RestRequest request) {
String mName = request.getMethod().toString();
String className = getMethodClassnameFromMethodName(mName);
try {
Class clazz = (Class) Class.forName(className);
if (className.endsWith("TraceMethod")) {
return clazz.getConstructor(String.class).newInstance("http://dummy.com");
} else {
return clazz.newInstance();
}
} catch (ClassNotFoundException e) {
throw new IllegalStateException(className + " not found: you may be using a too old or " + "too new version of HttpClient", e);
} catch (InstantiationException e) {
throw new IllegalStateException("An object of type " + className + " cannot be instantiated", e);
} catch (IllegalAccessException e) {
throw new IllegalStateException("The default ctor for type " + className + " cannot be accessed", e);
} catch (RuntimeException e) {
throw new IllegalStateException("Exception when instantiating: " + className, e);
} catch (InvocationTargetException e) {
throw new IllegalStateException("The ctor with String.class arg for type " + className + " cannot be invoked", e);
} catch (NoSuchMethodException e) {
throw new IllegalStateException("The ctor with String.class arg for type " + className + " doesn't exist", e);
}
}
private void addHeaders(HttpMethod m, RestRequest request) {
for (RestData.Header h : request.getHeaders()) {
m.addRequestHeader(h.getName(), h.getValue());
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy