org.asynchttpclient.netty.request.body.BodyChunkedInput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of async-http-client Show documentation

The Async Http Client (AHC) classes.

There is a newer version: LATEST_VERSION

/*
 *    Copyright (c) 2014-2024 AsyncHttpClient Project. All rights reserved.
 *
 *    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 org.asynchttpclient.netty.request.body;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.stream.ChunkedInput;
import org.asynchttpclient.request.body.Body;

import static java.util.Objects.requireNonNull;

/**
 * Adapts a {@link Body} to Netty's {@link ChunkedInput}.
 */
public class BodyChunkedInput implements ChunkedInput {

    public static final int DEFAULT_CHUNK_SIZE = 8 * 1024;

    private final Body body;
    private final int chunkSize;
    private final long contentLength;
    private boolean endOfInput;
    private long progress;

    BodyChunkedInput(Body body) {
        this.body = requireNonNull(body, "body");
        contentLength = body.getContentLength();
        if (contentLength <= 0) {
            chunkSize = DEFAULT_CHUNK_SIZE;
        } else {
            chunkSize = (int) Math.min(contentLength, DEFAULT_CHUNK_SIZE);
        }
    }

    @Override
    @Deprecated
    public ByteBuf readChunk(ChannelHandlerContext ctx) throws Exception {
        return readChunk(ctx.alloc());
    }

    @Override
    public ByteBuf readChunk(ByteBufAllocator alloc) throws Exception {
        if (endOfInput) {
            return null;
        }

        ByteBuf buffer = alloc.buffer(chunkSize);
        Body.BodyState state = body.transferTo(buffer);
        progress += buffer.writerIndex();
        switch (state) {
            case STOP:
                endOfInput = true;
                return buffer;
            case SUSPEND:
                // this will suspend the stream in ChunkedWriteHandler
                buffer.release();
                return null;
            case CONTINUE:
                return buffer;
            default:
                throw new IllegalStateException("Unknown state: " + state);
        }
    }

    @Override
    public boolean isEndOfInput() {
        return endOfInput;
    }

    @Override
    public void close() throws Exception {
        body.close();
    }

    @Override
    public long length() {
        return contentLength;
    }

    @Override
    public long progress() {
        return progress;
    }
}