Class ApiMethod

fluss-rpc/src/main/java/org/apache/fluss/rpc/protocol/ApiMethod.java:33–122

Describes an RPC API method to be invoked.

/** Describes an RPC API method to be invoked. */
@ThreadSafe
public class ApiMethod {
    /** The API key of the RPC API. */
    private final ApiKeys apiKey;

    /** The class of the request message. */
    private final Class<?> requestClass;

    /** The class of the response message. */
    private final Class<?> responseClass;

    /** The constructor of the request message. */
    private final Supplier<ApiMessage> requestConstructor;

    /** The constructor of the response message. */
    private final Supplier<ApiMessage> responseConstructor;

    /** The RPC method to be invoked. */
    final Method method;

    /** The providers of the RPC API. */
    final List<ServerType> providers;

    ApiMethod(
            ApiKeys apiKey,
            Class<?> requestClass,
            Class<?> responseClass,
            Method method,
            ImmutableList<ServerType> providers) {
        this.apiKey = apiKey;
        this.requestClass = requestClass;
        this.responseClass = responseClass;
        this.requestConstructor = new ApiMessageConstructor(requestClass);
        this.responseConstructor = new ApiMessageConstructor(responseClass);
        this.method = method;
        this.providers = providers;
    }

    public boolean inScope(ServerType provider) {
        return providers.contains(provider);
    }

    public ApiMethod copyAndAddProvider(ServerType additional) {
        List<ServerType> tmp = new ArrayList<>();
        tmp.add(additional);
        tmp.addAll(providers);
        return new ApiMethod(
                apiKey, requestClass, responseClass, method, ImmutableList.copyOf(tmp));
    }

    public ApiKeys getApiKey() {
        return apiKey;
    }

    public Supplier<ApiMessage> getRequestConstructor() {
        return requestConstructor;
    }