lib/mongo/operation/shared/op_msg_executable.rb in mongo-2.20.1 vs lib/mongo/operation/shared/op_msg_executable.rb in mongo-2.21.0

@@ -30,10 +30,13 @@
       # @param [ Operation::Context ] context The operation context.
       # @param [ Hash ] options Operation execution options.
       #
       # @return [ Mongo::Operation::Result ] The operation result.
       def execute(server, context:, options: {})
-        server.with_connection(connection_global_id: context.connection_global_id) do |connection|
+        server.with_connection(
+          connection_global_id: context.connection_global_id,
+          context: context
+        ) do |connection|
           execute_with_connection(connection, context: context, options: options)
         end
       end
 
       # Execute the operation.