org.asyncflows.protocol.http.common.content.ContentLengthOutput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of asyncflows-protocol-http Show documentation
This component contains implementation of HTTP 1.1 protocol.
There is a newer version: 0.1.1
Show newest version
/*
 * Copyright (c) 2018 Konstantin Plotnikov
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package org.asyncflows.protocol.http.common.content;

import org.asyncflows.io.util.ByteGeneratorContext;
import org.asyncflows.protocol.http.HttpException;
import org.asyncflows.core.Promise;
import org.asyncflows.core.function.AResolver;

import java.nio.ByteBuffer;

import static org.asyncflows.core.CoreFlows.aVoid;

/**
 * The output for the content length.
 */
public class ContentLengthOutput extends MessageOutput {
    /**
     * The content length.
     */
    private final long length;
    /**
     * The generated amount.
     */
    private long generated;

    /**
     * The constructor.
     *
     * @param output       the output
     * @param stateTracker the state tracker
     * @param length       the content length
     */
    protected ContentLengthOutput(final ByteGeneratorContext output, final AResolver stateTracker,
                                  final long length) {
        super(output, stateTracker);
        this.length = length;
    }

    @Override
    public Promise write(final ByteBuffer buffer) {
        return writes.run(() -> {
            ensureValidAndOpen();
            if (!buffer.hasRemaining()) {
                return aVoid();
            }
            if (generated == 0) {
                stateChanged(OutputState.DATA);
            }
            generated += buffer.remaining();
            if (generated > length) {
                throw new HttpException("Writing too much for Content-Length: " + generated + " > " + length);
            }
            if (output.isSendNeeded()) {
                return output.send().thenFlatGet(
                        () -> output.getOutput().write(buffer).thenFlatGet(() -> {
                            if (generated == length) {
                                stateChanged(OutputState.DATA_CLOSEABLE);
                            }
                            return aVoid();
                        }));
            } else {
                return output.getOutput().write(buffer);
            }
        }).listen(outcomeChecker());
    }

    @Override
    protected Promise closeAction() {
        stateChanged(generated == length ? OutputState.CLOSED : OutputState.CLOSED_LAST);
        return super.closeAction();
    }
}