Add writeAndFlush method in Response interface.

m0ver · m0ver · commit 31ba41eb9aba · 2025-05-19T00:25:21.000+08:00
diff --git a/src/main/java/org/tinystruct/http/Response.java b/src/main/java/org/tinystruct/http/Response.java
@@ -1,5 +1,7 @@
 package org.tinystruct.http;
 
+import org.tinystruct.ApplicationException;
+
 import java.io.IOException;
 
 public interface Response<T, O> extends Protocol {
@@ -28,5 +30,7 @@ public interface Response<T, O> extends Protocol {
 
     void sendRedirect(String url) throws IOException;
 
+    void writeAndFlush(byte[] bytes) throws ApplicationException;
+
     O get();
 }
diff --git a/src/main/java/org/tinystruct/http/ResponseBuilder.java b/src/main/java/org/tinystruct/http/ResponseBuilder.java
@@ -3,6 +3,7 @@
 import io.netty.handler.codec.http.FullHttpResponse;
 import io.netty.handler.codec.http.HttpHeaders;
 import io.netty.handler.codec.http.HttpResponseStatus;
+import org.tinystruct.ApplicationException;
 
 import java.io.IOException;
 
@@ -92,4 +93,13 @@ public void sendRedirect(String url) throws IOException {
         this.setStatus(ResponseStatus.TEMPORARY_REDIRECT);
     }
 
+    /**
+     * @param bytes
+     * @throws ApplicationException
+     */
+    @Override
+    public void writeAndFlush(byte[] bytes) throws ApplicationException {
+        this.get().content().writeBytes(bytes);
+    }
+
 }
diff --git a/src/main/java/org/tinystruct/http/servlet/ResponseBuilder.java b/src/main/java/org/tinystruct/http/servlet/ResponseBuilder.java
@@ -2,6 +2,7 @@
 
 import jakarta.servlet.ServletOutputStream;
 import jakarta.servlet.http.HttpServletResponse;
+import org.tinystruct.ApplicationException;
 import org.tinystruct.http.*;
 
 import java.io.IOException;
@@ -100,4 +101,18 @@ public void sendRedirect(String url) throws IOException {
         if (!this.response.isCommitted())
             this.response.sendRedirect(url);
     }
+
+    /**
+     * @param bytes
+     * @throws ApplicationException
+     */
+    @Override
+    public void writeAndFlush(byte[] bytes) throws ApplicationException {
+        try {
+            this.get().write(bytes);
+            this.get().flush();
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
 }
