Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.pulsar.client.admin.internal.BaseResource Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.pulsar.client.admin.internal;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.function.Function;
import java.util.function.Supplier;
import org.apache.pulsar.shade.javax.ws.rs.ClientErrorException;
import org.apache.pulsar.shade.javax.ws.rs.ServerErrorException;
import org.apache.pulsar.shade.javax.ws.rs.ServiceUnavailableException;
import org.apache.pulsar.shade.javax.ws.rs.WebApplicationException;
import org.apache.pulsar.shade.javax.ws.rs.client.Entity;
import org.apache.pulsar.shade.javax.ws.rs.client.Invocation.Builder;
import org.apache.pulsar.shade.javax.ws.rs.client.InvocationCallback;
import org.apache.pulsar.shade.javax.ws.rs.client.WebTarget;
import org.apache.pulsar.shade.javax.ws.rs.core.GenericType;
import org.apache.pulsar.shade.javax.ws.rs.core.MediaType;
import org.apache.pulsar.shade.javax.ws.rs.core.Response;
import org.apache.pulsar.client.admin.PulsarAdminException;
import org.apache.pulsar.client.admin.PulsarAdminException.ConflictException;
import org.apache.pulsar.client.admin.PulsarAdminException.ConnectException;
import org.apache.pulsar.client.admin.PulsarAdminException.GettingAuthenticationDataException;
import org.apache.pulsar.client.admin.PulsarAdminException.NotAllowedException;
import org.apache.pulsar.client.admin.PulsarAdminException.NotAuthorizedException;
import org.apache.pulsar.client.admin.PulsarAdminException.NotFoundException;
import org.apache.pulsar.client.admin.PulsarAdminException.PreconditionFailedException;
import org.apache.pulsar.client.admin.PulsarAdminException.ServerSideErrorException;
import org.apache.pulsar.client.api.Authentication;
import org.apache.pulsar.client.api.AuthenticationDataProvider;
import org.apache.pulsar.client.api.PulsarClientException;
import org.apache.pulsar.common.policies.data.ErrorData;
import org.apache.pulsar.common.util.ObjectMapperFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Abstract base class for all admin resources.
*/
public abstract class BaseResource {
private static final Logger log = LoggerFactory.getLogger(BaseResource.class);
protected final Authentication auth;
protected final long readTimeoutMs;
protected BaseResource(Authentication auth, long readTimeoutMs) {
this.auth = auth;
this.readTimeoutMs = readTimeoutMs;
}
public Builder request(final WebTarget target) throws PulsarAdminException {
try {
return requestAsync(target).get();
} catch (Exception e) {
throw new GettingAuthenticationDataException(e);
}
}
// do the authentication stage, and once authentication completed return a Builder
public CompletableFuture requestAsync(final WebTarget target) {
CompletableFuture builderFuture = new CompletableFuture<>();
CompletableFuture> authFuture = new CompletableFuture<>();
try {
AuthenticationDataProvider authData = auth.getAuthData(target.getUri().getHost());
if (authData.hasDataForHttp()) {
auth.authenticationStage(target.getUri().toString(), authData, null, authFuture);
} else {
authFuture.complete(null);
}
// auth complete, return a new Builder
authFuture.whenComplete((respHeaders, ex) -> {
if (ex != null) {
log.warn("[{}] Failed to perform http request at auth stage: {}", target.getUri(),
ex.getMessage());
builderFuture.completeExceptionally(new PulsarClientException(ex));
return;
}
try {
Builder builder = target.request(MediaType.APPLICATION_JSON);
if (authData.hasDataForHttp()) {
Set> headers =
auth.newRequestHeader(target.getUri().toString(), authData, respHeaders);
if (headers != null) {
headers.forEach(entry -> builder.header(entry.getKey(), entry.getValue()));
}
}
builderFuture.complete(builder);
} catch (Throwable t) {
builderFuture.completeExceptionally(new GettingAuthenticationDataException(t));
}
});
} catch (Throwable t) {
builderFuture.completeExceptionally(new GettingAuthenticationDataException(t));
}
return builderFuture;
}
public CompletableFuture asyncPutRequest(final WebTarget target, Entity entity) {
final CompletableFuture future = new CompletableFuture<>();
try {
request(target).async().put(entity, new InvocationCallback() {
@Override
public void completed(ErrorData response) {
future.complete(null);
}
@Override
public void failed(Throwable throwable) {
log.warn("[{}] Failed to perform http put request: {}", target.getUri(), throwable.getMessage());
future.completeExceptionally(getApiException(throwable.getCause()));
}
});
} catch (PulsarAdminException cae) {
future.completeExceptionally(cae);
}
return future;
}
public void asyncPostRequestWithResponse(final WebTarget target, Entity entity,
InvocationCallback callback) {
try {
request(target).async().post(entity, callback);
} catch (PulsarAdminException cae) {
callback.failed(cae);
}
}
public CompletableFuture asyncPostRequest(final WebTarget target, Entity entity) {
final CompletableFuture future = new CompletableFuture<>();
try {
request(target).async().post(entity, new InvocationCallback() {
@Override
public void completed(ErrorData response) {
future.complete(null);
}
@Override
public void failed(Throwable throwable) {
log.warn("[{}] Failed to perform http post request: {}", target.getUri(), throwable.getMessage());
future.completeExceptionally(getApiException(throwable.getCause()));
}
});
} catch (PulsarAdminException cae) {
future.completeExceptionally(cae);
}
return future;
}
public void asyncGetRequest(final WebTarget target, InvocationCallback callback) {
try {
request(target).async().get(callback);
} catch (PulsarAdminException cae) {
callback.failed(cae);
}
}
public CompletableFuture asyncGetRequest(final WebTarget target, FutureCallback callback) {
asyncGetRequest(target, (InvocationCallback) callback);
return callback.future();
}
protected CompletableFuture asyncGetRequest(final WebTarget target, Class type) {
return asyncGetRequest(target, response -> response.readEntity(type));
}
protected CompletableFuture asyncGetRequest(final WebTarget target, GenericType type) {
return asyncGetRequest(target, response -> response.readEntity(type));
}
private CompletableFuture asyncGetRequest(final WebTarget target, Function readResponse) {
final CompletableFuture future = new CompletableFuture<>();
asyncGetRequest(target,
new InvocationCallback() {
@Override
public void completed(Response response) {
if (response.getStatus() != Response.Status.OK.getStatusCode()) {
future.completeExceptionally(getApiException(response));
} else {
try {
future.complete(readResponse.apply(response));
} catch (Exception e) {
future.completeExceptionally(getApiException(e));
}
}
}
@Override
public void failed(Throwable throwable) {
future.completeExceptionally(getApiException(throwable.getCause()));
}
});
return future;
}
public CompletableFuture asyncDeleteRequest(final WebTarget target) {
final CompletableFuture future = new CompletableFuture<>();
try {
request(target).async().delete(new InvocationCallback() {
@Override
public void completed(ErrorData response) {
future.complete(null);
}
@Override
public void failed(Throwable throwable) {
log.warn("[{}] Failed to perform http delete request: {}", target.getUri(), throwable.getMessage());
future.completeExceptionally(getApiException(throwable.getCause()));
}
});
} catch (PulsarAdminException cae) {
future.completeExceptionally(cae);
}
return future;
}
public void asyncDeleteRequest(final WebTarget target, InvocationCallback callback) {
try {
request(target).async().delete(callback);
} catch (PulsarAdminException cae) {
callback.failed(cae);
}
}
public static PulsarAdminException getApiException(Throwable e) {
if (e instanceof PulsarAdminException) {
return (PulsarAdminException) e;
} else if (e instanceof ServiceUnavailableException) {
if (e.getCause() instanceof java.net.ConnectException) {
return new ConnectException(e.getCause());
} else {
ServerErrorException see = (ServerErrorException) e;
int statusCode = see.getResponse().getStatus();
String httpError = getReasonFromServer(see);
return new PulsarAdminException(e, httpError, statusCode);
}
} else if (e instanceof WebApplicationException) {
// Handle 5xx exceptions
if (e instanceof ServerErrorException) {
ServerErrorException see = (ServerErrorException) e;
int statusCode = see.getResponse().getStatus();
String httpError = getReasonFromServer(see);
return new ServerSideErrorException(see, httpError, httpError, statusCode);
} else if (e instanceof ClientErrorException) {
// Handle 4xx exceptions
ClientErrorException cee = (ClientErrorException) e;
int statusCode = cee.getResponse().getStatus();
String httpError = getReasonFromServer(cee);
switch (statusCode) {
case 401:
case 403:
return new NotAuthorizedException(cee, httpError, statusCode);
case 404:
return new NotFoundException(cee, httpError, statusCode);
case 405:
return new NotAllowedException(cee, httpError, statusCode);
case 409:
return new ConflictException(cee, httpError, statusCode);
case 412:
return new PreconditionFailedException(cee, httpError, statusCode);
default:
return new PulsarAdminException(httpError, cee, httpError, statusCode);
}
} else {
WebApplicationException wae = (WebApplicationException) e;
int statusCode = wae.getResponse().getStatus();
String httpError = getReasonFromServer(wae);
return new PulsarAdminException(httpError, wae, httpError, statusCode);
}
} else {
return new PulsarAdminException(e);
}
}
public PulsarAdminException getApiException(Response response) {
if (response.getStatus() == Response.Status.OK.getStatusCode()) {
return null;
}
try {
if (response.getStatus() >= 500) {
throw new ServerErrorException(response);
} else if (response.getStatus() >= 400) {
throw new ClientErrorException(response);
} else {
throw new WebApplicationException(response);
}
} catch (Exception e) {
return getApiException(e);
}
}
public static String getReasonFromServer(WebApplicationException e) {
try {
return e.getResponse().readEntity(ErrorData.class).reason.toString();
} catch (Exception ex) {
try {
return ObjectMapperFactory.getMapper().reader().readValue(
e.getResponse().getEntity().toString(), ErrorData.class).reason;
} catch (Exception ex1) {
try {
return ObjectMapperFactory.getMapper().reader()
.readValue(e.getMessage(), ErrorData.class).reason;
} catch (Exception ex2) {
// could not parse output to ErrorData class
return e.getMessage();
}
}
}
}
protected T sync(Supplier> executor) throws PulsarAdminException {
try {
return executor.get().get(this.readTimeoutMs, TimeUnit.MILLISECONDS);
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
throw new PulsarAdminException(e);
} catch (TimeoutException e) {
throw new PulsarAdminException.TimeoutException(e);
} catch (ExecutionException e) {
// we want to have a stacktrace that points to this point, in order to return a meaningful
// stacktrace to the user, otherwise we will have a stacktrace
// related to another thread, because all Admin API calls are async
throw PulsarAdminException.wrap(getApiException(e.getCause()));
} catch (Exception e) {
throw PulsarAdminException.wrap(getApiException(e));
}
}
/**
* InvocationCallback that creates a CompletableFuture and completes it based on the response.
* Must be subclassed to provide runtime type information to the ReST client library.
* @param type to which the response body is parsed in case of success
*/
abstract static class FutureCallback implements InvocationCallback {
private final CompletableFuture future = new CompletableFuture<>();
@Override
public void completed(T value) {
future.complete(value);
}
@Override
public void failed(Throwable throwable) {
future.completeExceptionally(getApiException(throwable.getCause()));
}
public CompletableFuture future() {
return future;
}
}
}
