org.springframework.social.partnercenter.http.client.HttpRequestBuilder Maven / Gradle / Ivy

Go to download
package org.springframework.social.partnercenter.http.client;

import static java.util.Collections.singletonList;

import java.util.List;
import java.util.UUID;

import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;
import org.springframework.web.util.UriComponentsBuilder;

public class HttpRequestBuilder {
	public static final String MS_REQUEST_ID_HEADER_NAME = "MS-RequestId";
	public static final String MS_CORRELATION_ID_HEADER_NAME = "MS-CorrelationId";
	private HttpHeaders headers;
	private RestResource restResource;
	private UriComponentsBuilder uriBuilder;

	HttpRequestBuilder(RestResource restResource, String resourceBaseUri, String msRequestId, String msCorrelationId){
		this.headers = new HttpHeaders();
		addMicrosoftTrackingHeaders(msRequestId, msCorrelationId);
		this.restResource = restResource;
		this.uriBuilder = UriComponentsBuilder.fromUriString(resourceBaseUri);
	}

	HttpRequestBuilder(RestResource restResource, String resourceBaseUri){
		this.headers = new HttpHeaders();
		addMicrosoftTrackingHeaders();
		this.restResource = restResource;
		this.uriBuilder = UriComponentsBuilder.fromUriString(resourceBaseUri);
	}

	public HttpRequestBuilder header(String key, List value){
		headers.put(key, value);
		return this;
	}

	public HttpRequestBuilder header(String key, String value){
		headers.put(key, singletonList(value));
		return this;
	}

	public HttpRequestBuilder pathSegment(String ... pathSegments){
		uriBuilder.pathSegment(pathSegments);
		return this;
	}

	public HttpRequestBuilder queryParam(String name, Object... values){
		uriBuilder.queryParam(name, values);
		return this;
	}

	public HttpRequestBuilder path(String path){
		uriBuilder.path(path);
		return this;
	}

	public  ResponseEntity put(T payload, Class aClass){
		HttpEntity tHttpEntity = new HttpEntity<>(payload, headers);
		return restResource.put(uriBuilder.build().toUri(), tHttpEntity, aClass);
	}

	public  ResponseEntity post(T payload, Class aClass){
		HttpEntity tHttpEntity = new HttpEntity<>(payload, headers);
		return restResource.post(uriBuilder.build().toUri(), tHttpEntity, aClass);
	}

	public  ResponseEntity patch(T payload, Class aClass){
		HttpEntity tHttpEntity = new HttpEntity<>(payload, headers);
		return restResource.patch(uriBuilder.build().toUri(), tHttpEntity, aClass);
	}

	public  ResponseEntity get(Class aClass){
		HttpEntity tHttpEntity = new HttpEntity<>(headers);
		return restResource.get(uriBuilder.build().toUri(), aClass, headers);
	}

	public  ResponseEntity get(ParameterizedTypeReference aClass){
		return restResource.get(uriBuilder.build().toUri(), aClass, headers);
	}

	public ResponseEntity delete(){
		return restResource.delete(uriBuilder.build().toUri(), headers);
	}

	private void addMicrosoftTrackingHeaders() {
		headers.set(MS_REQUEST_ID_HEADER_NAME, UUID.randomUUID().toString());
		headers.set(MS_CORRELATION_ID_HEADER_NAME, UUID.randomUUID().toString());
	}
	private void addMicrosoftTrackingHeaders(String msRequestId, String msCorrelationId) {
		headers.set(MS_REQUEST_ID_HEADER_NAME, msRequestId);
		headers.set(MS_CORRELATION_ID_HEADER_NAME, msCorrelationId);
	}
}