fix force buffer

sonroyaalmerol · Sep 23, 2024 · e2f59fa · e2f59fa
1 parent 70790ed
commit e2f59fa
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/proxy/buffer.go b/proxy/buffer.go
@@ -38,9 +38,11 @@ func (b *Buffer) ReadChunk(size int, force bool) ([]byte, bool) {
 	defer b.mu.Unlock()
 
 	// Wait for buffer to have enough data
-	if !force {
-		for len(b.data) < size {
-			b.cond.Wait()
+	for len(b.data) < size {
+		b.cond.Wait()
+		if force {
+			size = len(b.data)
+			break
 		}
 	}