add read stream

reactphp-x · Nov 14, 2024 · 4f80c02 · 4f80c02
1 parent 5acc667
commit 4f80c02
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/Bandwidth.php b/src/Bandwidth.php
@@ -63,6 +63,19 @@ public function file(string $path, $p = 0, $length = -1)
         return $stream;
     }
 
+    public function stream($stream)
+    {
+        $stream = new \React\Stream\ThroughStream();
+        $stream->on('data', function ($data) use ($stream) {
+            $this->concurrent->concurrent(function() use ($stream, $data){
+                return $this->bucket->removeTokens(1024 * strlen($data))->then(function () use ($stream, $data) {
+                    $stream->write($data);
+                });
+            });
+        });
+        return $stream;
+    }
+
     protected function fileStream($file, $stream, $p, $size)
     {