feat(BigQuery): Exposing reservation field in JobConfiguration (#31777)

* chore(BigQuery): Adding missing parameters to load_job initializers and updaters

-

* feat(BigQuery): Exposing reservation field in JobConfiguration

Author: cy-yun

google-cloud-bigquery/lib/google/cloud/bigquery/copy_job.rb

@@ -194,7 +194,8 @@ def self.from_options service, source, target, options
               job_reference: job_ref,
               configuration: Google::Apis::BigqueryV2::JobConfiguration.new(
                 copy:    copy_cfg,
-                dry_run: options[:dryrun]
+                dry_run: options[:dryrun],
+                reservation: options[:reservation]
               )
             )
 
@@ -325,6 +326,18 @@ def labels= value
             @gapi.configuration.update! labels: value
           end
 
+          ##
+          # Sets the reservation that job would use. User can specify a reservation
+          # to execute the job. If reservation is not set, reservation is determined
+          # based on the rules defined by the reservation assignments. The expected
+          # format is `projects/`project`/locations/`location`/reservations/`reservation``.
+          # @param [String] value The reservation name.
+          #
+          # @!group Attributes
+          def reservation= value
+            @gapi.configuration.update! reservation: value
+          end
+
           def cancel
             raise "not implemented in #{self.class}"
           end

google-cloud-bigquery/lib/google/cloud/bigquery/dataset.rb

@@ -280,7 +280,8 @@ def self.from_options service, source, storage_files, options
               job_reference: job_ref,
               configuration: Google::Apis::BigqueryV2::JobConfiguration.new(
                 extract: extract_config,
-                dry_run: options[:dryrun]
+                dry_run: options[:dryrun],
+                reservation: options[:reservation]
               )
             )
 
@@ -438,6 +439,18 @@ def use_avro_logical_types= value
             @gapi.configuration.extract.use_avro_logical_types = value
           end
 
+          ##
+          # Sets the reservation that job would use. User can specify a reservation
+          # to execute the job. If reservation is not set, reservation is determined
+          # based on the rules defined by the reservation assignments. The expected
+          # format is `projects/`project`/locations/`location`/reservations/`reservation``.
+          # @param [String] value The reservation name.
+          #
+          # @!group Attributes
+          def reservation= value
+            @gapi.configuration.update! reservation: value
+          end
+
           def cancel
             raise "not implemented in #{self.class}"
           end

google-cloud-bigquery/lib/google/cloud/bigquery/extract_job.rb

@@ -737,7 +737,7 @@ def time_zone
         end
 
         ##
-        # The URI of thereference file with the reader schema. This file is only
+        # The URI of the reference file with the reader schema. This file is only
         # loaded if it is part of source URIs, but is not loaded otherwise.
         # It is enabled for the following formats: `AVRO`, `PARQUET`, `ORC`.
         #
@@ -2779,7 +2779,7 @@ def time_zone= time_zone
 
           ##
 
-          # Sets the URI of thereference file with the reader schema. This file
+          # Sets the URI of the reference file with the reader schema. This file
           # is only loaded if it is part of source URIs, but is not loaded
           # otherwise. It is enabled for the following formats: `AVRO`,
           # `PARQUET`, `ORC`.
@@ -2800,6 +2800,18 @@ def preserve_ascii_control_characters= val
             @gapi.configuration.load.update! preserve_ascii_control_characters: val
           end
 
+          ##
+          # Sets the reservation that job would use. User can specify a reservation
+          # to execute the job. If reservation is not set, reservation is determined
+          # based on the rules defined by the reservation assignments. The expected
+          # format is `projects/`project`/locations/`location`/reservations/`reservation``.
+          # @param [String] value The reservation name.
+          #
+          # @!group Attributes
+          def reservation= value
+            @gapi.configuration.update! reservation: value
+          end
+
           def cancel
             raise "not implemented in #{self.class}"
           end

google-cloud-bigquery/lib/google/cloud/bigquery/load_job.rb

@@ -545,6 +545,11 @@ def training_runs
         #   * The key portion of a label must be unique. However, you can use the
         #     same key with multiple resources.
         #   * Keys must start with a lowercase letter or international character.
+        # @param [String] reservation The reservation that job would use. User
+        #    can specify a reservation to execute the job. If reservation is not
+        #    set, reservation is determined based on the rules defined by the
+        #    reservation assignments. The expected format is
+        #    `projects/`project`/locations/`location`/reservations/`reservation``.
         #
         # @yield [job] a job configuration object
         # @yieldparam [Google::Cloud::Bigquery::ExtractJob::Updater] job a job
@@ -566,9 +571,9 @@ def training_runs
         #
         # @!group Data
         #
-        def extract_job extract_url, format: nil, job_id: nil, prefix: nil, labels: nil
+        def extract_job extract_url, format: nil, job_id: nil, prefix: nil, labels: nil, reservation: nil
           ensure_service!
-          options = { format: format, job_id: job_id, prefix: prefix, labels: labels }
+          options = { format: format, job_id: job_id, prefix: prefix, labels: labels, reservation: reservation }
           updater = ExtractJob::Updater.from_options service, model_ref, extract_url, options
           updater.location = location if location # may be model reference
 
@@ -603,6 +608,12 @@ def extract_job extract_url, format: nil, job_id: nil, prefix: nil, labels: nil
         #
         #   * `ml_tf_saved_model` - TensorFlow SavedModel
         #   * `ml_xgboost_booster` - XGBoost Booster
+        # @param [String] reservation The reservation that job would use. User
+        #    can specify a reservation to execute the job. If reservation is not
+        #    set, reservation is determined based on the rules defined by the
+        #    reservation assignments. The expected format is
+        #    `projects/`project`/locations/`location`/reservations/`reservation``.
+        #
         # @yield [job] a job configuration object
         # @yieldparam [Google::Cloud::Bigquery::ExtractJob::Updater] job a job
         #   configuration object for setting additional options.
@@ -620,8 +631,8 @@ def extract_job extract_url, format: nil, job_id: nil, prefix: nil, labels: nil
         #
         # @!group Data
         #
-        def extract extract_url, format: nil, &block
-          job = extract_job extract_url, format: format, &block
+        def extract extract_url, format: nil, reservation: nil, &block
+          job = extract_job extract_url, format: format, reservation: reservation, &block
           job.wait_until_done!
           ensure_job_succeeded! job
           true