view src/goodjava/webserver/ResponseOutputStream.java @ 1610:94ea190714dd

minor
author Franklin Schmidt <fschmidt@gmail.com>
date Thu, 13 May 2021 17:26:34 -0600
parents 27efb1fcbcb5
children
line wrap: on
line source

package goodjava.webserver;

import java.io.ByteArrayOutputStream;
import java.io.ByteArrayInputStream;
import java.io.IOException;


// plenty of room for improvement
public class ResponseOutputStream extends ByteArrayOutputStream {
	private final Response response;

	public ResponseOutputStream(Response response) {
		if(response==null) throw new NullPointerException();
		this.response = response;
	}

	@Override public void close() throws IOException {
		super.close();
		int size = size();
		response.body = new Response.Body( size, new ByteArrayInputStream(buf,0,size) );
	}
}