Check if Redis' socket classes are defined before using them

Some of these, such as `Redis::Connection::TCPSocket`, have been removed in redis-rb 5.0
basecamp · Mar 25, 2024 · 3d40ff7 · 3d40ff7
1 parent 60c2206
commit 3d40ff7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/console1984/shield.rb b/lib/console1984/shield.rb
@@ -47,8 +47,12 @@ def extend_sockets
       socket_classes = [TCPSocket, OpenSSL::SSL::SSLSocket]
       OpenSSL::SSL::SSLSocket.include(SSLSocketRemoteAddress)
 
-      if defined?(Redis::Connection)
-        socket_classes.push(*[Redis::Connection::TCPSocket, Redis::Connection::SSLSocket])
+      if defined?(Redis::Connection::TCPSocket)
+        socket_classes.push(Redis::Connection::TCPSocket)
+      end
+
+      if defined?(Redis::Connection::TCPSocket)
+        socket_classes.push(Redis::Connection::SSLSocket)
       end
 
       socket_classes.compact.each do |socket_klass|