io.edurt.datacap.pinot.com.ning.http.client.providers.grizzly.GrizzlyResponseFuture Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of datacap-shaded-pinot Show documentation
DataCap - Shaded for Apache Pinot
There is a newer version: 2024.03.7
/*
 * Copyright (c) 2012-2015 Sonatype, Inc. All rights reserved.
 *
 * This program is licensed to you under the Apache License Version 2.0,
 * and you may not use this file except in compliance with the Apache License Version 2.0.
 * You may obtain a copy of the Apache License Version 2.0 at http://www.apache.org/licenses/LICENSE-2.0.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the Apache License Version 2.0 is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Apache License Version 2.0 for the specific language governing permissions and limitations there under.
 */

package io.edurt.datacap.pinot.com.ning.http.client.providers.grizzly;

import io.edurt.datacap.pinot.com.ning.http.client.AsyncHandler;
import io.edurt.datacap.pinot.com.ning.http.client.listenable.AbstractListenableFuture;

import java.util.concurrent.CancellationException;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import org.glassfish.grizzly.CompletionHandler;
import org.glassfish.grizzly.impl.FutureImpl;
import org.glassfish.grizzly.utils.Futures;

/**
 * {@link AbstractListenableFuture} implementation adaptation of Grizzly's
 * {@link FutureImpl}.
 *
 * @author The Grizzly Team
 * @since 1.7.0
 */
final class GrizzlyResponseFuture extends AbstractListenableFuture
        implements CompletionHandler {

    private final FutureImpl delegate;
//    private final GrizzlyAsyncHttpProvider provider;
//    private Request request;
//    private Connection connection;
    private AsyncHandler asyncHandler;
    
    // transaction context. Not null if connection is established
    private volatile HttpTransactionContext transactionCtx;


    // ------------------------------------------------------------ Constructors


    GrizzlyResponseFuture(final AsyncHandler asyncHandler) {
        this.asyncHandler = asyncHandler;
        
        delegate = Futures.createSafeFuture();
        delegate.addCompletionHandler(this);
    }


    // ----------------------------------- Methods from AbstractListenableFuture


    public void done() {
        done(null);
    }

    public void done(V result) {
        delegate.result(result);
    }

    public void abort(Throwable t) {

        delegate.failure(t);

    }

    public void touch() {
        final HttpTransactionContext tx = transactionCtx;
        if (tx != null) {
            tx.touchConnection();
        }

    }


    public boolean getAndSetWriteHeaders(boolean writeHeaders) {

        // TODO This doesn't currently do anything - and may not make sense
        // with our implementation.  Needs further analysis.
        return writeHeaders;

    }


    public boolean getAndSetWriteBody(boolean writeBody) {

        // TODO This doesn't currently do anything - and may not make sense
        // with our implementation.  Needs further analysis.
        return writeBody;

    }


    // ----------------------------------------------------- Methods from Future


    public boolean cancel(boolean mayInterruptIfRunning) {
        return delegate.cancel(mayInterruptIfRunning);
    }


    public boolean isCancelled() {

        return delegate.isCancelled();

    }


    public boolean isDone() {

        return delegate.isDone();

    }


    public V get() throws InterruptedException, ExecutionException {

        return delegate.get();

    }


    public V get(long timeout, TimeUnit unit)
            throws InterruptedException, ExecutionException, TimeoutException {

        return delegate.get(timeout, unit);

    }

    // ----------------------------------------------------- Methods from CompletionHandler

    @Override
    public void cancelled() {
        final AsyncHandler ah = asyncHandler;
        if (ah != null) {
            try {
                ah.onThrowable(new CancellationException());
            } catch (Throwable ignore) {
            }
        }
        
        runListeners();
    }

    @Override
    public void failed(final Throwable t) {
        final AsyncHandler ah = asyncHandler;
        if (ah != null) {
            try {
                ah.onThrowable(t);
            } catch (Throwable ignore) {
            }
        }
            
        final HttpTransactionContext tx = transactionCtx;
        if (tx != null) {
            tx.closeConnection();
        }

        runListeners();
    }

    @Override
    public void completed(V result) {
        runListeners();
    }

    @Override
    public void updated(V result) {
    }        

    // ------------------------------------------------- Package Private Methods

    AsyncHandler getAsyncHandler() {
        return asyncHandler;
    }

    void setAsyncHandler(final AsyncHandler asyncHandler) {
        this.asyncHandler = asyncHandler;
    }

    /**
     * @return {@link HttpTransactionContext}, or null if connection is
     *          not established
     */
    HttpTransactionContext getHttpTransactionCtx() {
        return transactionCtx;
    }

    /**
     * @param transactionCtx
     * @return true if we can continue request/response processing,
     *          or false if future has been aborted
     */
    boolean setHttpTransactionCtx(
            final HttpTransactionContext transactionCtx) {
        this.transactionCtx = transactionCtx;
        return !delegate.isDone();
    }
}