[EPIC] v0.2.0: Streaming Support & High-Performance I/O #11

.github/labeler.yml

@@ -0,0 +1,8 @@
+# .github/labeler.yml
+# Adds the 'java' label to any PR that touches .java files
+java:
+  - '**/*.java'
+
+# Adds the 'ci' label if you touch the workflow files
+ci:
+  - '.github/workflows/**/*'

.github/workflows/codeql.yml

@@ -13,9 +13,9 @@ name: "CodeQL Advanced"
 
 on:
   push:
-    branches: [ "main" ]
+    branches: [ "main", "development" ]
   pull_request:
-    branches: [ "main" ]
+    branches: [ "main", development ]
   schedule:
     - cron: '20 2 * * 6'
 

.gitignore

@@ -41,4 +41,6 @@ build/
 
 Main.java
 keystore.p12
-public
+public
+development/
+random.txt

README.md

@@ -2,7 +2,6 @@
 ![Maven Central](https://img.shields.io/maven-central/v/io.github.bernardusz/levtus-core)
 ![License](https://img.shields.io/github/license/Bernardusz/levtus)
 ![Maven Release](https://img.shields.io/github/actions/workflow/status/Bernardusz/levtus/release.yml?label=maven%20release)
-[![Qodana](https://github.com/Bernardusz/levtus/actions/workflows/qodana_code_quality.yml/badge.svg?branch=main)](https://github.com/Bernardusz/levtus/actions/workflows/qodana_code_quality.yml)
 
 **Levtus** (Latin: *Levis Conatus* - "Light Effort") is a high-performance, zero-dependency HTTP/1.1 engine built from the ground up for the modern JVM. It is designed to be lightweight, secure, and incredibly fast by leveraging the power of **Java 21+ Virtual Threads (Project Loom)**.
 

pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>io.github.bernardusz</groupId>
     <artifactId>levtus-core</artifactId>
-    <version>0.1.1</version>
+    <version>0.2.0</version>
 
     <name>Levtus Core</name>
     <description>A high-performance, zero-dependency HTTP/1.1 engine built for the modern JVM using Virtual Threads.</description>

src/main/java/io/github/bernardusz/levtus/Levtus.java

@@ -3,6 +3,7 @@
 import io.github.bernardusz.levtus.engine.LevtusEngine;
 import io.github.bernardusz.levtus.http.LevtusContext;
 import io.github.bernardusz.levtus.routing.Middleware;
+import io.github.bernardusz.levtus.routing.Node;
 import io.github.bernardusz.levtus.routing.Router;
 import java.util.function.Consumer;
 
@@ -44,9 +45,10 @@ public static Levtus create() {
    *
    * @param path the exact URL path to match (must not be null, e.g., "/api/users")
    * @param handler the functional callback executed upon a successful match (must not be null)
+   * @return the registered route Node
    */
-  public void get(String path, Consumer<LevtusContext> handler) {
-    router.get(path, handler);
+  public Node get(String path, Consumer<LevtusContext> handler) {
+    return router.get(path, handler);
   }
 
   /**
@@ -56,29 +58,32 @@ public void get(String path, Consumer<LevtusContext> handler) {
    *
    * @param path the exact URL path to match (must not be null)
    * @param handler the functional callback executed upon a successful match (must not be null)
+   * @return the registered route Node
    */
-  public void post(String path, Consumer<LevtusContext> handler) {
-    router.post(path, handler);
+  public Node post(String path, Consumer<LevtusContext> handler) {
+    return router.post(path, handler);
   }
 
   /**
    * Maps an HTTP PUT request to a specific handler function, typically used for updating resources.
    *
    * @param path the exact URL path to match (must not be null)
    * @param handler the functional callback executed upon a successful match (must not be null)
+   * @return the registered route Node
    */
-  public void put(String path, Consumer<LevtusContext> handler) {
-    router.put(path, handler);
+  public Node put(String path, Consumer<LevtusContext> handler) {
+    return router.put(path, handler);
   }
 
   /**
    * Maps an HTTP DELETE request to a specific handler function.
    *
    * @param path the exact URL path to match (must not be null)
    * @param handler the functional callback executed upon a successful match (must not be null)
+   * @return the registered route Node
    */
-  public void delete(String path, Consumer<LevtusContext> handler) {
-    router.delete(path, handler);
+  public Node delete(String path, Consumer<LevtusContext> handler) {
+    return router.delete(path, handler);
   }
 
   /**
@@ -192,6 +197,28 @@ public Levtus staticFiles(String staticFilesPath) {
     return this;
   }
 
+  /**
+   * Set the initial socket timeout for all incoming HTTP requests
+   *
+   * @param initialSocketTimeout the initial socket timeout
+   * @return The current Levtus instance for method chaining
+   */
+  public Levtus initialSocketTimeout(int initialSocketTimeout) {
+    engine.setInitialSocketTimeout(initialSocketTimeout);
+    return this;
+  }
+
+  /**
+   * Set the processing socket timeout for all incoming HTTP requests
+   *
+   * @param processingSocketTimeout the processing socket timeout
+   * @return The current Levtus instance for method chaining
+   */
+  public Levtus processingSocketTimeout(int processingSocketTimeout) {
+    engine.setProcessingSocketTimeout(processingSocketTimeout);
+    return this;
+  }
+
   /**
    * Binds the Levtus application to a specific TCP port and boots the server.
    *

src/main/java/io/github/bernardusz/levtus/engine/HttpConnectionHandler.java

@@ -1,54 +1,84 @@
 package io.github.bernardusz.levtus.engine;
 
-import io.github.bernardusz.levtus.exception.LevtusHttpException;
+import io.github.bernardusz.levtus.exception.http.LevtusHttpException;
 import io.github.bernardusz.levtus.http.Request;
 import io.github.bernardusz.levtus.http.Response;
 import io.github.bernardusz.levtus.routing.Router;
 import io.github.bernardusz.levtus.spi.ConnectionHandler;
 import java.io.BufferedInputStream;
-import java.io.BufferedOutputStream;
 import java.io.EOFException;
+import java.io.OutputStream;
 import java.net.Socket;
 import java.net.SocketTimeoutException;
 
 class HttpConnectionHandler implements ConnectionHandler {
   final Router router;
   final HttpParser parser;
   private int maxEmptyLines = 10;
-  private int maxBodySize = 10 * 1024 * 1024;
+  private long maxBodySize = 10 * 1024 * 1024;
   private int maxHeaderCount = 100;
   private int maxLineSize = 8192; // 8 KB Limit
   private int maxHeaderSize = 8192; // 8 KB Limit
   private String staticFilesPath = "./public";
+  private long maxChunkSize = 1024 * 1024; // 1 MB Limit
+  private long maxChunkCount = 1000;
+  private int initialSocketTimeout = 5000;  // Initial request data
+  private int processingSocketTimeout = 20000;  // Processing phase
 
+  /**
+   * The constructor for HttpConnectionHandler.
+   *
+   * @param router the router for HTTP Handler
+   * @param parser the parser for HTTP Parser
+   */
   public HttpConnectionHandler(Router router, HttpParser parser) {
     this.router = router;
     this.parser = parser;
   }
 
+  /**
+   * The main method that handles a client connection.
+   *
+   * @param client the client socket to process
+   */
   public void handle(Socket client) {
     try (client;
         BufferedInputStream inputStream = new BufferedInputStream(client.getInputStream());
-        BufferedOutputStream outputStream = new BufferedOutputStream(client.getOutputStream())) {
-      client.setSoTimeout(5000);
-      Response res = new Response(outputStream, staticFilesPath);
-
+        OutputStream outputStream = client.getOutputStream()) {
+      client.setSoTimeout(initialSocketTimeout);
+      Response res = new Response(outputStream, staticFilesPath, false); // At first, it is false,
+      // Because we won't continue sending responses after the error
       try {
         Request req;
-        while ((req = parser.parseRequest(this, inputStream)) != null) {
-          res = new Response(outputStream, staticFilesPath);
-          client.setSoTimeout(20000);
+        while ((req = parser.parseRequest(this, inputStream, res)) != null) {
+          res = new Response(outputStream, staticFilesPath, req.isKeepAlive()); // Now when we respond, that's when the isKeepAlive is needed
+          client.setSoTimeout(processingSocketTimeout);
           router.handle(req, res);
-          if (!res.isSent()) {
-            res.status(404).send("404 - Not Found");
+          if (!res.isSent()){
+            if (res.isChunked()){
+              res.finishChunkedResponse();
+            }
+            else {
+              res.status(404).send("404 - Not Found");
+            }
           }
           try {
-            if (!req.isCached()) {
-              inputStream.skipNBytes(req.contentLength() - req.bytesRead());
+            if (!req.isCached()){
+              if (req.isChunked()){
+                req.body();
+              }
+              else {
+                inputStream.skipNBytes(req.contentLength() - req.bytesRead());
+              }
             }
           } catch (EOFException e) {
             break;
           }
+
+          if (!req.isKeepAlive()) {
+            break;
+          }
+
         }
       } catch (IllegalArgumentException e) {
         res.status(400).send("400 - Bad Request (Malformed URL)");
@@ -59,6 +89,8 @@ public void handle(Socket client) {
       } catch (Exception e) {
         res.status(500).send("500 - Internal Server Error");
         throw e;
+      } finally {
+        res.finishChunkedResponse();
       }
 
     } catch (Exception e) {
@@ -90,7 +122,7 @@ public void setMaxEmptyLines(int maxEmptyLines) {
    *
    * @return return the max body size in a request
    */
-  int getMaxBodySize() {
+  long getMaxBodySize() {
     return maxBodySize;
   }
 
@@ -99,7 +131,7 @@ int getMaxBodySize() {
    *
    * @param maxBodySize the max body size
    */
-  public void setMaxBodySize(int maxBodySize) {
+  public void setMaxBodySize(long maxBodySize) {
     this.maxBodySize = maxBodySize;
   }
 
@@ -157,6 +189,22 @@ public void setMaxHeaderSize(int maxHeaderSize) {
     this.maxHeaderSize = maxHeaderSize;
   }
 
+  long getMaxChunkSize() {
+    return maxChunkSize;
+  }
+
+  public void setMaxChunkSize(int maxChunkSize) {
+    this.maxChunkSize = maxChunkSize;
+  }
+
+  long getMaxChunkCount() {
+    return maxChunkCount;
+  }
+
+  public void setMaxChunkCount(long maxChunkCount) {
+    this.maxChunkCount = maxChunkCount;
+  }
+
   /**
    * Sets the path/directory in which all static files are set.
    *
@@ -167,4 +215,41 @@ public void setMaxHeaderSize(int maxHeaderSize) {
   public void setStaticFiles(String staticFilesPath) {
     this.staticFilesPath = staticFilesPath;
   }
+
+  /**
+   * Get the initial socket timeout for all incoming HTTP requests.
+   *
+   * @return return the initial socket timeout for all incoming HTTP requests
+   */
+  int getInitialSocketTimeout() {
+    return initialSocketTimeout;
+  }
+
+  /**
+   * Set the initial socket timeout for all incoming HTTP requests
+   *
+   * @param initialSocketTimeout the initial socket timeout
+   */
+  public void setInitialSocketTimeout(int initialSocketTimeout){
+    this.initialSocketTimeout = initialSocketTimeout;
+  }
+
+  /**
+   * Get the processing socket timeout for all incoming HTTP requests.
+   *
+   * @return return the processing socket timeout for all incoming HTTP requests
+   */
+  int getProcessingSocketTimeout() {
+    return processingSocketTimeout;
+  }
+
+  /**
+   * Set the processing socket timeout for all incoming HTTP requests
+   *
+   * @param processingSocketTimeout the processing socket timeout
+   */
+  public void setProcessingSocketTimeout(int processingSocketTimeout){
+    this.processingSocketTimeout = processingSocketTimeout;
+  }
+
 }