com.clevercloud.biscuitpulsar.AuthenticationDataBiscuit Maven / Gradle / Ivy

Go to download
package com.clevercloud.biscuitpulsar;

import org.apache.pulsar.client.api.AuthenticationDataProvider;

import java.util.Collections;
import java.util.Map;
import java.util.Set;
import java.util.function.Supplier;

public class AuthenticationDataBiscuit implements AuthenticationDataProvider {
    public final static String HTTP_HEADER_NAME = "Authorization";

    private final Supplier biscuitSupplier;

    public AuthenticationDataBiscuit(Supplier biscuitSupplier) {
        this.biscuitSupplier = biscuitSupplier;
    }

    @Override
    public boolean hasDataForHttp() {
        return true;
    }

    @Override
    public Set> getHttpHeaders() {
        return Collections.singletonMap(HTTP_HEADER_NAME, "Bearer " + getBiscuit()).entrySet();
    }

    @Override
    public boolean hasDataFromCommand() {
        return true;
    }

    @Override
    public String getCommandData() {
        return getBiscuit();
    }

    private String getBiscuit() {
        try {
            return biscuitSupplier.get();
        } catch (Throwable t) {
            throw new RuntimeException("failed to get client biscuit", t);
        }
    }
}