org.asynchttpclient.request.body.generator.PushBody 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: 3.0.0

/*
 * Copyright (c) 2015 AsyncHttpClient Project. 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 org.asynchttpclient.request.body.generator;

import io.netty.buffer.ByteBuf;
import org.asynchttpclient.request.body.Body;

import java.util.Queue;

public final class PushBody implements Body {

  private final Queue queue;
  private BodyState state = BodyState.CONTINUE;

  public PushBody(Queue queue) {
    this.queue = queue;
  }

  @Override
  public long getContentLength() {
    return -1;
  }

  @Override
  public BodyState transferTo(final ByteBuf target) {
    switch (state) {
      case CONTINUE:
        return readNextChunk(target);
      case STOP:
        return BodyState.STOP;
      default:
        throw new IllegalStateException("Illegal process state.");
    }
  }

  private BodyState readNextChunk(ByteBuf target) {
    BodyState res = BodyState.SUSPEND;
    while (target.isWritable() && state != BodyState.STOP) {
      BodyChunk nextChunk = queue.peek();
      if (nextChunk == null) {
        // Nothing in the queue. suspend stream if nothing was read. (reads == 0)
        return res;
      } else if (!nextChunk.buffer.isReadable() && !nextChunk.last) {
        // skip empty buffers
        queue.remove();
      } else {
        res = BodyState.CONTINUE;
        readChunk(target, nextChunk);
      }
    }
    return res;
  }

  private void readChunk(ByteBuf target, BodyChunk part) {
    target.writeBytes(part.buffer);
    if (!part.buffer.isReadable()) {
      if (part.last) {
        state = BodyState.STOP;
      }
      queue.remove();
    }
  }

  @Override
  public void close() {
  }
}