command_succeeded.rb in mongo-2.15.0

- old
+ new

@@ -1,5 +1,8 @@
+# frozen_string_literal: true
+# encoding: utf-8
+
 # Copyright (C) 2015-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -41,10 +44,19 @@
         attr_reader :operation_id
 
         # @return [ Integer ] request_id The request id.
         attr_reader :request_id
 
+        # @return [ nil | Object ] The service id, if any.
+        attr_reader :service_id
+
+        # @return [ Monitoring::Event::CommandStarted ] started_event The corresponding
+        #   started event.
+        #
+        # @api private
+        attr_reader :started_event
+
         # Create the new event.
         #
         # @example Create the event.
         #
         # @param [ String ] command_name The name of the command.
@@ -52,19 +64,26 @@
         # @param [ Server::Address ] address The server address.
         # @param [ Integer ] request_id The request id.
         # @param [ Integer ] operation_id The operation id.
         # @param [ BSON::Document ] reply The command reply.
         # @param [ Float ] duration The duration the command took in seconds.
+        # @param [ Monitoring::Event::CommandStarted ] started_event The corresponding
+        #   started event.
+        # @param [ Object ] service_id The service id, if any.
         #
         # @since 2.1.0
         # @api private
-        def initialize(command_name, database_name, address, request_id, operation_id, reply, duration)
+        def initialize(command_name, database_name, address, request_id,
+          operation_id, reply, duration, started_event:, service_id: nil
+        )
           @command_name = command_name.to_s
           @database_name = database_name
           @address = address
           @request_id = request_id
           @operation_id = operation_id
+          @service_id = service_id
+          @started_event = started_event
           @reply = redacted(command_name, reply)
           @duration = duration
         end
 
         # Returns a concise yet useful summary of the event.
@@ -86,23 +105,30 @@
         # @param [ Server::Address ] address The server address.
         # @param [ Integer ] operation_id The operation id.
         # @param [ Hash ] command_payload The command message payload.
         # @param [ Hash ] reply_payload The reply message payload.
         # @param [ Float ] duration The duration of the command in seconds.
+        # @param [ Monitoring::Event::CommandStarted ] started_event The corresponding
+        #   started event.
+        # @param [ Object ] service_id The service id, if any.
         #
         # @return [ CommandCompleted ] The event.
         #
         # @since 2.1.0
         # @api private
-        def self.generate(address, operation_id, command_payload, reply_payload, duration)
+        def self.generate(address, operation_id, command_payload,
+          reply_payload, duration, started_event:, service_id: nil
+        )
           new(
             command_payload[:command_name],
             command_payload[:database_name],
             address,
             command_payload[:request_id],
             operation_id,
             generate_reply(command_payload, reply_payload),
-            duration
+            duration,
+            started_event: started_event,
+            service_id: service_id,
           )
         end
 
         private
 