Examples with CreateDelegationTokenResponseData org.apache.kafka.common.message.CreateDelegationTokenResponseData used on opensource projects

Example 1 with CreateDelegationTokenResponseData

use of org.apache.kafka.common.message.CreateDelegationTokenResponseData in project kafka by apache.

the class KafkaAdminClient method createDelegationToken.

@Override
public CreateDelegationTokenResult createDelegationToken(final CreateDelegationTokenOptions options) {
    final KafkaFutureImpl<DelegationToken> delegationTokenFuture = new KafkaFutureImpl<>();
    final long now = time.milliseconds();
    List<CreatableRenewers> renewers = new ArrayList<>();
    for (KafkaPrincipal principal : options.renewers()) {
        renewers.add(new CreatableRenewers().setPrincipalName(principal.getName()).setPrincipalType(principal.getPrincipalType()));
    }
    runnable.call(new Call("createDelegationToken", calcDeadlineMs(now, options.timeoutMs()), new LeastLoadedNodeProvider()) {

        @Override
        CreateDelegationTokenRequest.Builder createRequest(int timeoutMs) {
            return new CreateDelegationTokenRequest.Builder(new CreateDelegationTokenRequestData().setRenewers(renewers).setMaxLifetimeMs(options.maxlifeTimeMs()));
        }

        @Override
        void handleResponse(AbstractResponse abstractResponse) {
            CreateDelegationTokenResponse response = (CreateDelegationTokenResponse) abstractResponse;
            if (response.hasError()) {
                delegationTokenFuture.completeExceptionally(response.error().exception());
            } else {
                CreateDelegationTokenResponseData data = response.data();
                TokenInformation tokenInfo = new TokenInformation(data.tokenId(), new KafkaPrincipal(data.principalType(), data.principalName()), options.renewers(), data.issueTimestampMs(), data.maxTimestampMs(), data.expiryTimestampMs());
                DelegationToken token = new DelegationToken(tokenInfo, data.hmac());
                delegationTokenFuture.complete(token);
            }
        }

        @Override
        void handleFailure(Throwable throwable) {
            delegationTokenFuture.completeExceptionally(throwable);
        }
    }, now);
    return new CreateDelegationTokenResult(delegationTokenFuture);
}