com.dtsx.astra.sdk.db.domain.telemetry.SpecializedTelemetryClient Maven / Gradle / Ivy
package com.dtsx.astra.sdk.db.domain.telemetry;
import com.dtsx.astra.sdk.utils.ApiResponseHttp;
import com.dtsx.astra.sdk.utils.Assert;
import com.dtsx.astra.sdk.utils.HttpClientWrapper;
import com.dtsx.astra.sdk.utils.JsonUtils;
import com.fasterxml.jackson.core.type.TypeReference;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.net.HttpURLConnection;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
/**
* Kafka Client.
*
* @param
* type of data
*/
public class SpecializedTelemetryClient {
/** Logger for our Client. */
private static final Logger LOGGER = LoggerFactory.getLogger(SpecializedTelemetryClient.class);
/** Load Database responses. */
private final TypeReference> RESPONSE = new TypeReference>(){};
/** unique db identifier. */
private final String token;
/** unique db identifier. */
private final String key;
/** Reference to upper resource. */
private final String telemetryEndpoint;
/**
* Default constructor.
*
* @param token
* token client
* @param telemetryEndpoint
* endpoint
* @param key
* key for target system
*/
public SpecializedTelemetryClient(String token, String telemetryEndpoint, String key) {
Assert.notNull(token,"databasesClient");
Assert.hasLength(telemetryEndpoint, "telemetryEndpoint");
this.token = token;
this.key = key;
this.telemetryEndpoint = telemetryEndpoint;
}
/**
* Configure Astra Remote Telemetry.
*
* @param ktr
* config request
* @return
* http response
*/
public ApiResponseHttp setup(T ktr) {
Map bodyMap = new HashMap<>();
bodyMap.put(key, ktr);
return HttpClientWrapper.getInstance().POST(telemetryEndpoint, token, JsonUtils.mapAsJson(bodyMap));
}
/**
* Retrieve Remote Telemetry configuration
* https://docs.datastax.com/en/astra/docs/_attachments/devopsv2.html#operation/getTelemetryConfig
* @return
* telemetry request
*/
public Optional find() {
ApiResponseHttp res = HttpClientWrapper.getInstance().GET(telemetryEndpoint, token);
try{
if (res.getCode() == HttpURLConnection.HTTP_OK) {
return Optional.ofNullable(JsonUtils
.unmarshallType(res.getBody(), RESPONSE).get(key));
}
} catch(Exception e) {
LOGGER.warn("Cannot read telemetry configuration for " + key, e);
}
return Optional.empty();
}
}