com.dtsx.astra.sdk.streaming.TenantStatsClient Maven / Gradle / Ivy
package com.dtsx.astra.sdk.streaming;
import com.dtsx.astra.sdk.AbstractApiClient;
import com.dtsx.astra.sdk.utils.ApiLocator;
import com.dtsx.astra.sdk.utils.Assert;
import com.dtsx.astra.sdk.utils.AstraEnvironment;
import com.dtsx.astra.sdk.utils.HttpClientWrapper;
import com.dtsx.astra.sdk.streaming.domain.Statistics;
import com.dtsx.astra.sdk.streaming.domain.Tenant;
import com.dtsx.astra.sdk.utils.JsonUtils;
import com.fasterxml.jackson.core.type.TypeReference;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Stream;
/**
* Access metrics on a tenant.
*/
public class TenantStatsClient extends AbstractApiClient {
/** Load Database responses. */
private static final TypeReference> TYPE_LIST_STATISTICS =
new TypeReference>(){};
/**
* Unique db identifier.
*/
private final Tenant tenant;
/**
* As immutable object use builder to initiate the object.
*
* @param tenantId
* unique tenant identifier
* @param token
* authenticated token
*/
public TenantStatsClient(String token, String tenantId) {
this(token, AstraEnvironment.PROD, tenantId);
}
/**
* As immutable object use builder to initiate the object.
*
* @param env
* define target environment to be used
* @param token
* authenticated token
* @param tenantId
* unique tenant identifier
*/
public TenantStatsClient(String token, AstraEnvironment env, String tenantId) {
super(token, env);
Assert.hasLength(tenantId, "tenantId");
this.tenant = new AstraStreamingClient(token, env).get(tenantId);
}
/**
* Retrieve Statistics for all namespaces.
*
* @return
* statistics
*/
public Stream namespaces() {
return JsonUtils
.unmarshallType(
HttpClientWrapper.getInstance().GET_PULSAR(getEndpointStatisticsNamespaces(),
tenant.getPulsarToken(), tenant.getClusterName(),
tenant.getOrganizationId().toString()).getBody(), TYPE_LIST_STATISTICS)
.values()
.stream();
}
/**
* Retrieve Statistics for one namespace.
*
* @param namespace
* current pulsar namespace
* @return
* statistics
*/
public Optional namespace(String namespace) {
Map map = JsonUtils
.unmarshallType(
HttpClientWrapper
.getInstance()
.GET_PULSAR(
getEndpointStatisticsNamespaces() + "/" + namespace,
tenant.getPulsarToken(), tenant.getClusterName(),
tenant.getOrganizationId().toString())
.getBody(), TYPE_LIST_STATISTICS);
return Optional.ofNullable(map.get(tenant.getTenantName() + "/" + namespace));
}
/**
* Retrieve Statistics for all topics.
*
* @return
* statistics
*/
public Stream topics() {
return JsonUtils
.unmarshallType(
HttpClientWrapper.getInstance().GET_PULSAR(getEndpointStatisticsTopics(),
tenant.getPulsarToken(), tenant.getClusterName(),
tenant.getOrganizationId().toString()).getBody(), TYPE_LIST_STATISTICS)
.values()
.stream();
}
/**
* Retrieve Statistics for topics of a namespace
*
* @param namespace
* current pulsar namespace
* @return
* statistics
*/
public Stream topics(String namespace) {
return JsonUtils
.unmarshallType(
HttpClientWrapper.getInstance().GET_PULSAR(getEndpointStatisticsTopics() + "/" + namespace,
tenant.getPulsarToken(), tenant.getClusterName(),
tenant.getOrganizationId().toString()).getBody(), TYPE_LIST_STATISTICS)
.values()
.stream();
}
/**
* Endpoint for admin
*
* @return
* database endpoint
*/
public String getEndpointStreamingAdminV2() {
return ApiLocator.getApiStreamingV2Endpoint(environment, tenant.getClusterName());
}
/**
* Endpoint for namespace statistics.
*
* @return
* endpoint
*/
public String getEndpointStatisticsNamespaces() {
return getEndpointStreamingAdminV2() + "/stats/namespaces/" + tenant.getTenantName();
}
/**
* Endpoint for topics statistics.
*
* @return
* endpoint
*/
public String getEndpointStatisticsTopics() {
return getEndpointStreamingAdminV2() + "/stats/topics/" + tenant.getTenantName();
}
}