Merge pull request #69 from longnguyen2004/nosleep

dank074 · Feb 28, 2024 · 8510166 · 8510166
2 parents 3094aba + dfb0ba5
commit 8510166
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 8 deletions.
diff --git a/src/media/AudioStream.ts b/src/media/AudioStream.ts
@@ -6,12 +6,14 @@ class AudioStream extends Writable {
     public count: number;
     public sleepTime: number;
     public startTime?: number;
-
-    constructor(udp: MediaUdp) {
+    private noSleep: boolean;
+
+    constructor(udp: MediaUdp, noSleep = false) {
         super();
         this.udp = udp;
         this.count = 0;
         this.sleepTime = 20;
+        this.noSleep = noSleep;
     }
 
     _write(chunk: any, _: BufferEncoding, callback: (error?: Error | null) => void) {
@@ -22,9 +24,17 @@ class AudioStream extends Writable {
         this.udp.sendAudioFrame(chunk);
 
         const next = ((this.count + 1) * this.sleepTime) - (performance.now() - this.startTime);
-        setTimeout(() => {
+
+        if (this.noSleep)
+        {
             callback();
-        }, next);
+        }
+        else
+        {
+            setTimeout(() => {
+                callback();
+            }, next);
+        }
     }
 }
 

diff --git a/src/media/VideoStream.ts b/src/media/VideoStream.ts
@@ -6,12 +6,14 @@ export class VideoStream extends Writable {
     public count: number;
     public sleepTime: number;
     public startTime?: number;
-
-    constructor(udp: MediaUdp, fps: number = 30) {
+    private noSleep: boolean;
+
+    constructor(udp: MediaUdp, fps: number = 30, noSleep = false) {
         super();
         this.udp = udp;
         this.count = 0;
         this.sleepTime = 1000 / fps;
+        this.noSleep = noSleep;
     }
 
     public setSleepTime(time: number) {
@@ -27,8 +29,15 @@ export class VideoStream extends Writable {
 
         const next = ( (this.count + 1) * this.sleepTime) - (performance.now() - this.startTime);
 
-       setTimeout(() => {
+        if (this.noSleep)
+        {
             callback();
-        }, next);
+        }
+        else
+        {
+            setTimeout(() => {
+                callback();
+            }, next);
+        }
     }
 }