1137
|
1 package luan.webserver;
|
|
2
|
|
3 import java.io.ByteArrayOutputStream;
|
|
4 import java.io.ByteArrayInputStream;
|
|
5 import java.io.IOException;
|
|
6
|
|
7
|
|
8 // plenty of room for improvement
|
|
9 public class ResponseOutputStream extends ByteArrayOutputStream {
|
|
10 private final Response response;
|
|
11
|
|
12 public ResponseOutputStream(Response response) {
|
|
13 this.response = response;
|
|
14 }
|
|
15
|
|
16 @Override public void close() throws IOException {
|
|
17 super.close();
|
|
18 int size = size();
|
|
19 response.body = new Response.Body( size, new ByteArrayInputStream(buf,0,size) );
|
|
20 }
|
|
21 }
|