Fix buffered IO signatures for Ruby 4.1

Assisted-By: devx/ab4e61c0-6c66-4d82-b517-93ef8659abb1

Author: samuel-williams-shopify

lib/async/scheduler.rb

@@ -336,16 +336,35 @@ def io_wait(io, events, timeout = nil)
 		end
 
 		if ::IO::Event::Support.buffer?
-			# Read from the specified IO into the buffer.
-			#
-			# @public Since *Async v2* and Ruby with `IO::Buffer` support.
-			# @asynchronous May be non-blocking.
-			#
-			# @parameter io [IO] The IO object to read from.
-			# @parameter buffer [IO::Buffer] The buffer to read into.
-			# @parameter length [Integer] The minimum number of bytes to read.
-			# @parameter offset [Integer] The offset within the buffer to read into.
-			def io_read(io, buffer, length, offset = 0)
+			if defined?(IO::Buffer::VERSION) && IO::Buffer::VERSION >= 3
+				# Read at most the specified number of bytes from the IO into the buffer.
+				#
+				# @public Since *Async v2* and Ruby with `IO::Buffer` support.
+				# @asynchronous May be non-blocking.
+				#
+				# @parameter io [IO] The IO object to read from.
+				# @parameter buffer [IO::Buffer] The buffer to read into.
+				# @parameter offset [Integer] The offset within the buffer to read into.
+				# @parameter length [Integer] The maximum number of bytes to read.
+				def io_read(io, buffer, offset, length)
+					io_read_with_timeout(io, buffer, offset, length)
+				end
+			else
+				# Read from the specified IO into the buffer.
+				#
+				# @public Since *Async v2* and Ruby with `IO::Buffer` support.
+				# @asynchronous May be non-blocking.
+				#
+				# @parameter io [IO] The IO object to read from.
+				# @parameter buffer [IO::Buffer] The buffer to read into.
+				# @parameter length [Integer] The minimum number of bytes to read.
+				# @parameter offset [Integer] The offset within the buffer to read into.
+				def io_read(io, buffer, length, offset = 0)
+					io_read_with_timeout(io, buffer, length, offset)
+				end
+			end
+			
+			private def io_read_with_timeout(io, buffer, first, second)
 				fiber = Fiber.current
 
 				if timeout = io.timeout
@@ -354,22 +373,41 @@ def io_read(io, buffer, length, offset = 0)
 					end
 				end
 
-				@selector.io_read(fiber, io, buffer, length, offset)
+				@selector.io_read(fiber, io, buffer, first, second)
 			ensure
 				timer&.cancel!
 			end
 
 			if RUBY_ENGINE != "ruby" || RUBY_VERSION >= "3.3.1"
-				# Write the specified buffer to the IO.
-				#
-				# @public Since *Async v2* and *Ruby v3.3.1* with `IO::Buffer` support.
-				# @asynchronous May be non-blocking.
-				#
-				# @parameter io [IO] The IO object to write to.
-				# @parameter buffer [IO::Buffer] The buffer to write from.
-				# @parameter length [Integer] The minimum number of bytes to write.
-				# @parameter offset [Integer] The offset within the buffer to write from.
-				def io_write(io, buffer, length, offset = 0)
+				if defined?(IO::Buffer::VERSION) && IO::Buffer::VERSION >= 3
+					# Write at most the specified number of bytes from the buffer to the IO.
+					#
+					# @public Since *Async v2* and *Ruby v3.3.1* with `IO::Buffer` support.
+					# @asynchronous May be non-blocking.
+					#
+					# @parameter io [IO] The IO object to write to.
+					# @parameter buffer [IO::Buffer] The buffer to write from.
+					# @parameter offset [Integer] The offset within the buffer to write from.
+					# @parameter length [Integer] The maximum number of bytes to write.
+					def io_write(io, buffer, offset, length)
+						io_write_with_timeout(io, buffer, offset, length)
+					end
+				else
+					# Write the specified buffer to the IO.
+					#
+					# @public Since *Async v2* and *Ruby v3.3.1* with `IO::Buffer` support.
+					# @asynchronous May be non-blocking.
+					#
+					# @parameter io [IO] The IO object to write to.
+					# @parameter buffer [IO::Buffer] The buffer to write from.
+					# @parameter length [Integer] The minimum number of bytes to write.
+					# @parameter offset [Integer] The offset within the buffer to write from.
+					def io_write(io, buffer, length, offset = 0)
+						io_write_with_timeout(io, buffer, length, offset)
+					end
+				end
+				
+				private def io_write_with_timeout(io, buffer, first, second)
 					fiber = Fiber.current
 
 					if timeout = io.timeout
@@ -378,7 +416,7 @@ def io_write(io, buffer, length, offset = 0)
 						end
 					end
 
-					@selector.io_write(fiber, io, buffer, length, offset)
+					@selector.io_write(fiber, io, buffer, first, second)
 				ensure
 					timer&.cancel!
 				end

test/async/scheduler.rb

@@ -125,7 +125,11 @@
 				scheduler.instance_variable_set(:@selector, selector)
 
 				expect do
-					scheduler.io_read(io, IO::Buffer.new(1024), 1024)
+					if defined?(IO::Buffer::VERSION) && IO::Buffer::VERSION >= 3
+						scheduler.io_read(io, IO::Buffer.new(1024), 0, 1024)
+					else
+						scheduler.io_read(io, IO::Buffer.new(1024), 1024)
+					end
 				end.to raise_exception(::IO::TimeoutError)
 			ensure
 				scheduler&.close