google-cloud-orchestration-airflow-service-v1 0.2.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8c779ef4e48955daae1fbf8fa52fc979b3f4f10efdec8df9c2f35912da327f3
-  data.tar.gz: f51e6e5da1e50b0e889473ba8a80a152ed1e71749a348c92fa3c8d9bf2d3b22a
+  metadata.gz: 7fd74eac5e4a724c6b7e660ede76650d39bc14070b69a99f1d7b6ff4b8bfb5e1
+  data.tar.gz: aa154b38a7c6b7fdce3562f67a7de4e1c7efe58dce8a9d7be603da67a9310c81
 SHA512:
-  metadata.gz: 57c651bca41a62ac151b6491cc03cfe86b8b1811e3afe692d6b812870513bed1119da741dc4a769b8776787dd330046cde4d9aff2026b86637b3bf64ae0891e3
-  data.tar.gz: 82b4d353aa144d664f986b228ba993c01f3e7451369f950eb0d9c76f3ca10cb74a746abad9a1633b6333a9dfdfeff72e3eeb1c61e230e23b70bb052cbd48b732
+  metadata.gz: f71781256e53466e202765f048a81d03ff7f62752edc7188caeffd65f8e6f1a4ad6a7f98e78e5351bf15ecec39cfa139baa628c19f3384c20afd9e0fd2b77357
+  data.tar.gz: b3a5ecb299ff686bbe708d718249aed3b783cdb87e3ed0f72d69c9ccda8430c1088ff2a91f30f321f2386cf634d51a9e0ffd20a2d162b92880aa4c6fe3f19af7

data/AUTHENTICATION.md

@@ -112,7 +112,7 @@ credentials are discovered.
 To configure your system for this, simply:
 
 1. [Download and install the Cloud SDK](https://cloud.google.com/sdk)
-2. Authenticate using OAuth 2.0 `$ gcloud auth login`
+2. Authenticate using OAuth 2.0 `$ gcloud auth application-default login`
 3. Write code as if already authenticated.
 
 **NOTE:** This is _not_ recommended for running in production. The Cloud SDK

data/README.md

@@ -46,7 +46,7 @@ for general usage information.
 ## Enabling Logging
 
 To enable logging for this library, set the logger for the underlying [gRPC](https://github.com/grpc/grpc/tree/master/src/ruby) library.
-The logger that you set may be a Ruby stdlib [`Logger`](https://ruby-doc.org/stdlib/libdoc/logger/rdoc/Logger.html) as shown below,
+The logger that you set may be a Ruby stdlib [`Logger`](https://ruby-doc.org/current/stdlibs/logger/Logger.html) as shown below,
 or a [`Google::Cloud::Logging::Logger`](https://googleapis.dev/ruby/google-cloud-logging/latest)
 that will write logs to [Cloud Logging](https://cloud.google.com/logging/). See [grpc/logconfig.rb](https://github.com/grpc/grpc/blob/master/src/ruby/lib/grpc/logconfig.rb)
 and the gRPC [spec_helper.rb](https://github.com/grpc/grpc/blob/master/src/ruby/spec/spec_helper.rb) for additional information.

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

@@ -549,13 +549,10 @@ module Google
                 #     * `config.nodeCount`
                 #         * Horizontally scale the number of nodes in the environment. An integer
                 #           greater than or equal to 3 must be provided in the `config.nodeCount`
-                #           field.
+                #           field. Supported for Cloud Composer environments in versions
+                #           composer-1.*.*-airflow-*.*.*.
                 #     * `config.webServerNetworkAccessControl`
                 #         * Replace the environment's current `WebServerNetworkAccessControl`.
-                #     * `config.databaseConfig`
-                #         * Replace the environment's current `DatabaseConfig`.
-                #     * `config.webServerConfig`
-                #         * Replace the environment's current `WebServerConfig`.
                 #     * `config.softwareConfig.airflowConfigOverrides`
                 #         * Replace all Apache Airflow config overrides. If a replacement config
                 #           overrides map is not included in `environment`, all config overrides
@@ -573,9 +570,22 @@ module Google
                 #     * `config.softwareConfig.envVariables`
                 #         * Replace all environment variables. If a replacement environment
                 #           variable map is not included in `environment`, all custom environment
-                #           variables  are cleared.
-                #           It is an error to provide both this mask and a mask specifying one or
-                #           more individual environment variables.
+                #           variables are cleared.
+                #     * `config.softwareConfig.schedulerCount`
+                #         * Horizontally scale the number of schedulers in Airflow. A positive
+                #           integer not greater than the number of nodes must be provided in the
+                #           `config.softwareConfig.schedulerCount` field. Supported for Cloud
+                #           Composer environments in versions composer-1.*.*-airflow-2.*.*.
+                #     * `config.databaseConfig.machineType`
+                #         * Cloud SQL machine type used by Airflow database.
+                #           It has to be one of: db-n1-standard-2, db-n1-standard-4,
+                #           db-n1-standard-8 or db-n1-standard-16. Supported for Cloud Composer
+                #           environments in versions composer-1.*.*-airflow-*.*.*.
+                #     * `config.webServerConfig.machineType`
+                #         * Machine type on which Airflow web server is running.
+                #           It has to be one of: composer-n1-webserver-2, composer-n1-webserver-4
+                #           or composer-n1-webserver-8. Supported for Cloud Composer environments
+                #           in versions composer-1.*.*-airflow-*.*.*.
                 #
                 # @yield [response, operation] Access the result along with the RPC operation
                 # @yieldparam response [::Gapic::Operation]
@@ -743,6 +753,218 @@ module Google
                   raise ::Google::Cloud::Error.from_error(e)
                 end
 
+                ##
+                # Creates a snapshots of a Cloud Composer environment.
+                #
+                # As a result of this operation, snapshot of environment's state is stored
+                # in a location specified in the SaveSnapshotRequest.
+                #
+                # @overload save_snapshot(request, options = nil)
+                #   Pass arguments to `save_snapshot` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::SaveSnapshotRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::SaveSnapshotRequest, ::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 save_snapshot(environment: nil, snapshot_location: nil)
+                #   Pass arguments to `save_snapshot` 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 source environment in the form:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}"
+                #   @param snapshot_location [::String]
+                #     Location in a Cloud Storage where the snapshot is going to be stored, e.g.:
+                #     "gs://my-bucket/snapshots".
+                #
+                # @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::SaveSnapshotRequest.new
+                #
+                #   # Call the save_snapshot method.
+                #   result = client.save_snapshot request
+                #
+                #   # The returned object is of type Gapic::Operation. You can use this
+                #   # object to check the status of an operation, cancel it, or wait
+                #   # for results. Here is how to block until completion:
+                #   result.wait_until_done! timeout: 60
+                #   if result.response?
+                #     p result.response
+                #   else
+                #     puts "Error!"
+                #   end
+                #
+                def save_snapshot request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::SaveSnapshotRequest
+
+                  # 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.save_snapshot.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.save_snapshot.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.save_snapshot.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :save_snapshot, 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
+
+                ##
+                # Loads a snapshot of a Cloud Composer environment.
+                #
+                # As a result of this operation, a snapshot of environment's specified in
+                # LoadSnapshotRequest is loaded into the environment.
+                #
+                # @overload load_snapshot(request, options = nil)
+                #   Pass arguments to `load_snapshot` via a request object, either of type
+                #   {::Google::Cloud::Orchestration::Airflow::Service::V1::LoadSnapshotRequest} or an equivalent Hash.
+                #
+                #   @param request [::Google::Cloud::Orchestration::Airflow::Service::V1::LoadSnapshotRequest, ::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 load_snapshot(environment: nil, snapshot_path: nil, skip_pypi_packages_installation: nil, skip_environment_variables_setting: nil, skip_airflow_overrides_setting: nil, skip_gcs_data_copying: nil)
+                #   Pass arguments to `load_snapshot` 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 target environment in the form:
+                #     "projects/\\{projectId}/locations/\\{locationId}/environments/\\{environmentId}"
+                #   @param snapshot_path [::String]
+                #     A Cloud Storage path to a snapshot to load, e.g.:
+                #     "gs://my-bucket/snapshots/project_location_environment_timestamp".
+                #   @param skip_pypi_packages_installation [::Boolean]
+                #     Whether or not to skip installing Pypi packages when loading the
+                #     environment's state.
+                #   @param skip_environment_variables_setting [::Boolean]
+                #     Whether or not to skip setting environment variables when loading the
+                #     environment's state.
+                #   @param skip_airflow_overrides_setting [::Boolean]
+                #     Whether or not to skip setting Airflow overrides when loading the
+                #     environment's state.
+                #   @param skip_gcs_data_copying [::Boolean]
+                #     Whether or not to skip copying Cloud Storage data when loading the
+                #     environment's state.
+                #
+                # @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::LoadSnapshotRequest.new
+                #
+                #   # Call the load_snapshot method.
+                #   result = client.load_snapshot request
+                #
+                #   # The returned object is of type Gapic::Operation. You can use this
+                #   # object to check the status of an operation, cancel it, or wait
+                #   # for results. Here is how to block until completion:
+                #   result.wait_until_done! timeout: 60
+                #   if result.response?
+                #     p result.response
+                #   else
+                #     puts "Error!"
+                #   end
+                #
+                def load_snapshot request, options = nil
+                  raise ::ArgumentError, "request must be provided" if request.nil?
+
+                  request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Orchestration::Airflow::Service::V1::LoadSnapshotRequest
+
+                  # 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.load_snapshot.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.load_snapshot.timeout,
+                                         metadata:     metadata,
+                                         retry_policy: @config.rpcs.load_snapshot.retry_policy
+
+                  options.apply_defaults timeout:      @config.timeout,
+                                         metadata:     @config.metadata,
+                                         retry_policy: @config.retry_policy
+
+                  @environments_stub.call_rpc :load_snapshot, 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
+
                 ##
                 # Configuration class for the Environments API.
                 #
@@ -903,6 +1125,16 @@ module Google
                     # @return [::Gapic::Config::Method]
                     #
                     attr_reader :delete_environment
+                    ##
+                    # RPC-specific configuration for `save_snapshot`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :save_snapshot
+                    ##
+                    # RPC-specific configuration for `load_snapshot`
+                    # @return [::Gapic::Config::Method]
+                    #
+                    attr_reader :load_snapshot
 
                     # @private
                     def initialize parent_rpcs = nil
@@ -916,6 +1148,10 @@ 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
+                      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
 
                       yield self if block_given?
                     end

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

@@ -36,7 +36,7 @@ module Google
             ##
             # Managed Apache Airflow Environments.
             #
-            # To load this service and instantiate a client:
+            # @example Load this service and instantiate a gRPC client
             #
             #     require "google/cloud/orchestration/airflow/service/v1/environments"
             #     client = ::Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new

data/lib/google/cloud/orchestration/airflow/service/v1/environments_pb.rb

@@ -7,7 +7,9 @@ require 'google/api/annotations_pb'
 require 'google/api/client_pb'
 require 'google/api/field_behavior_pb'
 require 'google/api/resource_pb'
+require 'google/cloud/orchestration/airflow/service/v1/operations_pb'
 require 'google/longrunning/operations_pb'
+require 'google/protobuf/empty_pb'
 require 'google/protobuf/field_mask_pb'
 require 'google/protobuf/timestamp_pb'
 
@@ -37,6 +39,23 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :environment, :message, 1, "google.cloud.orchestration.airflow.service.v1.Environment"
       optional :update_mask, :message, 3, "google.protobuf.FieldMask"
     end
+    add_message "google.cloud.orchestration.airflow.service.v1.SaveSnapshotRequest" do
+      optional :environment, :string, 1
+      optional :snapshot_location, :string, 2
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.SaveSnapshotResponse" do
+      optional :snapshot_path, :string, 1
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.LoadSnapshotRequest" do
+      optional :environment, :string, 1
+      optional :snapshot_path, :string, 2
+      optional :skip_pypi_packages_installation, :bool, 3
+      optional :skip_environment_variables_setting, :bool, 4
+      optional :skip_airflow_overrides_setting, :bool, 5
+      optional :skip_gcs_data_copying, :bool, 6
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.LoadSnapshotResponse" do
+    end
     add_message "google.cloud.orchestration.airflow.service.v1.EnvironmentConfig" do
       optional :gke_cluster, :string, 1
       optional :dag_gcs_prefix, :string, 2
@@ -48,7 +67,18 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :database_config, :message, 9, "google.cloud.orchestration.airflow.service.v1.DatabaseConfig"
       optional :web_server_config, :message, 10, "google.cloud.orchestration.airflow.service.v1.WebServerConfig"
       optional :encryption_config, :message, 11, "google.cloud.orchestration.airflow.service.v1.EncryptionConfig"
+      optional :maintenance_window, :message, 12, "google.cloud.orchestration.airflow.service.v1.MaintenanceWindow"
+      optional :workloads_config, :message, 15, "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig"
+      optional :environment_size, :enum, 16, "google.cloud.orchestration.airflow.service.v1.EnvironmentConfig.EnvironmentSize"
       optional :airflow_uri, :string, 6
+      optional :master_authorized_networks_config, :message, 17, "google.cloud.orchestration.airflow.service.v1.MasterAuthorizedNetworksConfig"
+      optional :recovery_config, :message, 18, "google.cloud.orchestration.airflow.service.v1.RecoveryConfig"
+    end
+    add_enum "google.cloud.orchestration.airflow.service.v1.EnvironmentConfig.EnvironmentSize" do
+      value :ENVIRONMENT_SIZE_UNSPECIFIED, 0
+      value :ENVIRONMENT_SIZE_SMALL, 1
+      value :ENVIRONMENT_SIZE_MEDIUM, 2
+      value :ENVIRONMENT_SIZE_LARGE, 3
     end
     add_message "google.cloud.orchestration.airflow.service.v1.WebServerNetworkAccessControl" do
       repeated :allowed_ip_ranges, :message, 1, "google.cloud.orchestration.airflow.service.v1.WebServerNetworkAccessControl.AllowedIpRange"
@@ -66,12 +96,18 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "google.cloud.orchestration.airflow.service.v1.EncryptionConfig" do
       optional :kms_key_name, :string, 1
     end
+    add_message "google.cloud.orchestration.airflow.service.v1.MaintenanceWindow" do
+      optional :start_time, :message, 1, "google.protobuf.Timestamp"
+      optional :end_time, :message, 2, "google.protobuf.Timestamp"
+      optional :recurrence, :string, 3
+    end
     add_message "google.cloud.orchestration.airflow.service.v1.SoftwareConfig" do
       optional :image_version, :string, 1
       map :airflow_config_overrides, :string, :string, 2
       map :pypi_packages, :string, :string, 3
       map :env_variables, :string, :string, 4
       optional :python_version, :string, 6
+      optional :scheduler_count, :int32, 7
     end
     add_message "google.cloud.orchestration.airflow.service.v1.IPAllocationPolicy" do
       optional :use_ip_aliases, :bool, 1
@@ -94,18 +130,72 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :service_account, :string, 7
       repeated :tags, :string, 8
       optional :ip_allocation_policy, :message, 9, "google.cloud.orchestration.airflow.service.v1.IPAllocationPolicy"
+      optional :enable_ip_masq_agent, :bool, 11
     end
     add_message "google.cloud.orchestration.airflow.service.v1.PrivateClusterConfig" do
       optional :enable_private_endpoint, :bool, 1
       optional :master_ipv4_cidr_block, :string, 2
       optional :master_ipv4_reserved_range, :string, 3
     end
+    add_message "google.cloud.orchestration.airflow.service.v1.NetworkingConfig" do
+      optional :connection_type, :enum, 1, "google.cloud.orchestration.airflow.service.v1.NetworkingConfig.ConnectionType"
+    end
+    add_enum "google.cloud.orchestration.airflow.service.v1.NetworkingConfig.ConnectionType" do
+      value :CONNECTION_TYPE_UNSPECIFIED, 0
+      value :VPC_PEERING, 1
+      value :PRIVATE_SERVICE_CONNECT, 2
+    end
     add_message "google.cloud.orchestration.airflow.service.v1.PrivateEnvironmentConfig" do
       optional :enable_private_environment, :bool, 1
       optional :private_cluster_config, :message, 2, "google.cloud.orchestration.airflow.service.v1.PrivateClusterConfig"
       optional :web_server_ipv4_cidr_block, :string, 3
       optional :cloud_sql_ipv4_cidr_block, :string, 4
       optional :web_server_ipv4_reserved_range, :string, 5
+      optional :cloud_composer_network_ipv4_cidr_block, :string, 7
+      optional :cloud_composer_network_ipv4_reserved_range, :string, 8
+      optional :enable_privately_used_public_ips, :bool, 6
+      optional :cloud_composer_connection_subnetwork, :string, 9
+      optional :networking_config, :message, 10, "google.cloud.orchestration.airflow.service.v1.NetworkingConfig"
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig" do
+      optional :scheduler, :message, 1, "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.SchedulerResource"
+      optional :web_server, :message, 2, "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.WebServerResource"
+      optional :worker, :message, 3, "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.WorkerResource"
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.SchedulerResource" do
+      optional :cpu, :float, 1
+      optional :memory_gb, :float, 2
+      optional :storage_gb, :float, 3
+      optional :count, :int32, 4
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.WebServerResource" do
+      optional :cpu, :float, 1
+      optional :memory_gb, :float, 2
+      optional :storage_gb, :float, 3
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.WorkerResource" do
+      optional :cpu, :float, 1
+      optional :memory_gb, :float, 2
+      optional :storage_gb, :float, 3
+      optional :min_count, :int32, 4
+      optional :max_count, :int32, 5
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.RecoveryConfig" do
+      optional :scheduled_snapshots_config, :message, 1, "google.cloud.orchestration.airflow.service.v1.ScheduledSnapshotsConfig"
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.ScheduledSnapshotsConfig" do
+      optional :enabled, :bool, 1
+      optional :snapshot_location, :string, 6
+      optional :snapshot_creation_schedule, :string, 3
+      optional :time_zone, :string, 5
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.MasterAuthorizedNetworksConfig" do
+      optional :enabled, :bool, 1
+      repeated :cidr_blocks, :message, 2, "google.cloud.orchestration.airflow.service.v1.MasterAuthorizedNetworksConfig.CidrBlock"
+    end
+    add_message "google.cloud.orchestration.airflow.service.v1.MasterAuthorizedNetworksConfig.CidrBlock" do
+      optional :display_name, :string, 1
+      optional :cidr_block, :string, 2
     end
     add_message "google.cloud.orchestration.airflow.service.v1.Environment" do
       optional :name, :string, 1
@@ -151,17 +241,33 @@ module Google
             ListEnvironmentsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.ListEnvironmentsResponse").msgclass
             DeleteEnvironmentRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.DeleteEnvironmentRequest").msgclass
             UpdateEnvironmentRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.UpdateEnvironmentRequest").msgclass
+            SaveSnapshotRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.SaveSnapshotRequest").msgclass
+            SaveSnapshotResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.SaveSnapshotResponse").msgclass
+            LoadSnapshotRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.LoadSnapshotRequest").msgclass
+            LoadSnapshotResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.LoadSnapshotResponse").msgclass
             EnvironmentConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.EnvironmentConfig").msgclass
+            EnvironmentConfig::EnvironmentSize = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.EnvironmentConfig.EnvironmentSize").enummodule
             WebServerNetworkAccessControl = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WebServerNetworkAccessControl").msgclass
             WebServerNetworkAccessControl::AllowedIpRange = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WebServerNetworkAccessControl.AllowedIpRange").msgclass
             DatabaseConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.DatabaseConfig").msgclass
             WebServerConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WebServerConfig").msgclass
             EncryptionConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.EncryptionConfig").msgclass
+            MaintenanceWindow = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.MaintenanceWindow").msgclass
             SoftwareConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.SoftwareConfig").msgclass
             IPAllocationPolicy = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.IPAllocationPolicy").msgclass
             NodeConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.NodeConfig").msgclass
             PrivateClusterConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.PrivateClusterConfig").msgclass
+            NetworkingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.NetworkingConfig").msgclass
+            NetworkingConfig::ConnectionType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.NetworkingConfig.ConnectionType").enummodule
             PrivateEnvironmentConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.PrivateEnvironmentConfig").msgclass
+            WorkloadsConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WorkloadsConfig").msgclass
+            WorkloadsConfig::SchedulerResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.SchedulerResource").msgclass
+            WorkloadsConfig::WebServerResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.WebServerResource").msgclass
+            WorkloadsConfig::WorkerResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.WorkloadsConfig.WorkerResource").msgclass
+            RecoveryConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.RecoveryConfig").msgclass
+            ScheduledSnapshotsConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.ScheduledSnapshotsConfig").msgclass
+            MasterAuthorizedNetworksConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.MasterAuthorizedNetworksConfig").msgclass
+            MasterAuthorizedNetworksConfig::CidrBlock = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.MasterAuthorizedNetworksConfig.CidrBlock").msgclass
             Environment = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.Environment").msgclass
             Environment::State = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.Environment.State").enummodule
             CheckUpgradeResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.orchestration.airflow.service.v1.CheckUpgradeResponse").msgclass

data/lib/google/cloud/orchestration/airflow/service/v1/environments_services_pb.rb

@@ -45,6 +45,16 @@ module Google
                 rpc :UpdateEnvironment, ::Google::Cloud::Orchestration::Airflow::Service::V1::UpdateEnvironmentRequest, ::Google::Longrunning::Operation
                 # Delete an environment.
                 rpc :DeleteEnvironment, ::Google::Cloud::Orchestration::Airflow::Service::V1::DeleteEnvironmentRequest, ::Google::Longrunning::Operation
+                # Creates a snapshots of a Cloud Composer environment.
+                #
+                # As a result of this operation, snapshot of environment's state is stored
+                # in a location specified in the SaveSnapshotRequest.
+                rpc :SaveSnapshot, ::Google::Cloud::Orchestration::Airflow::Service::V1::SaveSnapshotRequest, ::Google::Longrunning::Operation
+                # Loads a snapshot of a Cloud Composer environment.
+                #
+                # As a result of this operation, a snapshot of environment's specified in
+                # LoadSnapshotRequest is loaded into the environment.
+                rpc :LoadSnapshot, ::Google::Cloud::Orchestration::Airflow::Service::V1::LoadSnapshotRequest, ::Google::Longrunning::Operation
               end
 
               Stub = Service.rpc_stub_class

data/lib/google/cloud/orchestration/airflow/service/v1/image_versions.rb

@@ -34,7 +34,7 @@ module Google
             ##
             # Readonly service to query available ImageVersions.
             #
-            # To load this service and instantiate a client:
+            # @example Load this service and instantiate a gRPC client
             #
             #     require "google/cloud/orchestration/airflow/service/v1/image_versions"
             #     client = ::Google::Cloud::Orchestration::Airflow::Service::V1::ImageVersions::Client.new

data/lib/google/cloud/orchestration/airflow/service/v1/operations_pb.rb

@@ -4,7 +4,6 @@
 require 'google/protobuf'
 
 require 'google/protobuf/timestamp_pb'
-require 'google/api/annotations_pb'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/cloud/orchestration/airflow/service/v1/operations.proto", :syntax => :proto3) do
@@ -30,6 +29,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :DELETE, 2
       value :UPDATE, 3
       value :CHECK, 4
+      value :SAVE_SNAPSHOT, 5
+      value :LOAD_SNAPSHOT, 6
     end
   end
 end

data/lib/google/cloud/orchestration/airflow/service/v1/version.rb

@@ -23,7 +23,7 @@ module Google
       module Airflow
         module Service
           module V1
-            VERSION = "0.2.0"
+            VERSION = "0.3.1"
           end
         end
       end

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

@@ -26,9 +26,9 @@ module Google
       module Airflow
         module Service
           ##
-          # To load this package, including all its services, and instantiate a client:
+          # API client module.
           #
-          # @example
+          # @example Load this package, including all its services, and instantiate a gRPC client
           #
           #     require "google/cloud/orchestration/airflow/service/v1"
           #     client = ::Google::Cloud::Orchestration::Airflow::Service::V1::Environments::Client.new