From 41c912ee6a10885ef3967f78eb228a29ea9acb75 Mon Sep 17 00:00:00 2001
From: Samuel Williams <samuel.williams@oriontransfer.co.nz>
Date: Tue, 17 Nov 2020 11:54:24 +1300
Subject: [PATCH] Prefer `IO#wait_readable`/`IO#wait_writable` rather than
 `IO.select`.

This allows the Ruby 3 scheduler hooks to improve concurrency of Redis.
---
 lib/redis/connection/ruby.rb | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/lib/redis/connection/ruby.rb b/lib/redis/connection/ruby.rb
index 756ef8db9..8638a6312 100644
--- a/lib/redis/connection/ruby.rb
+++ b/lib/redis/connection/ruby.rb
@@ -52,11 +52,11 @@ def _read_from_socket(nbytes)
         loop do
           case chunk = read_nonblock(nbytes, exception: false)
           when :wait_readable
-            unless IO.select([self], nil, nil, @timeout)
+            unless wait_readable(@timeout)
               raise Redis::TimeoutError
             end
           when :wait_writable
-            unless IO.select(nil, [self], nil, @timeout)
+            unless wait_writable(@timeout)
               raise Redis::TimeoutError
             end
           when nil
@@ -72,11 +72,11 @@ def _write_to_socket(data)
         loop do
           case bytes_written = write_nonblock(data, exception: false)
           when :wait_readable
-            unless IO.select([self], nil, nil, @write_timeout)
+            unless wait_readable(@write_timeout)
               raise Redis::TimeoutError
             end
           when :wait_writable
-            unless IO.select(nil, [self], nil, @write_timeout)
+            unless wait_writable(@write_timeout)
               raise Redis::TimeoutError
             end
           when nil
@@ -139,7 +139,7 @@ def self.connect(path, timeout)
             raise TimeoutError
           end
 
-          # JRuby raises Errno::EAGAIN on #read_nonblock even when IO.select
+          # JRuby raises Errno::EAGAIN on #read_nonblock even when it
           # says it is readable (1.6.6, in both 1.8 and 1.9 mode).
           # Use the blocking #readpartial method instead.
 
@@ -164,7 +164,7 @@ def self.connect_addrinfo(addrinfo, port, timeout)
           begin
             sock.connect_nonblock(sockaddr)
           rescue Errno::EINPROGRESS
-            raise TimeoutError if IO.select(nil, [sock], nil, timeout).nil?
+            raise TimeoutError unless sock.wait_writable(timeout)
 
             begin
               sock.connect_nonblock(sockaddr)
@@ -219,7 +219,7 @@ def self.connect(path, timeout)
           begin
             sock.connect_nonblock(sockaddr)
           rescue Errno::EINPROGRESS
-            raise TimeoutError if IO.select(nil, [sock], nil, timeout).nil?
+            raise TimeoutError unless sock.wait_writable(timeout)
 
             begin
               sock.connect_nonblock(sockaddr)
@@ -258,13 +258,13 @@ def self.connect(host, port, timeout, ssl_params)
             # Instead, you have to retry.
             ssl_sock.connect_nonblock
           rescue Errno::EAGAIN, Errno::EWOULDBLOCK, IO::WaitReadable
-            if IO.select([ssl_sock], nil, nil, timeout)
+            if ssl_sock.wait_readable(timeout)
               retry
             else
               raise TimeoutError
             end
           rescue IO::WaitWritable
-            if IO.select(nil, [ssl_sock], nil, timeout)
+            if ssl_sock.wait_writable(timeout)
               retry
             else
               raise TimeoutError