Simplify implementation of scheduler io_read and io_write. (#6527)

ruby · Oct 12, 2022 · 04d291a · 04d291a
1 parent 4e29ca0
commit 04d291a
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/test/fiber/scheduler.rb b/test/fiber/scheduler.rb
@@ -269,15 +269,14 @@ def io_read(io, buffer, length, offset)
     total = 0
     io.nonblock = true
 
-    while length >= 0
+    while true
       maximum_size = buffer.size - offset
-
       result = blocking{buffer.read(io, maximum_size, offset)}
 
       if result > 0
         total += result
         offset += result
-        break if result >= length
+        break if total >= length
       elsif result == 0
         break
       elsif result == EAGAIN
@@ -298,15 +297,14 @@ def io_write(io, buffer, length, offset)
     total = 0
     io.nonblock = true
 
-    while length >= 0
+    while true
       maximum_size = buffer.size - offset
-
       result = blocking{buffer.write(io, maximum_size, offset)}
 
       if result > 0
         total += result
         offset += result
-        break if result >= length
+        break if total >= length
       elsif result == 0
         break
       elsif result == EAGAIN