NIO开发Http服务器（4）：Response封装和响应

 1 public class Cookie {
 2 
 3     private String name;
 4     private String value;
 5     private long age;
 6     private String path = "/";
 7     private String domain;
 8 
 9     public Cookie() {
10         super();
11     }
12 
13     public Cookie(String name, String value, long age) {
14         super();
15         this.name = name;
16         this.value = value;
17         this.age = age;
18     }
19 
20     // getter and setter
21 }

 1 // 设置http响应状态码
 2 void setResponseCode(int status);
 3 
 4 // 设置http响应的Content-Type
 5 void setContentType(String contentType);
 6 
 7 // 设置header
 8 void setHeader(String headerName, String headerValue);
 9 
10 // 添加一个cookie到响应中
11 void addCookie(Cookie cookie);
12 
13 // 设置响应编码字符集
14 void setCharsetEncoding(String charsetName);
15 
16 // 响应
17 void response();
18 
19 // 获取当前请求所对应的客户端socket通道
20 @Deprecated
21 SocketChannel getOut();
22 
23 // 把指定的字符串写入响应缓冲区
24 void print(String line);
25 
26 // 把指定的字符串写入响应缓冲区，末尾有换行符
27 void println(String line);

 1 // 时间格式化工具
 2 private static SimpleDateFormat sdf = new SimpleDateFormat("EEE, d MMM yyyy HH:mm:ss ‘GMT‘", Locale.US);
 4 
 5 // 编码字符集
 6 private CharsetEncoder encoder;
 7 
 8 // 响应的Content-Type
 9 private String contentType = "text/html;charset=utf-8";
10 
11 // 响应状态码
12 private int status = 0;
13 
14 // 响应头
15 private Map<String, String> headers = new HashMap<String, String>();
16 
17 // 响应cookie
18 private List<Cookie> cookies = new ArrayList<Cookie>();
19 
20 // 本地资源输入通道
21 private FileChannel in;
22 
23 // 客户端输出通道
24 private SocketChannel out;
25 
26 // 动态资源生成的数据
27 private StringBuilder content = new StringBuilder();
28 
29 // 获取服务器配置
30 HttpServerConfig config = HttpServerConfig.getInstance();

 1 public HttpResponse(SocketChannel sChannel) {
 2     // 获取GBK字符集
 3     Charset c1 = Charset.forName(config.getResponseCharset());
 4     // 获取编码器
 5     this.encoder = c1.newEncoder();
 6     // 获取Content-Type
 7     this.setContentType(ContentTypeUtil.getContentType(ContentTypeUtil.HTML));
 8     this.headers.put("Date", sdf.format(new Date()));
 9     this.headers.put("Server", "nginx");
10     this.headers.put("Connection", "keep-alive");
11     // 客户端输出通道
12     this.out = sChannel;
13 }

 1 public HttpResponse(Request req, SocketChannel sChannel) {
 2 
 3     this(sChannel);
 4 
 5     // 获取请求资源URI
 6     String uri = req.getRequestURI();
 7 
 8     // 获取本地输入通道
 9     this.getLocalFileChannel(uri);
10 
11     // 设置Content-Type
12     this.setContentType(req.getContentType());
13 
14     // 设置静态资源过期响应头
15     int expires = config.getExpiresMillis(this.contentType);
16     if (expires > 0) {
17         long expiresTimeStamp = System.currentTimeMillis() + expires;
18         this.headers.put("Expires", sdf.format(new Date(expiresTimeStamp)));
19     }
20 }

 1 private void getLocalFileChannel(String uri) {
 2     // 打开本地文件
 3     try {
 4         this.in = FileChannel.open(Paths.get(config.getRoot(), uri),
 5                 StandardOpenOption.READ);
 6         // 设置Content-Length响应头
 7         this.setHeader("Content-Length", String.valueOf(in.size()));
 8         // 设置响应状态码200
 9         this.setResponseCode(ResponseUtil.RESPONSE_CODE_200);
10     } catch (NoSuchFileException e) {
11         // 没有本地资源被找到
12         // 设置响应状态码404
13         this.setResponseCode(ResponseUtil.RESPONSE_CODE_404);
14         // 关闭本地文件通道
15         this.closeLocalFileChannel();
16     } catch (IOException e) {
17         // 打开资源时出现异常
18         // 设置响应状态码500
19         this.setResponseCode(ResponseUtil.RESPONSE_CODE_500);
20         // 关闭本地文件通道
21         this.closeLocalFileChannel();
22     }
23 }

1 public void setCharsetEncoding(String charsetName) {
2     // 获取GBK字符集
3     Charset c1 = Charset.forName(charsetName);
4     // 获取编码器
5     this.encoder = c1.newEncoder();
6 }

 1 private void newLine() throws IOException {
 2     this.write("\n");
 3 }
 4 
 5 private void write(String content) throws IOException {
 6     CharBuffer cBuf = CharBuffer.allocate(content.length());
 7     cBuf.put(content);
 8     cBuf.flip();
 9     ByteBuffer bBuf = this.encoder.encode(cBuf);
10     this.out.write(bBuf);
11 }