com.hazelcast.cloud.maven.client.HazelcastCloudClient Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hazelcast-cloud-maven-plugin Show documentation

Maven plugin for the operations with Hazelcast Cloud

There is a newer version: 0.2.0

Show newest version

package com.hazelcast.cloud.maven.client;

import java.io.File;
import java.time.Duration;
import java.util.HashMap;

import lombok.var;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpMethod;
import org.springframework.http.client.reactive.ReactorClientHttpConnector;
import org.springframework.http.codec.ServerSentEvent;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Flux;
import reactor.netty.http.client.HttpClient;

import com.hazelcast.cloud.maven.model.Cluster;

import static com.hazelcast.cloud.maven.auth.AuthUtils.headersWithAuth;
import static org.springframework.http.HttpHeaders.AUTHORIZATION;
import static org.springframework.http.MediaType.MULTIPART_FORM_DATA;

public class HazelcastCloudClient {

    private final String apiBaseUrl;

    private final String token;

    private final RestTemplate restTemplate = new RestTemplate();

    public HazelcastCloudClient(String apiBaseUrl, String token) {
        this.apiBaseUrl = apiBaseUrl;
        this.token = token;
    }

    public Cluster getClusterStatus(String clusterId) {
        return restTemplate.exchange(
            url("/cluster/{clusterId}"),
            HttpMethod.GET,
            new HttpEntity<>(headersWithAuth(token)),
            Cluster.class,
            clusterId).getBody();
    }

    public Flux> getClusterLogs(String clusterId) {
        return WebClient.builder()
            .clientConnector(new ReactorClientHttpConnector(
                HttpClient
                    .create()
                    .responseTimeout(Duration.ofMinutes(5))))
            .baseUrl(String.format("%s/cluster/%s/logstream", apiBaseUrl, clusterId))
            .build().get()
            .header(AUTHORIZATION, "Bearer " + token)
            .retrieve()
            .bodyToFlux(new ParameterizedTypeReference>() {

            });
    }

    public void uploadCustomClasses(String clusterId, File file) {
        var headers = headersWithAuth(token);
        headers.setContentType(MULTIPART_FORM_DATA);

        var body = new LinkedMultiValueMap();
        body.add("customClassesFile", new FileSystemResource(file));

        var pathParams = new HashMap();
        pathParams.put("clusterId", clusterId);

        restTemplate.postForEntity(
            url("/cluster/{clusterId}/custom_classes"),
            new HttpEntity<>(body, headers),
            String.class,
            pathParams);
    }

    private String url(String uri) {
        return this.apiBaseUrl + uri;
    }

}