org.glassfish.grizzly.http2.utils.ChunkedCompletionHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grizzly-http-all Show documentation
The newest version!
/*
 * Copyright (c) 2014, 2020 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package org.glassfish.grizzly.http2.utils;

import java.util.logging.Logger;

import org.glassfish.grizzly.CompletionHandler;
import org.glassfish.grizzly.WriteResult;

/**
 *
 * @author oleksiys
 */
public class ChunkedCompletionHandler implements CompletionHandler {
    private static final Logger LOG = Logger.getLogger(ChunkedCompletionHandler.class.getName());
    private final CompletionHandler parentCompletionHandler;

    private boolean isDone;
    private int chunksCounter = 1;
    private long writtenSize;

    /**
     * @param parentCompletionHandler - can be null
     */
    public ChunkedCompletionHandler(final CompletionHandler parentCompletionHandler) {
        this.parentCompletionHandler = parentCompletionHandler;
    }

    public void incChunks() {
        chunksCounter++;
    }

    @Override
    public void cancelled() {
        LOG.finest("cancelled()");
        if (done()) {
            if (parentCompletionHandler != null) {
                parentCompletionHandler.cancelled();
            }
        }
    }

    @Override
    public void failed(Throwable throwable) {
        // we don't need a stacktrace here, but we want to see why we are here.
        LOG.finest(() -> String.format("failed(throwable=%s)", throwable));
        if (done()) {
            if (parentCompletionHandler != null) {
                parentCompletionHandler.failed(throwable);
            }
        }
    }

    @Override
    public void completed(final WriteResult result) {
        LOG.finest(() -> String.format("completed(result=%s)", result));
        if (isDone) {
            return;
        }

        if (--chunksCounter == 0) {
            done();

            final long initialWrittenSize = result.getWrittenSize();
            writtenSize += initialWrittenSize;

            if (parentCompletionHandler != null) {
                try {
                    result.setWrittenSize(writtenSize);
                    parentCompletionHandler.completed(result);
                } finally {
                    result.setWrittenSize(initialWrittenSize);
                }
            }
        } else {
            updated(result);
            writtenSize += result.getWrittenSize();
        }
    }

    @Override
    public void updated(final WriteResult result) {
        LOG.finest(() -> String.format("updated(result=%s)", result));
        if (parentCompletionHandler != null) {
            final long initialWrittenSize = result.getWrittenSize();
            try {
                result.setWrittenSize(writtenSize + initialWrittenSize);
                parentCompletionHandler.updated(result);
            } finally {
                result.setWrittenSize(initialWrittenSize);
            }
        }
    }

    private boolean done() {
        if (isDone) {
            return false;
        }

        isDone = true;

        done0();
        return true;
    }


    /**
     * This method does nothing but can be overriden to implement some action executed before
     * the parent completition handler is executed.
     */
    protected void done0() {
    }
}