io.imunity.furms.unity.client.UnityClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of furms-unity-client Show documentation
FURMS Unity DAO/Client module
There is a newer version: 4.3.1
/*
 * Copyright (c) 2020 Bixbit s.c. All rights reserved.
 *  See LICENSE file for licensing information.
 */

package io.imunity.furms.unity.client;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.util.UriBuilder;
import org.springframework.web.util.UriComponentsBuilder;
import reactor.util.function.Tuple2;

import java.net.URI;
import java.util.List;
import java.util.Map;

import static java.util.Collections.emptyMap;
import static java.util.stream.Collectors.toMap;

@Component
public class UnityClient {

	private static final Logger LOG = LoggerFactory.getLogger(UnityClient.class);


	private final WebClient webClient;

	public UnityClient(WebClient webClient) {
		this.webClient = webClient;
	}

	public  T get(String path, Class clazz) {
		return get(path, clazz, emptyMap());
	}

	public  T get(String path, ParameterizedTypeReference typeReference) {
		return get(path, typeReference, emptyMap());
	}

	public  T get(String path, Class clazz, Map queryParams) {
		MultiValueMap params = createStringParams(queryParams);
		return webClient.get()
				.uri(uriBuilder -> uri(uriBuilder, path, params))
				.retrieve()
				.bodyToMono(clazz)
				.block();
	}

	public  T getWithListParam(String path, ParameterizedTypeReference typeReference, Map> queryParams) {
		return get(path, typeReference, createListParams(queryParams));
	}
	
	public  T get(String path, ParameterizedTypeReference typeReference, Map queryParams) {
		return get(path, typeReference, createStringParams(queryParams));
	}

	public  T get(String path, Map> queryParams, ParameterizedTypeReference typeReference) {
		return get(path, typeReference, new LinkedMultiValueMap<>(queryParams));
	}

	private  T get(String path, ParameterizedTypeReference typeReference, MultiValueMap params) {
		return webClient.get()
			.uri(uriBuilder -> uri(uriBuilder, path, params))
			.retrieve()
			.bodyToMono(typeReference)
			.elapsed()
			.doOnNext(tuple -> LOG.info("Path {}, Time {}", path, tuple.getT1()))
			.map(Tuple2::getT2)
			.block();
	}

	public void post(String path) {
		post(path, null);
	}

	public void post(String path, Object body) {
		post(path, body, emptyMap());
	}

	public  T post(String path, Object body, Map queryParams, ParameterizedTypeReference typeReference) {
		MultiValueMap params = createStringParams(queryParams);
		return webClient.post()
			.uri(uriBuilder -> uri(uriBuilder, path, params))
			.bodyValue(body == null ? "" : body)
			.retrieve()
			.bodyToMono(typeReference)
			.block();
	}

	public void post(String path, Object body, Map queryParams) {
		MultiValueMap params = createStringParams(queryParams);
		webClient.post()
				.uri(uriBuilder -> uri(uriBuilder, path, params))
				.bodyValue(body == null ? "" : body)
				.retrieve()
				.bodyToMono(Void.class).block();
	}

	public void post(URI uri, MediaType mediaType) {
		webClient.post()
				.uri(u -> uri)
				.contentType(mediaType)
				.bodyValue("")
				.retrieve()
				.bodyToMono(Void.class).block();
	}

	public void put(String path, Object body) {
		webClient.put()
				.uri(uriBuilder -> uri(uriBuilder, path))
				.bodyValue(body == null ? "" : body)
				.retrieve()
				.bodyToMono(Void.class).block();
	}

	public void put(String path) {
		put(path, null);
	}
	
	public void put(String path, Object body, Map queryParams) {
		MultiValueMap params = createStringParams(queryParams);
		webClient.put()
				.uri(uriBuilder -> uri(uriBuilder, path, params))
				.bodyValue(body == null ? "" : body)
				.retrieve()
				.bodyToMono(Void.class).block();
	}

	public void delete(String path, Map queryParams) {
		MultiValueMap params = createStringParams(queryParams);
		webClient.delete()
				.uri(uriBuilder -> uri(uriBuilder, path, params))
				.retrieve()
				.bodyToMono(Void.class).block();
	}

	private MultiValueMap createListParams(Map> queryParams) {
		return new LinkedMultiValueMap<>(queryParams);
	}

	private MultiValueMap createStringParams(Map queryParams) {
		Map> mutatedQueryParams = queryParams.entrySet().stream()
				.collect(toMap(Map.Entry::getKey, entry -> List.of(entry.getValue())));
		return new LinkedMultiValueMap<>(mutatedQueryParams);
	}

	private URI uri(final UriBuilder uriBuilder, String path) {
		return uri(uriBuilder, path, null);
	}

	private URI uri(final UriBuilder uriBuilder, String path, MultiValueMap queryParams) {
		return UriComponentsBuilder
				.fromUri(uriBuilder.build())
				.path(path)
				.queryParams(queryParams)
				.build(true)
				.toUri();
	}


}