fix for onread listener

wizzardo · Sep 30, 2023 · c7bd1bb · c7bd1bb
1 parent d095c9a
commit c7bd1bb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/main/java/com/wizzardo/http/HttpConnection.java b/src/main/java/com/wizzardo/http/HttpConnection.java
@@ -296,6 +296,16 @@ public void setCloseOnFinishWriting(boolean closeOnFinishWriting) {
 
     @Override
     public void onRead(ByteBufferProvider bufferProvider) throws IOException {
+        if (readListener != null) {
+            if (inputStream != null && processingBy.get() != null) // there is a worker waiting for wakeup
+                readListener.onRead(this, bufferProvider);
+
+            // todo should probably reset processingBy immediately when there is 0 bytes to read
+            if (inputStream == null && processingBy.get() == null) // there is no worker trying to read
+                readListener.onRead(this, bufferProvider);
+            return;
+        }
+
         server.process(this, bufferProvider);
     }