Class: Mongo::Tracing::OpenTelemetry::CommandTracer Private

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 271

def collection_name(message)
  case command_name(message)
  when 'getMore'
    message.documents.first['collection'].to_s
  when 'listCollections', 'listDatabases', 'commitTransaction', 'abortTransaction'
    nil
  else
    value = message.documents.first.values.first
    # Return nil if the value is not a string (e.g., for admin commands that have numeric values)
    value.is_a?(String) ? value : nil
  end
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 289

def command_name(message)
  message.documents.first.keys.first.to_s
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 204

def connection_attributes(connection)
  {
    'server.port' => connection.address.port,
    'server.address' => connection.address.host,
    'network.transport' => connection.transport.to_s,
    'db.mongodb.server_connection_id' => connection.server.description.server_connection_id,
    'db.mongodb.driver_connection_id' => connection.id
  }
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 133

def create_command_span(message, connection)
  @otel_tracer.start_span(
    command_name(message),
    attributes: span_attributes(message, connection),
    kind: :client
  )
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 314

def cursor_id(message)
  return unless command_name(message) == 'getMore'

  message.documents.first['getMore'].value
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 298

def database(message)
  message.documents.first['$db'].to_s
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 154

def handle_command_exception(span, exception)
  if exception.is_a?(Mongo::Error::OperationFailure)
    span.set_attribute('db.response.status_code', exception.code.to_s)
  end
  span.record_exception(exception)
  span.status = ::OpenTelemetry::Trace::Status.error("Unhandled exception of type: #{exception.class}")
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 325

def lsid(message)
  lsid_doc = message.documents.first['lsid']
  return unless lsid_doc

  lsid_doc['id'].to_uuid
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 242

def maybe_trace_error(result, span)
  return if result.successful?

  span.set_attribute('db.response.status_code', result.error.code.to_s)
  begin
    result.validate!
  rescue Mongo::Error::OperationFailure => e
    span.record_exception(e)
  end
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 234

def normalize_cursor_id(cursor_id)
  cursor_id.is_a?(BSON::Int64) ? cursor_id.value : cursor_id
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 145

def process_command_result(result, span)
  process_cursor_context(result, span)
  maybe_trace_error(result, span)
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 218

def process_cursor_context(result, span)
  cursor_id = normalize_cursor_id(result.cursor_id)
  return unless cursor_id.positive?

  span.set_attribute('db.mongodb.cursor_id', cursor_id)
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 258

def query_summary(message)
  if (coll_name = collection_name(message))
    "#{command_name(message)} #{database(message)}.#{coll_name}"
  else
    "#{command_name(message)} #{database(message)}"
  end
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 355

def query_text(message)
  return unless query_text?

  text = message
         .payload['command']
         .reject { |key, _| EXCLUDED_KEYS.include?(key) }
         .to_json
  if text.length > @query_text_max_length
    "#{text[0...@query_text_max_length]}#{ELLIPSIS}"
  else
    text
  end
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 120

def skip_tracing?(message)
  name = command_name(message)
  return true if HELLO_COMMANDS.include?(name)

  sensitive?(command_name: name, document: message.documents.first)
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 172

def span_attributes(message, connection)
  {
    'db.system.name' => 'mongodb',
    'db.namespace' => database(message),
    'db.collection.name' => collection_name(message),
    'db.command.name' => command_name(message)
  }.compact.merge(connection_attributes(connection))
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 51

def start_span(message, operation_context, connection); end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 67

def trace_command(message, _operation_context, connection, &block)
  return yield if skip_tracing?(message)

  # Commands should always be nested under their operation span, not directly under
  # the transaction span. Don't pass with_parent to use automatic parent resolution
  # from the currently active span (the operation span).
  span = create_command_span(message, connection)
  # With no SDK and no valid parent, the API returns a span with an
  # invalid context. It cannot be made current, recorded or
  # propagated, so it gets no work at all, not even finishing it.
  return yield unless span.context.valid?

  trace_with_span(span, message, &block)
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 95

def trace_with_span(span, message)
  recording = span.recording?
  apply_deferred_attributes(span, message) if recording
  ::OpenTelemetry::Trace.with_span(span) do |s, _c|
    yield.tap do |result|
      process_command_result(result, s) if recording
    end
  end
rescue Exception => e
  handle_command_exception(span, e) if recording
  raise e
ensure
  span.finish
end

# File 'lib/mongo/tracing/open_telemetry/command_tracer.rb', line 337

def txn_number(message)
  txn_num = message.documents.first['txnNumber']
  return unless txn_num

  txn_num.value
end