Implement reaping for lb

Author: comandeo-mongo

Diff for: gemfiles/standard.rb

@@ -65,6 +65,7 @@ def standard_dependencies
     gem 'tilt'
     # solargraph depends on rbs, which won't build on jruby for some reason
     gem 'solargraph', platforms: :mri
+    gem 'ruby-lsp', platforms: :mri
   end
 
   gem 'libmongocrypt-helper', '~> 1.8.0' if ENV['FLE'] == 'helper'

Diff for: lib/mongo/cluster/reapers/cursor_reaper.rb

@@ -194,7 +194,12 @@ def kill_cursors
             server_api: server.options[:server_api],
             connection_global_id: kill_spec.connection_global_id,
           }
-          op.execute(server, context: Operation::Context.new(options: options))
+          if connection = kill_spec.connection
+            op.execute_with_connection(connection, context: Operation::Context.new(options: options))
+            connection.connection_pool.check_in(connection)
+          else
+            op.execute(server, context: Operation::Context.new(options: options))
+          end
 
           if session = kill_spec.session
             if session.implicit?

Diff for: lib/mongo/cursor.rb

@@ -100,8 +100,10 @@ def initialize(view, result, server, options = {})
         check_in_connection
       else
         register
-        ObjectSpace.define_finalizer(self, self.class.finalize(kill_spec(@connection_global_id),
-          cluster))
+        ObjectSpace.define_finalizer(
+          self,
+          self.class.finalize(kill_spec(@connection_global_id), cluster)
+        )
       end
     end
 
@@ -406,6 +408,7 @@ def kill_spec(connection_global_id)
         connection_global_id: connection_global_id,
         server_address: server.address,
         session: @session,
+        connection: @connection
       )
     end
 

Diff for: lib/mongo/cursor/kill_spec.rb

@@ -31,22 +31,25 @@ def initialize(
         db_name:,
         connection_global_id:,
         server_address:,
-        session:
+        session:,
+        connection: nil
       )
         @cursor_id = cursor_id
         @coll_name = coll_name
         @db_name = db_name
         @connection_global_id = connection_global_id
         @server_address = server_address
         @session = session
+        @connection = connection
       end
 
       attr_reader :cursor_id,
       :coll_name,
       :db_name,
       :connection_global_id,
       :server_address,
-      :session
+      :session,
+      :connection
 
       def ==(other)
         cursor_id == other.cursor_id &&

Diff for: spec/integration/cursor_reaping_spec.rb

@@ -24,7 +24,7 @@
   let(:subscriber) { Mrss::EventSubscriber.new }
 
   let(:client) do
-    authorized_client.tap do |client|
+    authorized_client.with(max_pool_size: 10).tap do |client|
       client.subscribe(Mongo::Monitoring::COMMAND, subscriber)
     end
   end