io.crnk.spring.client.RestTemplateRequest Maven / Gradle / Ivy

Go to download
package io.crnk.spring.client;

import java.io.IOException;
import java.util.Arrays;

import io.crnk.client.http.HttpAdapterRequest;
import io.crnk.client.http.HttpAdapterResponse;

import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.client.HttpClientErrorException;
import org.springframework.web.client.RestTemplate;

public class RestTemplateRequest implements HttpAdapterRequest {

	private static final MediaType CONTENT_TYPE =
			MediaType.parseMediaType(io.crnk.core.engine.http.HttpHeaders.JSONAPI_CONTENT_TYPE_AND_CHARSET);

	private final RestTemplate template;

	private final String requestBody;

	private final io.crnk.core.engine.http.HttpMethod method;

	private final String url;

	private HttpHeaders headers;

	public RestTemplateRequest(RestTemplate template, String url, io.crnk.core.engine.http.HttpMethod method,
			String requestBody) {
		this.template = template;
		this.requestBody = requestBody;
		this.url = url;
		this.method = method;

		headers = new HttpHeaders();
		headers.setAccept(Arrays.asList(CONTENT_TYPE));
		if (requestBody != null) {
			headers.setContentType(CONTENT_TYPE);
		}
	}

	@Override
	public void header(String name, String value) {
		headers.set(name, value);
	}

	@Override
	public RestTemplateResponse execute() throws IOException {
		HttpEntity entityReq = new HttpEntity<>(requestBody, headers);
		try {
			ResponseEntity response = template.exchange(url, HttpMethod.resolve(method.name()), entityReq, String.class);
			return new RestTemplateResponse(response);
		}
		catch (HttpClientErrorException e) {
			return new RestTemplateResponse(e.getRawStatusCode(), e.getStatusCode().getReasonPhrase(), e.getResponseBodyAsString
					(), e.getResponseHeaders());
		}
	}

}