Mercurial Hosting > luan
annotate src/luan/webserver/Connection.java @ 1138:4189027691b7
webserver - add LogHandler
| author | Franklin Schmidt <fschmidt@gmail.com> |
|---|---|
| date | Mon, 29 Jan 2018 19:48:21 -0700 |
| parents | c123ee15f99b |
| children | 0f59eab45f3d |
| rev | line source |
|---|---|
| 1137 | 1 package luan.webserver; |
| 2 | |
| 3 import java.io.InputStream; | |
| 4 import java.io.OutputStream; | |
| 5 import java.io.IOException; | |
| 6 import java.net.Socket; | |
| 7 import org.slf4j.Logger; | |
| 8 import org.slf4j.LoggerFactory; | |
| 9 import luan.lib.parser.ParseException; | |
| 10 | |
| 11 | |
| 12 final class Connection { | |
| 13 private static final Logger logger = LoggerFactory.getLogger(Connection.class); | |
| 14 | |
| 15 private final Server server; | |
| 16 private final Socket socket; | |
| 17 | |
| 18 Connection(Server server,Socket socket) { | |
| 19 this.server = server; | |
| 20 this.socket = socket; | |
| 21 handle(); | |
| 22 } | |
| 23 | |
| 24 private void handle() { | |
| 25 try { | |
| 26 InputStream in = socket.getInputStream(); | |
| 27 byte[] a = new byte[8192]; | |
| 28 int endOfHeader; | |
| 29 int size = 0; | |
| 30 int left = a.length; | |
| 31 outer: while(true) { | |
| 32 int n = in.read(a,size,left); | |
|
1138
4189027691b7
webserver - add LogHandler
Franklin Schmidt <fschmidt@gmail.com>
parents:
1137
diff
changeset
|
33 if( n == -1 ) { |
|
4189027691b7
webserver - add LogHandler
Franklin Schmidt <fschmidt@gmail.com>
parents:
1137
diff
changeset
|
34 if( size == 0 ) { |
|
4189027691b7
webserver - add LogHandler
Franklin Schmidt <fschmidt@gmail.com>
parents:
1137
diff
changeset
|
35 socket.close(); |
|
4189027691b7
webserver - add LogHandler
Franklin Schmidt <fschmidt@gmail.com>
parents:
1137
diff
changeset
|
36 return; |
|
4189027691b7
webserver - add LogHandler
Franklin Schmidt <fschmidt@gmail.com>
parents:
1137
diff
changeset
|
37 } |
| 1137 | 38 throw new IOException("unexpected end of input at "+size); |
|
1138
4189027691b7
webserver - add LogHandler
Franklin Schmidt <fschmidt@gmail.com>
parents:
1137
diff
changeset
|
39 } |
| 1137 | 40 size += n; |
| 41 for( int i=0; i<=size-4; i++ ) { | |
| 42 if( a[i]=='\r' && a[i+1]=='\n' && a[i+2]=='\r' && a[i+3]=='\n' ) { | |
| 43 endOfHeader = i + 4; | |
| 44 break outer; | |
| 45 } | |
| 46 } | |
| 47 left -= n; | |
| 48 if( left == 0 ) { | |
| 49 byte[] a2 = new byte[2*a.length]; | |
| 50 System.arraycopy(a,0,a2,0,size); | |
| 51 a = a2; | |
| 52 left = a.length - size; | |
| 53 } | |
| 54 } | |
| 55 String rawRequest = new String(a,0,endOfHeader); | |
| 56 //System.out.println(rawRequest); | |
| 57 Request request = RequestHeadParser.parse(rawRequest); | |
| 58 //System.out.println(request.headers); | |
| 59 | |
| 60 Response response = server.handler.handle(request); | |
| 61 response.headers.put("Connection","close"); | |
| 62 response.headers.put("Content-Length",Long.toString(response.body.length)); | |
| 63 byte[] header = response.toHeaderString().getBytes(); | |
| 64 | |
| 65 OutputStream out = socket.getOutputStream(); | |
| 66 out.write(header); | |
| 67 copyAll(response.body.content,out); | |
| 68 out.close(); | |
| 69 socket.close(); | |
| 70 } catch(IOException e) { | |
| 71 logger.info("",e); | |
| 72 } catch(ParseException e) { | |
| 73 logger.info("",e); | |
| 74 } | |
| 75 } | |
| 76 | |
| 77 private static void copyAll(InputStream in,OutputStream out) | |
| 78 throws IOException | |
| 79 { | |
| 80 byte[] a = new byte[8192]; | |
| 81 int n; | |
| 82 while( (n=in.read(a)) != -1 ) { | |
| 83 out.write(a,0,n); | |
| 84 } | |
| 85 } | |
| 86 | |
| 87 } |