feat: Add Deno.TcpConn class, change return type from Deno.connect (#13714)

1 files changed, 15 insertions, 9 deletions

diff --git a/ext/net/01_net.js b/ext/net/01_net.js
index 2970dd817..8c17942da 100644
--- a/ext/net/01_net.js
+++ b/ext/net/01_net.js
@@ -163,14 +163,6 @@
       return shutdown(this.rid);
     }
 
-    setNoDelay(nodelay = true) {
-      return core.opSync("op_set_nodelay", this.rid, nodelay);
-    }
-
-    setKeepAlive(keepalive = true) {
-      return core.opSync("op_set_keepalive", this.rid, keepalive);
-    }
-
     get readable() {
       if (this.#readable === undefined) {
         this.#readable = readableStreamForRid(this.rid);
@@ -186,6 +178,16 @@
     }
   }
 
+  class TcpConn extends Conn {
+    setNoDelay(nodelay = true) {
+      return core.opSync("op_set_nodelay", this.rid, nodelay);
+    }
+
+    setKeepAlive(keepalive = true) {
+      return core.opSync("op_set_keepalive", this.rid, keepalive);
+    }
+  }
+
   class Listener {
     #rid = 0;
     #addr = null;
@@ -205,6 +207,9 @@
 
     async accept() {
       const res = await opAccept(this.rid, this.addr.transport);
+      if (this.addr.transport == "tcp") {
+        return new TcpConn(res.rid, res.remoteAddr, res.localAddr);
+      }
       return new Conn(res.rid, res.remoteAddr, res.localAddr);
     }
 
@@ -318,12 +323,13 @@
       });
     }
 
-    return new Conn(res.rid, res.remoteAddr, res.localAddr);
+    return new TcpConn(res.rid, res.remoteAddr, res.localAddr);
   }
 
   window.__bootstrap.net = {
     connect,
     Conn,
+    TcpConn,
     opConnect,
     listen,
     opListen,