com.qubole.qds.sdk.java.details.InvocationCallbackBase Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of qds-sdk-java Show documentation

A Java library that provides the tools you need to authenticate with, and use the Qubole Data Service API.

There is a newer version: 1.3.0

/**
 * Copyright 2014- Qubole Inc.
 *
 * Licensed 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 com.qubole.qds.sdk.java.details;

import com.google.common.base.Preconditions;
import com.qubole.qds.sdk.java.api.InvokableBuilder;
import com.qubole.qds.sdk.java.api.RawInvokableBuilder;
import javax.ws.rs.client.InvocationCallback;
import javax.ws.rs.core.Response;
import java.util.concurrent.Future;

abstract class InvocationCallbackBase implements InvokableBuilder
{
    private InvocationCallback callback;

    public InvokableBuilder withCallback(InvocationCallback callback)
    {
        this.callback = callback;
        return this;
    }

    @Override
    public final RawInvokableBuilder raw()
    {
        return new RawInvokableBuilder()
        {
            @Override
            public Future invoke()
            {
                Preconditions.checkState(callback == null, "withCallback() cannot be used in combination with invokeRaw()");

                InvokeArguments invokeArguments = getInvokeArguments();
                return invokeArguments.getClient().invokeRequest(invokeArguments.getForPage(), invokeArguments.getEntity(), Response.class, invokeArguments.getAdditionalPaths());
            }
        };
    }

    @Override
    public final Future invoke()
    {
        InvokeArguments invokeArguments = getInvokeArguments();

        if ( callback != null )
        {
            return invokeArguments.getClient().invokeRequest(invokeArguments.getForPage(), invokeArguments.getEntity(), callback, invokeArguments.getAdditionalPaths());
        }

        if ( invokeArguments.getGenericResponseType() != null )
        {
            return invokeArguments.getClient().invokeRequest(invokeArguments.getForPage(), invokeArguments.getEntity(), invokeArguments.getGenericResponseType(), invokeArguments.getAdditionalPaths());
        }

        return invokeArguments.getClient().invokeRequest(invokeArguments.getForPage(), invokeArguments.getEntity(), invokeArguments.getResponseType(), invokeArguments.getAdditionalPaths());
    }

    protected abstract InvokeArguments getInvokeArguments();
}