com.hibegin.http.io.ChunkedOutputStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of simplewebserver Show documentation

Simple, flexible, less dependent, more extended. Less memory footprint, can quickly build Web project. Can quickly run embedded, Android devices

There is a newer version: 0.3.162

Show newest version

package com.hibegin.http.io;

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

public class ChunkedOutputStream extends FilterOutputStream {
    static final byte[] CRLF = "\r\n".getBytes();
    static final byte[] LAST_TOKEN = "0\r\n\r\n".getBytes();
    boolean eos = false;

    public ChunkedOutputStream(OutputStream os) {
        super(os);
    }

    @Override
    public void write(int b) throws IOException {
        write(new byte[]{(byte) b}, 0, 1);
    }

    @Override
    public void write(byte[] b) throws IOException {
        write(b, 0, b.length);
    }

    @Override
    public void write(byte[] b, int off, int len) throws IOException {
        if (len == 0) {
            return;
        }
        out.write((Integer.toHexString(len)).getBytes());
        out.write(CRLF);
        out.write(b, off, len);
        out.write(CRLF);
    }

    private void eos() throws IOException {
        synchronized (this) {
            if (eos) {
                return;
            }
            eos = true;
        }
        out.write(LAST_TOKEN);
        out.flush();
    }

    @Override
    public void close() throws IOException {
        eos();
        out.close();
    }
}