google-cloud-orchestration-airflow-service-v1 0.6.1 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d522d39356360108b9fc91e5b47f6b3e7638457795bb86bae304162ca8963735
-  data.tar.gz: ebf85b4136b3a64ff2bf706f8cbd0358b431e4592bb93903ca50aacdff0e7185
+  metadata.gz: a83d1c7f7236b009e1afb6acdd1966ee048ab9ef08dd3dd96642084d69c73d3f
+  data.tar.gz: b18f02313e2eadd27b12ca3a4890a148f6d6645409b61d0a4dbd8d870e8eece1
 SHA512:
-  metadata.gz: 2bd8ccba4fa83394fe667aab7fe9e79c981335fbf742e1f915e63b027c896869947c8a7acae78fa45875831d5b1209a96e6163f1090bc8cf1ea613c92f9992b4
-  data.tar.gz: 5d3d0ea5649ecb513ba18b8a96bd5bee72bed17fb2be9b50fb61ac2233ff18e0d1edfd031cf42e63571bf59710b5aba4cbd9a445d9e1da3b900a5d134397bb49
+  metadata.gz: 895d3d84ad12e8e54c1ba3c778b8c498080ef160f4eb97dfd9bb6553f8a8c1fe07d3a5adac53bb176f251a7e8be04692e88ea57d283991b41a6898f014867bd5
+  data.tar.gz: cb5b3fa3ad08cbdbeeaa81fd98800b3903ce0c8c03ebb65cc279e6f6df17d48de5c52444617b841dbcfbe7ea97fc187c71c6a8f1b3ac6ccf93d7861062e3e2b4

data/lib/google/cloud/orchestration/airflow/service/v1/environments/client.rb

@@ -146,7 +146,8 @@ module Google
                     credentials:  credentials,
                     endpoint:     @config.endpoint,
                     channel_args: @config.channel_args,
-                    interceptors: @config.interceptors
+                    interceptors: @config.interceptors,
+                    channel_pool_config: @config.channel_pool
                   )
                 end
 
@@ -751,6 +752,290 @@ module Google
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
+                ##
+                # Executes Airflow CLI command.
+                #
+                # @overload execute_airflow_command(request, options = nil)
+                #   Pass arguments to `execute_airflow_command` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandRequest, ::Hash]
+                #     A request object representing the call parameters. Required. To specify no
+                #     parameters, or to keep all the default parameter values, pass an empty Hash.
+                #   @param options [::Gapic::CallOptions, ::Hash]
+                #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+                #
+                # @overload execute_airflow_command(environment: nil, command: nil, subcommand: nil, parameters: nil)
+                #   Pass arguments to `execute_airflow_command` via keyword arguments. Note that at
+                #   least one keyword argument is required. To specify no parameters, or to keep all
+                #   the default parameter values, pass an empty Hash as a request object (see above).
+                #
+                #   @param environment [::String]
+                #     The resource name of the environment in the form:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}".
+                #   @param command [::String]
+                #     Airflow command.
+                #   @param subcommand [::String]
+                #     Airflow subcommand.
+                #   @param parameters [::Array<::String>]
+                #     Parameters for the Airflow command/subcommand as an array of arguments.
+                #     It may contain positional arguments like `["my-dag-id"]`, key-value
+                #     parameters like `["--foo=bar"]` or `["--foo","bar"]`,
+                #     or other flags like `["-f"]`.
+                #
+                # @yield [response, operation] Access the result along with the RPC operation
+                # @yieldparam response [::Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandResponse]
+                # @yieldparam operation [::GRPC::ActiveCall::Operation]
+                #
+                # @return [::Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandResponse]
+                #
+                # @raise [::Google::Cloud::Error] if the RPC is aborted.
+                #
+                # @example Basic example
+                #   require "google/cloud/orchestration/airflow/service/v1"
+                #
+                #   # Create a client object. The client can be reused for multiple calls.
+                #   client = Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new
+                #
+                #   # Create a request. To set request fields, pass in keyword arguments.
+                #   request = Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandRequest.new
+                #
+                #   # Call the execute_airflow_command method.
+                #   result = client.execute_airflow_command request
+                #
+                #   # The returned object is of type Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandResponse.
+                #   p result
+                #
+                def execute_airflow_command request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::ExecuteAirflowCommandRequest
+
+                  # Converts hash and nil to an options object
+                  options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+                  # Customize the options with defaults
+                  metadata = @config.rpcs.execute_airflow_command.metadata.to_h
+
+                  # Set x-goog-api-client and x-goog-user-project headers
+                  metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                    lib_name: @config.lib_name, lib_version: @config.lib_version,
+                    gapic_version: ::Google::Cloud::Orchestration::Airflow::Service::V1::VERSION
+                  metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                  header_params = {}
+                  if request.environment
+                    header_params["environment"] = request.environment
+                  end
+
+                  request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+                  metadata[:"x-goog-request-params"] ||= request_params_header
+
+                  options.apply_defaults timeout:      @config.rpcs.execute_airflow_command.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.execute_airflow_command.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :execute_airflow_command, request, options: options do |response, operation|
+                    yield response, operation if block_given?
+                    return response
+                  end
+                rescue ::GRPC::BadStatus => e
+                  raise ::Google::Cloud::Error.from_error(e)
+                end
+
+                ##
+                # Stops Airflow CLI command execution.
+                #
+                # @overload stop_airflow_command(request, options = nil)
+                #   Pass arguments to `stop_airflow_command` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandRequest, ::Hash]
+                #     A request object representing the call parameters. Required. To specify no
+                #     parameters, or to keep all the default parameter values, pass an empty Hash.
+                #   @param options [::Gapic::CallOptions, ::Hash]
+                #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+                #
+                # @overload stop_airflow_command(environment: nil, execution_id: nil, pod: nil, pod_namespace: nil, force: nil)
+                #   Pass arguments to `stop_airflow_command` via keyword arguments. Note that at
+                #   least one keyword argument is required. To specify no parameters, or to keep all
+                #   the default parameter values, pass an empty Hash as a request object (see above).
+                #
+                #   @param environment [::String]
+                #     The resource name of the environment in the form:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}".
+                #   @param execution_id [::String]
+                #     The unique ID of the command execution.
+                #   @param pod [::String]
+                #     The name of the pod where the command is executed.
+                #   @param pod_namespace [::String]
+                #     The namespace of the pod where the command is executed.
+                #   @param force [::Boolean]
+                #     If true, the execution is terminated forcefully (SIGKILL). If false, the
+                #     execution is stopped gracefully, giving it time for cleanup.
+                #
+                # @yield [response, operation] Access the result along with the RPC operation
+                # @yieldparam response [::Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandResponse]
+                # @yieldparam operation [::GRPC::ActiveCall::Operation]
+                #
+                # @return [::Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandResponse]
+                #
+                # @raise [::Google::Cloud::Error] if the RPC is aborted.
+                #
+                # @example Basic example
+                #   require "google/cloud/orchestration/airflow/service/v1"
+                #
+                #   # Create a client object. The client can be reused for multiple calls.
+                #   client = Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new
+                #
+                #   # Create a request. To set request fields, pass in keyword arguments.
+                #   request = Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandRequest.new
+                #
+                #   # Call the stop_airflow_command method.
+                #   result = client.stop_airflow_command request
+                #
+                #   # The returned object is of type Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandResponse.
+                #   p result
+                #
+                def stop_airflow_command request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::StopAirflowCommandRequest
+
+                  # Converts hash and nil to an options object
+                  options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+                  # Customize the options with defaults
+                  metadata = @config.rpcs.stop_airflow_command.metadata.to_h
+
+                  # Set x-goog-api-client and x-goog-user-project headers
+                  metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                    lib_name: @config.lib_name, lib_version: @config.lib_version,
+                    gapic_version: ::Google::Cloud::Orchestration::Airflow::Service::V1::VERSION
+                  metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                  header_params = {}
+                  if request.environment
+                    header_params["environment"] = request.environment
+                  end
+
+                  request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+                  metadata[:"x-goog-request-params"] ||= request_params_header
+
+                  options.apply_defaults timeout:      @config.rpcs.stop_airflow_command.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.stop_airflow_command.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :stop_airflow_command, request, options: options do |response, operation|
+                    yield response, operation if block_given?
+                    return response
+                  end
+                rescue ::GRPC::BadStatus => e
+                  raise ::Google::Cloud::Error.from_error(e)
+                end
+
+                ##
+                # Polls Airflow CLI command execution and fetches logs.
+                #
+                # @overload poll_airflow_command(request, options = nil)
+                #   Pass arguments to `poll_airflow_command` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandRequest, ::Hash]
+                #     A request object representing the call parameters. Required. To specify no
+                #     parameters, or to keep all the default parameter values, pass an empty Hash.
+                #   @param options [::Gapic::CallOptions, ::Hash]
+                #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+                #
+                # @overload poll_airflow_command(environment: nil, execution_id: nil, pod: nil, pod_namespace: nil, next_line_number: nil)
+                #   Pass arguments to `poll_airflow_command` via keyword arguments. Note that at
+                #   least one keyword argument is required. To specify no parameters, or to keep all
+                #   the default parameter values, pass an empty Hash as a request object (see above).
+                #
+                #   @param environment [::String]
+                #     The resource name of the environment in the form:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}"
+                #   @param execution_id [::String]
+                #     The unique ID of the command execution.
+                #   @param pod [::String]
+                #     The name of the pod where the command is executed.
+                #   @param pod_namespace [::String]
+                #     The namespace of the pod where the command is executed.
+                #   @param next_line_number [::Integer]
+                #     Line number from which new logs should be fetched.
+                #
+                # @yield [response, operation] Access the result along with the RPC operation
+                # @yieldparam response [::Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandResponse]
+                # @yieldparam operation [::GRPC::ActiveCall::Operation]
+                #
+                # @return [::Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandResponse]
+                #
+                # @raise [::Google::Cloud::Error] if the RPC is aborted.
+                #
+                # @example Basic example
+                #   require "google/cloud/orchestration/airflow/service/v1"
+                #
+                #   # Create a client object. The client can be reused for multiple calls.
+                #   client = Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new
+                #
+                #   # Create a request. To set request fields, pass in keyword arguments.
+                #   request = Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandRequest.new
+                #
+                #   # Call the poll_airflow_command method.
+                #   result = client.poll_airflow_command request
+                #
+                #   # The returned object is of type Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandResponse.
+                #   p result
+                #
+                def poll_airflow_command request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::PollAirflowCommandRequest
+
+                  # Converts hash and nil to an options object
+                  options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+                  # Customize the options with defaults
+                  metadata = @config.rpcs.poll_airflow_command.metadata.to_h
+
+                  # Set x-goog-api-client and x-goog-user-project headers
+                  metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                    lib_name: @config.lib_name, lib_version: @config.lib_version,
+                    gapic_version: ::Google::Cloud::Orchestration::Airflow::Service::V1::VERSION
+                  metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                  header_params = {}
+                  if request.environment
+                    header_params["environment"] = request.environment
+                  end
+
+                  request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+                  metadata[:"x-goog-request-params"] ||= request_params_header
+
+                  options.apply_defaults timeout:      @config.rpcs.poll_airflow_command.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.poll_airflow_command.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :poll_airflow_command, request, options: options do |response, operation|
+                    yield response, operation if block_given?
+                    return response
+                  end
+                rescue ::GRPC::BadStatus => e
+                  raise ::Google::Cloud::Error.from_error(e)
+                end
+
                 ##
                 # Creates a snapshots of a Cloud Composer environment.
                 #
@@ -963,6 +1248,186 @@ module Google
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
+                ##
+                # Triggers database failover (only for highly resilient environments).
+                #
+                # @overload database_failover(request, options = nil)
+                #   Pass arguments to `database_failover` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::DatabaseFailoverRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::DatabaseFailoverRequest, ::Hash]
+                #     A request object representing the call parameters. Required. To specify no
+                #     parameters, or to keep all the default parameter values, pass an empty Hash.
+                #   @param options [::Gapic::CallOptions, ::Hash]
+                #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+                #
+                # @overload database_failover(environment: nil)
+                #   Pass arguments to `database_failover` via keyword arguments. Note that at
+                #   least one keyword argument is required. To specify no parameters, or to keep all
+                #   the default parameter values, pass an empty Hash as a request object (see above).
+                #
+                #   @param environment [::String]
+                #     Target environment:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}"
+                #
+                # @yield [response, operation] Access the result along with the RPC operation
+                # @yieldparam response [::Gapic::Operation]
+                # @yieldparam operation [::GRPC::ActiveCall::Operation]
+                #
+                # @return [::Gapic::Operation]
+                #
+                # @raise [::Google::Cloud::Error] if the RPC is aborted.
+                #
+                # @example Basic example
+                #   require "google/cloud/orchestration/airflow/service/v1"
+                #
+                #   # Create a client object. The client can be reused for multiple calls.
+                #   client = Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new
+                #
+                #   # Create a request. To set request fields, pass in keyword arguments.
+                #   request = Google::Cloud::Orchestration::Airflow::Service::V1::DatabaseFailoverRequest.new
+                #
+                #   # Call the database_failover method.
+                #   result = client.database_failover request
+                #
+                #   # The returned object is of type Gapic::Operation. You can use it to
+                #   # check the status of an operation, cancel it, or wait for results.
+                #   # Here is how to wait for a response.
+                #   result.wait_until_done! timeout: 60
+                #   if result.response?
+                #     p result.response
+                #   else
+                #     puts "No response received."
+                #   end
+                #
+                def database_failover request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::DatabaseFailoverRequest
+
+                  # Converts hash and nil to an options object
+                  options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+                  # Customize the options with defaults
+                  metadata = @config.rpcs.database_failover.metadata.to_h
+
+                  # Set x-goog-api-client and x-goog-user-project headers
+                  metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                    lib_name: @config.lib_name, lib_version: @config.lib_version,
+                    gapic_version: ::Google::Cloud::Orchestration::Airflow::Service::V1::VERSION
+                  metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                  header_params = {}
+                  if request.environment
+                    header_params["environment"] = request.environment
+                  end
+
+                  request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+                  metadata[:"x-goog-request-params"] ||= request_params_header
+
+                  options.apply_defaults timeout:      @config.rpcs.database_failover.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.database_failover.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :database_failover, request, options: options do |response, operation|
+                    response = ::Gapic::Operation.new response, @operations_client, options: options
+                    yield response, operation if block_given?
+                    return response
+                  end
+                rescue ::GRPC::BadStatus => e
+                  raise ::Google::Cloud::Error.from_error(e)
+                end
+
+                ##
+                # Fetches database properties.
+                #
+                # @overload fetch_database_properties(request, options = nil)
+                #   Pass arguments to `fetch_database_properties` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesRequest, ::Hash]
+                #     A request object representing the call parameters. Required. To specify no
+                #     parameters, or to keep all the default parameter values, pass an empty Hash.
+                #   @param options [::Gapic::CallOptions, ::Hash]
+                #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+                #
+                # @overload fetch_database_properties(environment: nil)
+                #   Pass arguments to `fetch_database_properties` via keyword arguments. Note that at
+                #   least one keyword argument is required. To specify no parameters, or to keep all
+                #   the default parameter values, pass an empty Hash as a request object (see above).
+                #
+                #   @param environment [::String]
+                #     Required. The resource name of the environment, in the form:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}"
+                #
+                # @yield [response, operation] Access the result along with the RPC operation
+                # @yieldparam response [::Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesResponse]
+                # @yieldparam operation [::GRPC::ActiveCall::Operation]
+                #
+                # @return [::Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesResponse]
+                #
+                # @raise [::Google::Cloud::Error] if the RPC is aborted.
+                #
+                # @example Basic example
+                #   require "google/cloud/orchestration/airflow/service/v1"
+                #
+                #   # Create a client object. The client can be reused for multiple calls.
+                #   client = Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new
+                #
+                #   # Create a request. To set request fields, pass in keyword arguments.
+                #   request = Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesRequest.new
+                #
+                #   # Call the fetch_database_properties method.
+                #   result = client.fetch_database_properties request
+                #
+                #   # The returned object is of type Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesResponse.
+                #   p result
+                #
+                def fetch_database_properties request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::FetchDatabasePropertiesRequest
+
+                  # Converts hash and nil to an options object
+                  options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+                  # Customize the options with defaults
+                  metadata = @config.rpcs.fetch_database_properties.metadata.to_h
+
+                  # Set x-goog-api-client and x-goog-user-project headers
+                  metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                    lib_name: @config.lib_name, lib_version: @config.lib_version,
+                    gapic_version: ::Google::Cloud::Orchestration::Airflow::Service::V1::VERSION
+                  metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                  header_params = {}
+                  if request.environment
+                    header_params["environment"] = request.environment
+                  end
+
+                  request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+                  metadata[:"x-goog-request-params"] ||= request_params_header
+
+                  options.apply_defaults timeout:      @config.rpcs.fetch_database_properties.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.fetch_database_properties.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :fetch_database_properties, request, options: options do |response, operation|
+                    yield response, operation if block_given?
+                    return response
+                  end
+                rescue ::GRPC::BadStatus => e
+                  raise ::Google::Cloud::Error.from_error(e)
+                end
+
                 ##
                 # Configuration class for the Environments API.
                 #
@@ -1082,6 +1547,14 @@ module Google
                     end
                   end
 
+                  ##
+                  # Configuration for the channel pool
+                  # @return [::Gapic::ServiceStub::ChannelPool::Configuration]
+                  #
+                  def channel_pool
+                    @channel_pool ||= ::Gapic::ServiceStub::ChannelPool::Configuration.new
+                  end
+
                   ##
                   # Configuration RPC class for the Environments API.
                   #
@@ -1126,6 +1599,21 @@ module Google
                     #
                     attr_reader :delete_environment
                     ##
+                    # RPC-specific configuration for `execute_airflow_command`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :execute_airflow_command
+                    ##
+                    # RPC-specific configuration for `stop_airflow_command`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :stop_airflow_command
+                    ##
+                    # RPC-specific configuration for `poll_airflow_command`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :poll_airflow_command
+                    ##
                     # RPC-specific configuration for `save_snapshot`
                     # @return [::Gapic::Config::Method]
                     #
@@ -1135,6 +1623,16 @@ module Google
                     # @return [::Gapic::Config::Method]
                     #
                     attr_reader :load_snapshot
+                    ##
+                    # RPC-specific configuration for `database_failover`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :database_failover
+                    ##
+                    # RPC-specific configuration for `fetch_database_properties`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :fetch_database_properties
 
                     # @private
                     def initialize parent_rpcs = nil
@@ -1148,10 +1646,20 @@ module Google
                       @update_environment = ::Gapic::Config::Method.new update_environment_config
                       delete_environment_config = parent_rpcs.delete_environment if parent_rpcs.respond_to? :delete_environment
                       @delete_environment = ::Gapic::Config::Method.new delete_environment_config
+                      execute_airflow_command_config = parent_rpcs.execute_airflow_command if parent_rpcs.respond_to? :execute_airflow_command
+                      @execute_airflow_command = ::Gapic::Config::Method.new execute_airflow_command_config
+                      stop_airflow_command_config = parent_rpcs.stop_airflow_command if parent_rpcs.respond_to? :stop_airflow_command
+                      @stop_airflow_command = ::Gapic::Config::Method.new stop_airflow_command_config
+                      poll_airflow_command_config = parent_rpcs.poll_airflow_command if parent_rpcs.respond_to? :poll_airflow_command
+                      @poll_airflow_command = ::Gapic::Config::Method.new poll_airflow_command_config
                       save_snapshot_config = parent_rpcs.save_snapshot if parent_rpcs.respond_to? :save_snapshot
                       @save_snapshot = ::Gapic::Config::Method.new save_snapshot_config
                       load_snapshot_config = parent_rpcs.load_snapshot if parent_rpcs.respond_to? :load_snapshot
                       @load_snapshot = ::Gapic::Config::Method.new load_snapshot_config
+                      database_failover_config = parent_rpcs.database_failover if parent_rpcs.respond_to? :database_failover
+                      @database_failover = ::Gapic::Config::Method.new database_failover_config
+                      fetch_database_properties_config = parent_rpcs.fetch_database_properties if parent_rpcs.respond_to? :fetch_database_properties
+                      @fetch_database_properties = ::Gapic::Config::Method.new fetch_database_properties_config
 
                       yield self if block_given?
                     end

data/lib/google/cloud/orchestration/airflow/service/v1/environments/operations.rb

@@ -95,7 +95,8 @@ module Google
                     credentials:  credentials,
                     endpoint:     @config.endpoint,
                     channel_args: @config.channel_args,
-                    interceptors: @config.interceptors
+                    interceptors: @config.interceptors,
+                    channel_pool_config: @config.channel_pool
                   )
 
                   # Used by an LRO wrapper for some methods of this service
@@ -703,6 +704,14 @@ module Google
                     end
                   end
 
+                  ##
+                  # Configuration for the channel pool
+                  # @return [::Gapic::ServiceStub::ChannelPool::Configuration]
+                  #
+                  def channel_pool
+                    @channel_pool ||= ::Gapic::ServiceStub::ChannelPool::Configuration.new
+                  end
+
                   ##
                   # Configuration RPC class for the Operations API.
                   #