client.rb in google-cloud-bigquery-migration-v2-0.12.0

- old
+ new

@@ -168,14 +168,34 @@
 
                   @migration_service_stub = ::Google::Cloud::Bigquery::Migration::V2::MigrationService::Rest::ServiceStub.new(
                     endpoint: @config.endpoint,
                     endpoint_template: DEFAULT_ENDPOINT_TEMPLATE,
                     universe_domain: @config.universe_domain,
-                    credentials: credentials
+                    credentials: credentials,
+                    logger: @config.logger
                   )
+
+                  @migration_service_stub.logger(stub: true)&.info do |entry|
+                    entry.set_system_name
+                    entry.set_service
+                    entry.message = "Created client for #{entry.service}"
+                    entry.set_credentials_fields credentials
+                    entry.set "customEndpoint", @config.endpoint if @config.endpoint
+                    entry.set "defaultTimeout", @config.timeout if @config.timeout
+                    entry.set "quotaProject", @quota_project_id if @quota_project_id
+                  end
                 end
 
+                ##
+                # The logger used for request/response debug logging.
+                #
+                # @return [Logger]
+                #
+                def logger
+                  @migration_service_stub.logger
+                end
+
                 # Service calls
 
                 ##
                 # Creates a migration workflow.
                 #
@@ -250,11 +270,10 @@
                                          metadata:     @config.metadata,
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.create_migration_workflow request, options do |result, operation|
                     yield result, operation if block_given?
-                    return result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -332,11 +351,10 @@
                                          metadata:     @config.metadata,
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.get_migration_workflow request, options do |result, operation|
                     yield result, operation if block_given?
-                    return result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -428,11 +446,11 @@
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.list_migration_workflows request, options do |result, operation|
                     result = ::Gapic::Rest::PagedEnumerable.new @migration_service_stub, :list_migration_workflows, "migration_workflows", request, result, options
                     yield result, operation if block_given?
-                    return result
+                    throw :response, result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -508,11 +526,10 @@
                                          metadata:     @config.metadata,
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.delete_migration_workflow request, options do |result, operation|
                     yield result, operation if block_given?
-                    return result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -591,11 +608,10 @@
                                          metadata:     @config.metadata,
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.start_migration_workflow request, options do |result, operation|
                     yield result, operation if block_given?
-                    return result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -673,11 +689,10 @@
                                          metadata:     @config.metadata,
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.get_migration_subtask request, options do |result, operation|
                     yield result, operation if block_given?
-                    return result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -773,11 +788,11 @@
                                          retry_policy: @config.retry_policy
 
                   @migration_service_stub.list_migration_subtasks request, options do |result, operation|
                     result = ::Gapic::Rest::PagedEnumerable.new @migration_service_stub, :list_migration_subtasks, "migration_subtasks", request, result, options
                     yield result, operation if block_given?
-                    return result
+                    throw :response, result
                   end
                 rescue ::Gapic::Rest::Error => e
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
@@ -853,10 +868,15 @@
                 # @!attribute [rw] universe_domain
                 #   The universe domain within which to make requests. This determines the
                 #   default endpoint URL. The default value of nil uses the environment
                 #   universe (usually the default "googleapis.com" universe).
                 #   @return [::String,nil]
+                # @!attribute [rw] logger
+                #   A custom logger to use for request/response debug logging, or the value
+                #   `:default` (the default) to construct a default logger, or `nil` to
+                #   explicitly disable logging.
+                #   @return [::Logger,:default,nil]
                 #
                 class Configuration
                   extend ::Gapic::Config
 
                   # @private
@@ -874,9 +894,10 @@
                   config_attr :timeout,       nil, ::Numeric, nil
                   config_attr :metadata,      nil, ::Hash, nil
                   config_attr :retry_policy,  nil, ::Hash, ::Proc, nil
                   config_attr :quota_project, nil, ::String, nil
                   config_attr :universe_domain, nil, ::String, nil
+                  config_attr :logger, :default, ::Logger, nil, :default
 
                   # @private
                   def initialize parent_config = nil
                     @parent_config = parent_config unless parent_config.nil?
 