fix: handle EAGAIN reading from TCP connection (#186)

denoland · Jun 6, 2024 · f4ec4e6 · f4ec4e6
1 parent df64a4d
commit f4ec4e6
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/packages/shim-deno/src/deno/internal/Conn.ts b/packages/shim-deno/src/deno/internal/Conn.ts
@@ -1,6 +1,7 @@
 ///<reference path="../stable/lib.deno.d.ts" />
 
 import { Socket } from "net";
+import { once } from "events";
 
 import { FsFile } from "../stable/classes/FsFile.js";
 
@@ -40,6 +41,21 @@ export class Conn extends FsFile implements Deno.Conn {
   unref(): void {
     this.#socket.unref();
   }
+
+  async read(p: Uint8Array): Promise<number | null> {
+    try {
+      return await super.read(p);
+    } catch (error) {
+      if (
+        !(error instanceof Error && "code" in error &&
+          error.code == "EAGAIN")
+      ) {
+        throw error;
+      }
+    }
+    await once(this.#socket, "readable");
+    return await super.read(p);
+  }
 }
 
 export class TlsConn extends Conn implements Deno.TlsConn {