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 } |