feat: added events for connection recv/send and stream readable/writable

[ci skip]
MatrixAI · May 11, 2023 · af0bfbc · af0bfbc
1 parent ff2e533
commit af0bfbc
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 13 deletions.
diff --git a/src/QUICConnection.ts b/src/QUICConnection.ts
@@ -451,6 +451,7 @@ class QUICConnection extends EventTarget {
         }
         return;
       }
+      this.dispatchEvent(new events.QUICConnectionRecvEvent());
       // Here we can resolve our promises!
       if (this.conn.isEstablished()) {
         this.resolveEstablishedP();
@@ -482,6 +483,7 @@ class QUICConnection extends EventTarget {
             );
           }
           quicStream.read();
+          quicStream.dispatchEvent(new events.QUICStreamReadableEvent());
         }
         for (const streamId of this.conn.writable() as Iterable<StreamId>) {
           let quicStream = this.streamMap.get(streamId);
@@ -500,6 +502,7 @@ class QUICConnection extends EventTarget {
               new events.QUICConnectionStreamEvent({ detail: quicStream }),
             );
           }
+          quicStream.dispatchEvent(new events.QUICStreamWritableEvent());
           quicStream.write();
         }
         // Checking shortlist if streams have finished.
@@ -626,6 +629,7 @@ class QUICConnection extends EventTarget {
           );
           return;
         }
+        this.dispatchEvent(new events.QUICConnectionSendEvent());
       }
     } finally {
       this.logger.debug('SEND FINALLY');

diff --git a/src/QUICStream.ts b/src/QUICStream.ts
@@ -57,7 +57,7 @@ class QUICStream
 
   /**
    * For `reasonToCode`, return 0 means "unknown reason"
-   * It is the catch all for codes.
+   * It is the catch-all for codes.
    * So it is the default reason.
    *
    * It may receive any reason for cancellation.

diff --git a/src/events.ts b/src/events.ts
@@ -71,6 +71,18 @@ class QUICConnectionStreamEvent extends Event {
   }
 }
 
+class QUICConnectionSendEvent extends Event {
+  constructor(options?: EventInit) {
+    super('send', options);
+  }
+}
+
+class QUICConnectionRecvEvent extends Event {
+  constructor(options?: EventInit) {
+    super('recv', options);
+  }
+}
+
 class QUICConnectionDestroyEvent extends Event {
   constructor(options?: EventInit) {
     super('destroy', options);
@@ -89,18 +101,17 @@ class QUICConnectionErrorEvent extends Event {
   }
 }
 
-// TODO: use these or remove them
-// class QUICStreamReadableEvent extends Event {
-//   constructor(options?: EventInit) {
-//     super('readable', options);
-//   }
-// }
-//
-// class QUICStreamWritableEvent extends Event {
-//   constructor(options?: EventInit) {
-//     super('writable', options);
-//   }
-// }
+class QUICStreamReadableEvent extends Event {
+  constructor(options?: EventInit) {
+    super('readable', options);
+  }
+}
+
+class QUICStreamWritableEvent extends Event {
+  constructor(options?: EventInit) {
+    super('writable', options);
+  }
+}
 
 class QUICStreamDestroyEvent extends Event {
   constructor(options?: EventInit) {
@@ -133,8 +144,12 @@ export {
   QUICServerStopEvent,
   QUICServerErrorEvent,
   QUICConnectionStreamEvent,
+  QUICConnectionSendEvent,
+  QUICConnectionRecvEvent,
   QUICConnectionDestroyEvent,
   QUICConnectionErrorEvent,
+  QUICStreamReadableEvent,
+  QUICStreamWritableEvent,
   QUICStreamDestroyEvent,
   QUICClientDestroyEvent,
   QUICClientErrorEvent,