google-cloud-bigquery 1.52.1 → 1.61.1

data/lib/google/cloud/bigquery/project.rb

@@ -178,6 +178,12 @@ module Google
         #   * 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::CopyJob::Updater] job a job
         #   configuration object for setting additional options.
@@ -199,9 +205,11 @@ module Google
         #
         # @!group Data
         #
-        def copy_job source_table, destination_table, create: nil, write: nil, job_id: nil, prefix: nil, labels: nil
+        def copy_job source_table, destination_table, create: nil, write: nil, job_id: nil, prefix: nil, labels: nil,
+                     reservation: nil
           ensure_service!
-          options = { create: create, write: write, labels: labels, job_id: job_id, prefix: prefix }
+          options = { create: create, write: write, labels: labels, job_id: job_id, prefix: prefix,
+            reservation: reservation }
 
           updater = CopyJob::Updater.from_options(
             service,
@@ -258,6 +266,12 @@ module Google
         #   * `append` - BigQuery appends the data to the table.
         #   * `empty` - An error will be returned if the destination table
         #     already contains data.
+        # @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::CopyJob::Updater] job a job
         #   configuration object for setting additional options.
@@ -276,8 +290,8 @@ module Google
         #
         # @!group Data
         #
-        def copy source_table, destination_table, create: nil, write: nil, &block
-          job = copy_job source_table, destination_table, create: create, write: write, &block
+        def copy source_table, destination_table, create: nil, write: nil, reservation: nil, &block
+          job = copy_job source_table, destination_table, create: create, write: write, reservation: reservation, &block
           job.wait_until_done!
           ensure_job_succeeded! job
           true
@@ -479,6 +493,12 @@ module Google
         #   The default value is false.
         # @param [String] session_id The ID of an existing session. See also the
         #   `create_session` param and {Job#session_id}.
+        # @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::QueryJob::Updater] job a job
         #   configuration object for setting query options.
@@ -631,7 +651,8 @@ module Google
                       labels: nil,
                       udfs: nil,
                       create_session: nil,
-                      session_id: nil
+                      session_id: nil,
+                      reservation: nil
           ensure_service!
           project ||= self.project
           options = {
@@ -657,7 +678,8 @@ module Google
             labels: labels,
             udfs: udfs,
             create_session: create_session,
-            session_id: session_id
+            session_id: session_id,
+            reservation: reservation
           }
 
           updater = QueryJob::Updater.from_options service, query, options
@@ -783,6 +805,14 @@ module Google
         #   `flatten` is false. Optional. The default value is false.
         # @param [String] session_id The ID of an existing session. See the
         #   `create_session` param in {#query_job} and {Job#session_id}.
+        # @param [Boolean] format_options_use_int64_timestamp Output timestamp
+        #   as usec int64. Default is true.
+        # @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::QueryJob::Updater] job a job
         #   configuration object for setting additional options for the query.
@@ -929,6 +959,8 @@ module Google
                   standard_sql: nil,
                   legacy_sql: nil,
                   session_id: nil,
+                  format_options_use_int64_timestamp: true,
+                  reservation: nil,
                   &block
           job = query_job query,
                           params: params,
@@ -940,6 +972,7 @@ module Google
                           standard_sql: standard_sql,
                           legacy_sql: legacy_sql,
                           session_id: session_id,
+                          reservation: reservation,
                           &block
           job.wait_until_done!
 
@@ -953,7 +986,7 @@ module Google
             end
           end
 
-          job.data max: max
+          job.data max: max, format_options_use_int64_timestamp: format_options_use_int64_timestamp
         end
 
         ##
@@ -1112,6 +1145,56 @@ module Google
         #         else the property will be ignored by the backend.
         # @param [string] session_id Session ID in which the load job must run.
         # @param [string] project_id Project ID where the destination table exists.
+        # @param [Boolean] dryrun  If set, don't actually run this job. Behavior
+        #   is undefined however for non-query jobs and may result in an error.
+        #   Deprecated.
+        # @param [String] date_format Format used to parse DATE values.
+        #   Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [String] datetime_format Format used to parse DATETIME
+        #   values. Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [String] time_format Format used to parse TIME values.
+        #   Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [String] timestamp_format Format used to parse
+        #   TIMESTAMP values. Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [Array<String>] null_markers A list of strings represented as
+        #   SQL NULL value in a CSV file. null_marker and null_markers can't be
+        #   set at the same time. If null_marker is set, null_markers has to be
+        #   not set. If null_markers is set, null_marker has to be not set. If
+        #   both null_marker and null_markers are set at the same time, a user
+        #   error would be thrown. Any strings listed in null_markers, including
+        #   empty string would be interpreted as SQL NULL. This applies to all
+        #   column types.
+        # @param [String] source_column_match Controls the strategy used to
+        #   match loaded columns to the schema. If not set, a sensible default is
+        #   chosen based on how the schema is provided. If autodetect is used,
+        #   then columns are matched by name. Otherwise, columns are matched by
+        #   position. This is done to keep the behavior backward-compatible.
+        #
+        #   Acceptable values are:
+        #   * `POSITION` - matches by position. This assumes that the columns are
+        #     ordered the same way as the schema.
+        #   * `NAME` - matches by name. This reads the header row as column names
+        #     and reorders columns to match the field names in the schema.
+        # @param [String] time_zone The time zone used when parsing timestamp
+        #   values.
+        # @param [String] reference_file_schema_uri 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`.
+        # @param [Boolean] preserve_ascii_control_characters When source_format
+        #   is set to `CSV`, indicates if the embedded ASCII control characters
+        #   (the first 32 characters in the ASCII-table, from `\x00` to `\x1F`)
+        #   are preserved. By default, ASCII control characters are not
+        #   preserved.
+        # @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 [updater] A block for setting the schema and other
         #   options for the destination table. The schema can be omitted if the
@@ -1119,9 +1202,6 @@ module Google
         #   Google Cloud Datastore backup.
         # @yieldparam [Google::Cloud::Bigquery::LoadJob::Updater] updater An
         #   updater to modify the load job and its schema.
-        # @param [Boolean] dryrun  If set, don't actually run this job. Behavior
-        #   is undefined however for non-query jobs and may result in an error.
-        #   Deprecated.
         #
         # @return [Google::Cloud::Bigquery::LoadJob] A new load job object.
         #
@@ -1139,7 +1219,10 @@ module Google
                      projection_fields: nil, jagged_rows: nil, quoted_newlines: nil, encoding: nil,
                      delimiter: nil, ignore_unknown: nil, max_bad_records: nil, quote: nil,
                      skip_leading: nil, schema: nil, job_id: nil, prefix: nil, labels: nil, autodetect: nil,
-                     null_marker: nil, dryrun: nil, create_session: nil, session_id: nil, project_id: nil, &block
+                     null_marker: nil, dryrun: nil, create_session: nil, session_id: nil, project_id: nil,
+                     date_format: nil, datetime_format: nil, time_format: nil, timestamp_format: nil,
+                     null_markers: nil, source_column_match: nil, time_zone: nil, reference_file_schema_uri: nil,
+                     preserve_ascii_control_characters: nil, reservation: nil, &block
           ensure_service!
           dataset_id ||= "_SESSION" unless create_session.nil? && session_id.nil?
           session_dataset = dataset dataset_id, skip_lookup: true, project_id: project_id
@@ -1151,7 +1234,12 @@ module Google
                           max_bad_records: max_bad_records, quote: quote, skip_leading: skip_leading,
                           dryrun: dryrun, schema: schema, job_id: job_id, prefix: prefix, labels: labels,
                           autodetect: autodetect, null_marker: null_marker, create_session: create_session,
-                          session_id: session_id, &block
+                          session_id: session_id, date_format: date_format, datetime_format: datetime_format,
+                          time_format: time_format, timestamp_format: timestamp_format,
+                          null_markers: null_markers, source_column_match: source_column_match,
+                          time_zone: time_zone, reference_file_schema_uri: reference_file_schema_uri,
+                          preserve_ascii_control_characters: preserve_ascii_control_characters,
+                          reservation: reservation, &block
         end
 
         ##
@@ -1273,6 +1361,53 @@ module Google
         #   this option. Also note that for most use cases, the block yielded by
         #   this method is a more convenient way to configure the schema.
         # @param [string] session_id Session ID in which the load job must run.
+        # @param [String] date_format Format used to parse DATE values.
+        #   Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [String] datetime_format Format used to parse DATETIME
+        #   values. Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [String] time_format Format used to parse TIME values.
+        #   Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [String] timestamp_format Format used to parse
+        #   TIMESTAMP values. Supports SQL-style format strings. See
+        #   [date and time formatting guide](https://cloud.google.com/bigquery/docs/reference/standard-sql/format-elements#format_date_time_as_string)
+        # @param [Array<String>] null_markers A list of strings represented as
+        #   SQL NULL value in a CSV file. null_marker and null_markers can't be
+        #   set at the same time. If null_marker is set, null_markers has to be
+        #   not set. If null_markers is set, null_marker has to be not set. If
+        #   both null_marker and null_markers are set at the same time, a user
+        #   error would be thrown. Any strings listed in null_markers, including
+        #   empty string would be interpreted as SQL NULL. This applies to all
+        #   column types.
+        # @param [String] source_column_match Controls the strategy used to
+        #   match loaded columns to the schema. If not set, a sensible default is
+        #   chosen based on how the schema is provided. If autodetect is used,
+        #   then columns are matched by name. Otherwise, columns are matched by
+        #   position. This is done to keep the behavior backward-compatible.
+        #
+        #   Acceptable values are:
+        #   * `POSITION` - matches by position. This assumes that the columns are
+        #     ordered the same way as the schema.
+        #   * `NAME` - matches by name. This reads the header row as column names
+        #     and reorders columns to match the field names in the schema.
+        # @param [String] time_zone The time zone used when parsing timestamp
+        #   values.
+        # @param [String] reference_file_schema_uri 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`.
+        # @param [Boolean] preserve_ascii_control_characters When source_format
+        #   is set to `CSV`, indicates if the embedded ASCII control characters
+        #   (the first 32 characters in the ASCII-table, from `\x00` to `\x1F`)
+        #   are preserved. By default, ASCII control characters are not
+        #   preserved.
+        # @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 [updater] A block for setting the schema of the destination
         #   table and other options for the load job. The schema can be omitted
@@ -1302,13 +1437,22 @@ module Google
         def load table_id, files, dataset_id: "_SESSION", format: nil, create: nil, write: nil,
                  projection_fields: nil, jagged_rows: nil, quoted_newlines: nil, encoding: nil,
                  delimiter: nil, ignore_unknown: nil, max_bad_records: nil, quote: nil,
-                 skip_leading: nil, schema: nil, autodetect: nil, null_marker: nil, session_id: nil, &block
+                 skip_leading: nil, schema: nil, autodetect: nil, null_marker: nil, session_id: nil,
+                 date_format: nil, datetime_format: nil, time_format: nil, timestamp_format: nil,
+                 null_markers: nil, source_column_match: nil, time_zone: nil, reference_file_schema_uri: nil,
+                 preserve_ascii_control_characters: nil, reservation: nil, &block
           job = load_job table_id, files, dataset_id: dataset_id,
                         format: format, create: create, write: write, projection_fields: projection_fields,
                         jagged_rows: jagged_rows, quoted_newlines: quoted_newlines, encoding: encoding,
                         delimiter: delimiter, ignore_unknown: ignore_unknown, max_bad_records: max_bad_records,
                         quote: quote, skip_leading: skip_leading, schema: schema, autodetect: autodetect,
-                        null_marker: null_marker, session_id: session_id, &block
+                        null_marker: null_marker, session_id: session_id, date_format: date_format,
+                        datetime_format: datetime_format, time_format: time_format,
+                        timestamp_format: timestamp_format, null_markers: null_markers,
+                        source_column_match: source_column_match, time_zone: time_zone,
+                        reference_file_schema_uri: reference_file_schema_uri,
+                        preserve_ascii_control_characters: preserve_ascii_control_characters, reservation: reservation,
+                         &block
 
           job.wait_until_done!
           ensure_job_succeeded! job
@@ -1378,6 +1522,27 @@ module Google
         #   service. Calls made on this object will raise errors if the resource
         #   does not exist. Default is `false`. Optional.
         # @param [String] project_id The GCP Project where the dataset lives.
+        # @param [Integer] access_policy_version Optional. The version of the
+        #   provided access policy schema. Valid values are `0`, `1`, and `3`.
+        #   Requests specifying an invalid value will be rejected. This
+        #   version refers to the schema version of the access policy and not
+        #   the version of access policy. This field's value can be equal or
+        #   more than the access policy schema provided in the request. For
+        #   example, requests with conditional access policy binding in datasets
+        #   must specify version `3`. But dataset with no conditional role
+        #   bindings in access policy may specify any valid value or leave the
+        #   field unset. If unset or if `0` or `1` value is used for dataset with
+        #   conditional bindings, request will be rejected. This field will be
+        #   mapped to
+        #   [IAM Policy version](https://cloud.google.com/iam/docs/policies#versions)
+        #   and will be used to set policy in IAM.
+        # @param [String] dataset_view The dataset_view parameter is an optional
+        #   field in the GetDatasetRequest used to specify which information
+        #   about a BigQuery dataset should be returned in the response. By
+        #   controlling this parameter, users can request a partial or full
+        #   response, which helps enforce fine-grained access control based on
+        #   their permissions. {Google::Cloud::Bigquery::DatasetView} provides
+        #   constants for this parameter.
         #
         # @return [Google::Cloud::Bigquery::Dataset, nil] Returns `nil` if the
         #   dataset does not exist.
@@ -1405,12 +1570,13 @@ module Google
         #
         #   dataset = bigquery.dataset "my_dataset", skip_lookup: true
         #
-        def dataset dataset_id, skip_lookup: nil, project_id: nil
+        def dataset dataset_id, skip_lookup: nil, project_id: nil, access_policy_version: nil, dataset_view: nil
           ensure_service!
           project_id ||= project
           return Dataset.new_reference project_id, dataset_id, service if skip_lookup
-          gapi = service.get_project_dataset project_id, dataset_id
-          Dataset.from_gapi gapi, service
+          gapi = service.get_project_dataset project_id, dataset_id, access_policy_version: access_policy_version,
+dataset_view: dataset_view
+          Dataset.from_gapi gapi, service, access_policy_version: access_policy_version, dataset_view: dataset_view
         rescue Google::Cloud::NotFoundError
           nil
         end
@@ -1429,6 +1595,28 @@ module Google
         # @param [String] location The geographic location where the dataset
         #   should reside. Possible values include `EU` and `US`. The default
         #   value is `US`.
+        # @param [Integer] access_policy_version Optional. The version of the
+        #   provided access policy schema. Valid values are `0`, `1`, and `3`.
+        #   Requests specifying an invalid value will be rejected. This
+        #   version refers to the schema version of the access policy and not
+        #   the version of access policy. This field's value can be equal or
+        #   more than the access policy schema provided in the request. For
+        #   example, requests with conditional access policy binding in datasets
+        #   must specify version `3`. But dataset with no conditional role
+        #   bindings in access policy may specify any valid value or leave the
+        #   field unset. If unset or if `0` or `1` value is used for dataset with
+        #   conditional bindings, request will be rejected. This field will be
+        #   mapped to
+        #   [IAM Policy version](https://cloud.google.com/iam/docs/policies#versions)
+        #   and will be used to set policy in IAM.
+        # @param [String] dataset_view The dataset_view parameter is an optional
+        #   field in the GetDatasetRequest used to specify which information
+        #   about a BigQuery dataset should be returned in the response. By
+        #   controlling this parameter, users can request a partial or full
+        #   response, which helps enforce fine-grained access control based on
+        #   their permissions. {Google::Cloud::Bigquery::DatasetView} provides
+        #   constants for this parameter.
+        #
         # @yield [access] a block for setting rules
         # @yieldparam [Google::Cloud::Bigquery::Dataset] access the object
         #   accepting rules
@@ -1461,7 +1649,8 @@ module Google
         #   end
         #
         def create_dataset dataset_id, name: nil, description: nil,
-                           expiration: nil, location: nil
+                           expiration: nil, location: nil, access_policy_version: nil,
+                           dataset_view: nil
           ensure_service!
 
           new_ds = Google::Apis::BigqueryV2::Dataset.new(
@@ -1484,8 +1673,8 @@ module Google
             updater.check_for_mutated_access!
           end
 
-          gapi = service.insert_dataset new_ds
-          Dataset.from_gapi gapi, service
+          gapi = service.insert_dataset new_ds, access_policy_version: access_policy_version
+          Dataset.from_gapi gapi, service, access_policy_version: access_policy_version, dataset_view: dataset_view
         end
 
         ##
@@ -1983,6 +2172,12 @@ module Google
         #   * 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
         #   configuration object for setting additional options.
@@ -2011,10 +2206,10 @@ module Google
         # @!group Data
         #
         def extract_job source, extract_url, format: nil, compression: nil, delimiter: nil, header: nil, job_id: nil,
-                        prefix: nil, labels: nil
+                        prefix: nil, labels: nil, reservation: nil
           ensure_service!
           options = { format: format, compression: compression, delimiter: delimiter, header: header, job_id: job_id,
-                      prefix: prefix, labels: labels }
+                      prefix: prefix, labels: labels, reservation: reservation }
           source_ref = if source.respond_to? :model_ref
                          source.model_ref
                        else
@@ -2081,6 +2276,12 @@ module Google
         #   models.
         # @param [Boolean] header Whether to print out a header row in table
         #   exports. Default is `true`. Not applicable when extracting models.
+        # @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.
@@ -2106,12 +2307,14 @@ module Google
         #
         # @!group Data
         #
-        def extract source, extract_url, format: nil, compression: nil, delimiter: nil, header: nil, &block
+        def extract source, extract_url, format: nil, compression: nil, delimiter: nil, header: nil, reservation: nil,
+                    &block
           job = extract_job source, extract_url,
                             format:      format,
                             compression: compression,
                             delimiter:   delimiter,
                             header:      header,
+                            reservation: reservation,
                             &block
           job.wait_until_done!
           ensure_job_succeeded! job

data/lib/google/cloud/bigquery/query_job.rb

@@ -723,6 +723,8 @@ module Google
         #   identifying the result set.
         # @param [Integer] max Maximum number of results to return.
         # @param [Integer] start Zero-based index of the starting row to read.
+        # @param [Boolean] format_options_use_int64_timestamp Output timestamp
+        #   as usec int64. Default is true.
         #
         # @return [Google::Cloud::Bigquery::Data] An object providing access to
         #   data read from the destination table for the job.
@@ -745,20 +747,21 @@ module Google
         #   # Retrieve the next page of results
         #   data = data.next if data.next?
         #
-        def data token: nil, max: nil, start: nil
+        def data token: nil, max: nil, start: nil, format_options_use_int64_timestamp: true
           return nil unless done?
-          return Data.from_gapi_json({ rows: [] }, nil, @gapi, service) if dryrun?
+          return Data.from_gapi_json({ rows: [] }, nil, @gapi, service, format_options_use_int64_timestamp) if dryrun?
           if ddl? || dml? || !ensure_schema!
             data_hash = { totalRows: nil, rows: [] }
-            return Data.from_gapi_json data_hash, nil, @gapi, service
+            return Data.from_gapi_json data_hash, nil, @gapi, service, format_options_use_int64_timestamp
           end
 
           data_hash = service.list_tabledata destination_table_dataset_id,
                                              destination_table_table_id,
                                              token: token,
                                              max: max,
-                                             start: start
-          Data.from_gapi_json data_hash, destination_table_gapi, @gapi, service
+                                             start: start,
+                                             format_options_use_int64_timestamp: format_options_use_int64_timestamp
+          Data.from_gapi_json data_hash, destination_table_gapi, @gapi, service, format_options_use_int64_timestamp
         end
         alias query_results data
 
@@ -789,7 +792,8 @@ module Google
                   query: query,
                   default_dataset: dataset_config,
                   maximum_billing_tier: options[:maximum_billing_tier]
-                )
+                ),
+                reservation: options[:reservation]
               )
             )
 
@@ -1593,6 +1597,18 @@ module Google
             @gapi.configuration.query.clustering.fields = fields
           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

data/lib/google/cloud/bigquery/remote_function_options.rb

@@ -0,0 +1,156 @@
+# Copyright 2025 Google LLC
+#
+# 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
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Google
+  module Cloud
+    module Bigquery
+      ##
+      # # RemoteFunctionOptions
+      #
+      # Options for a remote user-defined function.
+      #
+      class RemoteFunctionOptions
+        ##
+        # @private The Google API Client object.
+        # @return [Google::Apis::BigqueryV2::RemoteFunctionOptions]
+        attr_accessor :gapi
+
+        ##
+        # Creates a new RemoteFunctionOptions object.
+        #
+        # @example
+        #   require "google/cloud/bigquery"
+        #
+        #   remote_function_options = Google::Cloud::Bigquery::RemoteFunctionOptions.new.tap do |rfo|
+        #     rfo.endpoint = "https://us-east1-my_gcf_project.cloudfunctions.net/remote_add"
+        #     rfo.connection = "projects/my-project/locations/us-east1/connections/my-connection"
+        #     rfo.user_defined_context = { "foo" => "bar" }
+        #   end
+        #
+        def initialize
+          @gapi = Google::Apis::BigqueryV2::RemoteFunctionOptions.new
+        end
+
+        ##
+        # The endpoint of the user-provided remote service, e.g.
+        # `https://us-east1-my_gcf_project.cloudfunctions.net/remote_add`.
+        #
+        # @return [String] The endpoint of the user-provided remote service.
+        #   Returns an empty string if the endpoint is not configured.
+        #
+        def endpoint
+          @gapi.endpoint || ""
+        end
+
+        ##
+        # Sets the endpoint of the user-provided remote service.
+        #
+        # @param [String, nil] new_endpoint The new endpoint. Passing `nil` will
+        #   clear the endpoint, indicating that no remote service is configured.
+        #
+        def endpoint= new_endpoint
+          @gapi.endpoint = new_endpoint
+        end
+
+        ##
+        # The fully qualified name of the user-provided connection object which
+        # holds the authentication information to send requests to the remote
+        # service.
+        #
+        # Format:
+        # `projects/{projectId}/locations/{locationId}/connections/{connectionId}`
+        #
+        # @return [String] The fully qualified name of the user-provided
+        #   connection object. Returns an empty string if the connection is not
+        #   configured.
+        #
+        def connection
+          @gapi.connection || ""
+        end
+
+        ##
+        # Sets the fully qualified name of the user-provided connection object.
+        #
+        # @param [String, nil] new_connection The new connection. Passing `nil`
+        #   will clear the connection, indicating that no authentication
+        #   information is configured for the remote service.
+        #
+        def connection= new_connection
+          @gapi.connection = new_connection
+        end
+
+        ##
+        # User-defined context as a set of key/value pairs, which will be sent
+        # as function invocation context together with batched arguments in the
+        # requests to the remote service. The total number of bytes of keys and
+        # values must be less than 8KB.
+        #
+        # @return [Hash] The user-defined context. Returns an empty hash if no
+        #   context is configured.
+        #
+        def user_defined_context
+          @gapi.user_defined_context || {}
+        end
+
+        ##
+        # Sets the user-defined context.
+        #
+        # @param [Hash, nil] new_user_defined_context The new user-defined
+        #   context. Passing `nil` will clear the context, meaning no
+        #   user-defined key-value pairs will be sent.
+        #
+        def user_defined_context= new_user_defined_context
+          @gapi.user_defined_context = new_user_defined_context
+        end
+
+        ##
+        # Max number of rows in each batch sent to the remote service. If absent
+        # or if 0, BigQuery dynamically decides the number of rows in a batch.
+        #
+        # @return [Integer] Max number of rows in each batch. Returns `0` if not
+        #   set, which indicates that BigQuery dynamically decides the number of
+        #   rows.
+        #
+        def max_batching_rows
+          @gapi.max_batching_rows || 0
+        end
+
+        ##
+        # Sets the max number of rows in each batch sent to the remote service.
+        #
+        # @param [Integer, nil] new_max_batching_rows The new max batching rows.
+        #   Passing `nil` or `0` will reset the batch size, indicating that
+        #   BigQuery should dynamically decide the number of rows in each batch.
+        #
+        def max_batching_rows= new_max_batching_rows
+          @gapi.max_batching_rows = new_max_batching_rows
+        end
+
+        # @private New RemoteFunctionOptions from a Google API Client object.
+        def self.from_gapi gapi
+          return nil if gapi.nil?
+          new.tap do |rfo|
+            rfo.instance_variable_set :@gapi, gapi
+          end
+        end
+
+        ##
+        # @private Returns the Google API Client object.
+        def to_gapi
+          @gapi
+        end
+      end
+    end
+  end
+end