io.avaje.json.stream.NimaJsonOutput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of avaje-json Show documentation

There is a newer version: 3.0-RC2

package io.avaje.json.stream;

import java.io.IOException;
import java.io.OutputStream;

import io.helidon.http.HeaderNames;
import io.helidon.webserver.http.ServerResponse;

final class NimaJsonOutput implements JsonOutput {

  private static final String CHUNKED = "chunked";

  private final ServerResponse res;
  private OutputStream os;
  private boolean fixedLength;

  NimaJsonOutput(ServerResponse res) {
    this.res = res;
  }

  private OutputStream os() {
    if (os == null) {
      if (!fixedLength) {
        // going to use chunked content
        res.header(HeaderNames.TRANSFER_ENCODING, CHUNKED);
      }
      os = res.outputStream();
    }
    return os;
  }

  @Override
  public void write(byte[] content, int offset, int length) throws IOException {
    os().write(content, offset, length);
  }

  @Override
  public void writeLast(byte[] content, int offset, int length) throws IOException {
    if (offset == 0 && os == null) {
      // going to use fixed length content
      fixedLength = true;
      res.contentLength(length);
    }
    os().write(content, offset, length);
  }

  @Override
  public void flush() throws IOException {
    if (os != null) {
      os.flush();
    }
  }

  @Override
  public void close() throws IOException {
    if (os != null) {
      os.close();
    }
  }

  @Override
  public OutputStream unwrapOutputStream() {
    return res.outputStream();
  }
}