RUBY-3723 Add tracing for atlas indexes (#2975)

Author: comandeo-mongo

lib/mongo/search_index/view.rb

@@ -4,6 +4,7 @@ module Mongo
   module SearchIndex
     # A class representing a view of search indexes.
     class View
+      extend Forwardable
       include Enumerable
       include Retryable
       include Collection::Helpers
@@ -21,6 +22,8 @@ class View
       #   when querying the available indexes.
       attr_reader :aggregate_options
 
+      def_delegators :@collection, :tracer
+
       # Create the new search index view.
       #
       # @param [ Collection ] collection The collection.
@@ -51,7 +54,10 @@ def initialize(collection, options = {})
       #
       # @return [ String ] the name of the new search index.
       def create_one(definition, name: nil, type: 'search')
-        create_many([ { name: name, definition: definition, type: type } ]).first
+        spec = { definition: definition, type: type }.tap do |sp|
+          sp[:name] = name unless name.nil?
+        end
+        create_many([ spec ]).first
       end
 
       # Create multiple search indexes with a single command.
@@ -64,8 +70,12 @@ def create_one(definition, name: nil, type: 'search')
       # @return [ Array<String> ] the names of the new search indexes.
       def create_many(indexes)
         spec = spec_with(indexes: indexes.map { |v| validate_search_index!(v) })
-        result = Operation::CreateSearchIndexes.new(spec).execute(next_primary, context: execution_context)
-        result.first['indexesCreated'].map { |idx| idx['name'] }
+        operation = Operation::CreateSearchIndexes.new(spec)
+        context = execution_context
+        tracer.trace_operation(operation, context, op_name: 'createSearchIndexes') do
+          result = operation.execute(next_primary, context: context)
+          result.first['indexesCreated'].map { |idx| idx['name'] }
+        end
       end
 
       # Drop the search index with the given id, or name. One or the other must
@@ -81,11 +91,14 @@ def drop_one(id: nil, name: nil)
 
         spec = spec_with(index_id: id, index_name: name)
         op = Operation::DropSearchIndex.new(spec)
+        context = execution_context
 
-        # per the spec:
-        # Drivers MUST suppress NamespaceNotFound errors for the
-        # ``dropSearchIndex`` helper.  Drop operations should be idempotent.
-        do_drop(op, nil, execution_context)
+        tracer.trace_operation(op, context, op_name: 'dropSearchIndex') do
+          # per the spec:
+          # Drivers MUST suppress NamespaceNotFound errors for the
+          # ``dropSearchIndex`` helper.  Drop operations should be idempotent.
+          do_drop(op, nil, context)
+        end
       end
 
       # Iterate over the search indexes.
@@ -127,7 +140,11 @@ def update_one(definition, id: nil, name: nil)
         validate_id_or_name!(id, name)
 
         spec = spec_with(index_id: id, index_name: name, index: definition)
-        Operation::UpdateSearchIndex.new(spec).execute(next_primary, context: execution_context)
+        op = Operation::UpdateSearchIndex.new(spec)
+        context = execution_context
+        tracer.trace_operation(op, context, op_name: 'updateSearchIndex') do
+          op.execute(next_primary, context: context)
+        end
       end
 
       # The following methods are to make the view act more like an array,

spec/spec_tests/open_telemetry_spec.rb

@@ -10,7 +10,7 @@
 base = "#{CURRENT_PATH}/spec_tests/data/open_telemetry"
 OTEL_UNIFIED_TESTS = Dir.glob("#{base}/**/*.yml").sort
 SKIPPED_OTEL_TESTS = [
-  'bulk_write.yml', 'map_reduce.yml', 'atlas_search.yml'
+  'bulk_write.yml', 'map_reduce.yml'
 ]
 
 TESTS_TO_RUN = OTEL_UNIFIED_TESTS.reject do |path|