update HttpServer

Author: litongjava

frameworks/Java/aio-socket/src/main/java/com/litongjava/aio/http/server/HttpServer.java

@@ -7,6 +7,9 @@
 import java.nio.channels.AsynchronousSocketChannel;
 import java.nio.channels.CompletionHandler;
 import java.nio.charset.StandardCharsets;
+import java.time.ZonedDateTime;
+import java.time.format.DateTimeFormatter;
+import java.time.ZoneId;
 import java.util.concurrent.ThreadFactory;
 
 import com.alibaba.fastjson.JSON;
@@ -26,31 +29,29 @@ public class HttpServer {
 
   public static void main(String[] args) throws Exception {
 
-    // 创建通道提供者，false 表示非低内存模式
+    // 创建异步通道提供者
     EnhanceAsynchronousChannelProvider provider = new EnhanceAsynchronousChannelProvider(false);
 
-    // 创建一个异步通道组，线程数设为2（根据需求调整）
+    // 创建通道组
     AsynchronousChannelGroup group = provider.openAsynchronousChannelGroup(2, new ThreadFactory() {
       @Override
       public Thread newThread(Runnable r) {
         return new Thread(r, "http-server-thread");
       }
     });
 
-    // 使用提供者创建服务器通道
+    // 创建服务器通道并绑定端口
     EnhanceAsynchronousServerSocketChannel server = (EnhanceAsynchronousServerSocketChannel) provider.openAsynchronousServerSocketChannel(group);
-    // 绑定端口，例如 80，设置 backlog
     server.bind(new InetSocketAddress(8080), 0);
 
     System.out.println("HTTP Server 正在监听端口 8080 ...");
 
-    // 异步接受连接请求
+    // 异步接受连接
     server.accept(null, new CompletionHandler<AsynchronousSocketChannel, Object>() {
       @Override
       public void completed(AsynchronousSocketChannel channel, Object attachment) {
-        // 接收到连接后，立即继续接受下一个连接
+        // 继续接收其他连接
         server.accept(null, this);
-        // 处理客户端连接
         handleClient(channel);
       }
 
@@ -60,16 +61,14 @@ public void failed(Throwable exc, Object attachment) {
       }
     });
 
-    // 主线程阻塞，以保证服务器运行
+    // 主线程阻塞
     Thread.currentThread().join();
   }
 
   private static void handleClient(AsynchronousSocketChannel channel) {
-    // 通过 BufferPage 池化获取一个 VirtualBuffer，分配 8192 字节空间
     VirtualBuffer virtualBuffer = bufferPage.allocate(8192);
     ByteBuffer buffer = virtualBuffer.buffer();
 
-    // 异步读取客户端请求，将 VirtualBuffer 作为附件传入
     channel.read(buffer, virtualBuffer, new CompletionHandler<Integer, VirtualBuffer>() {
       @Override
       public void completed(Integer result, VirtualBuffer attachment) {
@@ -78,25 +77,42 @@ public void completed(Integer result, VirtualBuffer attachment) {
             buffer.flip();
             byte[] bytes = new byte[buffer.remaining()];
             buffer.get(bytes);
-            // 将请求转换成字符串进行判断
             String request = new String(bytes, StandardCharsets.UTF_8);
-
+            
+            // 生成当前时间，格式为 RFC 1123 格式
+            String date = DateTimeFormatter.RFC_1123_DATE_TIME.format(ZonedDateTime.now(ZoneId.of("GMT")));
+            
             ByteBuffer responseBuffer;
-            // 如果请求以 "GET /plaintext" 开头，则直接返回 HELLO_WORLD_BYTES
             if (request.startsWith("GET /plaintext")) {
-              String httpResponse = "HTTP/1.1 200 OK\r\n" + "Content-Length: 13\r\n" + "Server: aio-socket\r\n" + "Content-Type: text/plain\r\n" + "\r\n" + "Hello, World!";
+              String body = "Hello, World!";
+              String httpResponse = "HTTP/1.1 200 OK\r\n" +
+                  "Content-Length: " + body.getBytes(StandardCharsets.UTF_8).length + "\r\n" +
+                  "Server: aio-socket\r\n" +
+                  "Content-Type: text/plain\r\n" +
+                  "Date: " + date + "\r\n" +
+                  "\r\n" +
+                  body;
               responseBuffer = ByteBuffer.wrap(httpResponse.getBytes(StandardCharsets.UTF_8));
 
             } else if (request.startsWith("GET /json")) {
               String jsonString = JSON.toJSONString(new Message(HELLO_WORLD));
-              int length = jsonString.length();
-              String httpResponse = "HTTP/1.1 200 OK\r\n" + "Content-Length: " + length + "\r\n" + "Server: aio-socket\r\n" + "Content-Type: application/json\r\n" + "\r\n" + jsonString;
-
+              int length = jsonString.getBytes(StandardCharsets.UTF_8).length;
+              String httpResponse = "HTTP/1.1 200 OK\r\n" +
+                  "Content-Length: " + length + "\r\n" +
+                  "Server: aio-socket\r\n" +
+                  "Content-Type: application/json\r\n" +
+                  "Date: " + date + "\r\n" +
+                  "\r\n" +
+                  jsonString;
               responseBuffer = ByteBuffer.wrap(httpResponse.getBytes(StandardCharsets.UTF_8));
-
             } else {
-              // 默认返回完整的 HTTP 响应
-              String httpResponse = "HTTP/1.1 200 OK\r\n" + "Content-Length: 13\r\n" + "Content-Type: text/plain\r\n" + "\r\n" + "Hello, World!";
+              String body = "Hello, World!";
+              String httpResponse = "HTTP/1.1 200 OK\r\n" +
+                  "Content-Length: " + body.getBytes(StandardCharsets.UTF_8).length + "\r\n" +
+                  "Content-Type: text/plain\r\n" +
+                  "Date: " + date + "\r\n" +
+                  "\r\n" +
+                  body;
               responseBuffer = ByteBuffer.wrap(httpResponse.getBytes(StandardCharsets.UTF_8));
             }
 
@@ -109,7 +125,6 @@ public void completed(Integer result, VirtualBuffer attachment) {
                 } catch (IOException e) {
                   e.printStackTrace();
                 } finally {
-                  // 写完响应后归还虚拟缓冲区
                   attachment.clean();
                 }
               }
@@ -122,13 +137,11 @@ public void failed(Throwable exc, VirtualBuffer attachment) {
                 } catch (IOException e) {
                   e.printStackTrace();
                 } finally {
-                  // 出现写异常时也归还虚拟缓冲区
                   attachment.clean();
                 }
               }
             });
           } else {
-            // 未读到数据，则关闭连接
             try {
               channel.close();
             } catch (IOException e) {
@@ -137,7 +150,6 @@ public void failed(Throwable exc, VirtualBuffer attachment) {
           }
         } finally {
           // 注意：如果在写操作中已经归还了虚拟缓冲区，则不要重复释放
-          // attachment.clean();
         }
       }