google-cloud-bigquery 1.52.1 → 1.55.0

data/lib/google/cloud/bigquery/external/csv_source.rb

@@ -367,6 +367,177 @@ module Google
             @gapi.csv_options.skip_leading_rows = row_count
           end
 
+          ##
+          # Specifies a string that represents a null value in a CSV file. For
+          # example, if you specify `\N`, BigQuery interprets `\N` as a null value when
+          # querying a CSV file. The default value is the empty string. If you set this
+          # property to a custom value, BigQuery throws an error if an empty string is
+          # present for all data types except for STRING and BYTE. For STRING and BYTE
+          # columns, BigQuery interprets the empty string as an empty value.
+          #
+          # @return [String, nil] The null marker string. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   csv_url = "gs://bucket/path/to/data.csv"
+          #   csv_table = bigquery.external csv_url do |csv|
+          #     csv.null_marker = "\N"
+          #   end
+          #
+          #   csv_table.null_marker #=> "\N"
+          #
+          def null_marker
+            @gapi.csv_options.null_marker
+          end
+
+          ##
+          # Sets a string that represents a null value in a CSV file. For
+          # example, if you specify `\N`, BigQuery interprets `\N` as a null value when
+          # querying a CSV file. The default value is the empty string. If you set this
+          # property to a custom value, BigQuery throws an error if an empty string is
+          # present for all data types except for STRING and BYTE. For STRING and BYTE
+          # columns, BigQuery interprets the empty string as an empty value.
+          #
+          # @param [String, nil] null_marker The null marker string. `nil` to unset.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   csv_url = "gs://bucket/path/to/data.csv"
+          #   csv_table = bigquery.external csv_url do |csv|
+          #     csv.null_marker = "\N"
+          #   end
+          #
+          #   csv_table.null_marker #=> "\N"
+          #
+          def null_marker= null_marker
+            frozen_check!
+            @gapi.csv_options.null_marker = null_marker
+          end
+
+          ##
+          # The 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.
+          #
+          # @return [Array<String>] The array of null marker strings.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   csv_url = "gs://bucket/path/to/data.csv"
+          #   csv_table = bigquery.external csv_url do |csv|
+          #     csv.null_markers = ["\N", "NULL"]
+          #   end
+          #
+          #   csv_table.null_markers #=> ["\N", "NULL"]
+          #
+          def null_markers
+            @gapi.csv_options.null_markers || []
+          end
+
+          ##
+          # Sets the 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 [Array<String>] null_markers The array of null marker strings.
+          #
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   csv_url = "gs://bucket/path/to/data.csv"
+          #   csv_table = bigquery.external csv_url do |csv|
+          #     csv.null_markers = ["\N", "NULL"]
+          #   end
+          #
+          #   csv_table.null_markers #=> ["\N", "NULL"]
+          #
+          def null_markers= null_markers
+            frozen_check!
+            @gapi.csv_options.null_markers = null_markers
+          end
+
+          # 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. Assumes columns are ordered the
+          #   same way as the schema.
+          # - `NAME`: matches by name. Reads the header row as column names and
+          #   reorders columns to match the schema.
+          #
+          # @return [String, nil] The source column match value. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   csv_url = "gs://bucket/path/to/data.csv"
+          #   csv_table = bigquery.external csv_url do |csv|
+          #     csv.source_column_match = "NAME"
+          #   end
+          #
+          #   csv_table.source_column_match #=> "NAME"
+          #
+          def source_column_match
+            @gapi.csv_options.source_column_match
+          end
+
+          # Sets 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. Optional.
+          #
+          # Acceptable values are:
+          # - `POSITION`: matches by position. Assumes columns are ordered the
+          #   same way as the schema.
+          # - `NAME`: matches by name. Reads the header row as column names and
+          #   reorders columns to match the schema.
+          #
+          # @param [String, nil] source_column_match The new source column match value. `nil` to unset.
+          #
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   csv_url = "gs://bucket/path/to/data.csv"
+          #   csv_table = bigquery.external csv_url do |csv|
+          #     csv.source_column_match = "NAME"
+          #   end
+          #
+          #   csv_table.source_column_match #=> "NAME"
+          #
+          def source_column_match= source_column_match
+            frozen_check!
+            @gapi.csv_options.source_column_match = source_column_match
+          end
+
           ##
           # The schema for the data.
           #

data/lib/google/cloud/bigquery/external/data_source.rb

@@ -744,6 +744,222 @@ module Google
             @gapi.hive_partitioning_options.source_uri_prefix = source_uri_prefix
           end
 
+          ##
+          # Time zone used when parsing timestamp values that do not have specific
+          # time zone information (e.g. `2024-04-20 12:34:56`). The expected format
+          # is an IANA timezone string (e.g. `America/Los_Angeles`).
+          #
+          # @return [String, nil] The IANA time zone name. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.time_zone = "America/Los_Angeles"
+          #   end
+          #
+          #   external_data.time_zone #=> "America/Los_Angeles"
+          #
+          def time_zone
+            @gapi.time_zone
+          end
+
+          ##
+          # Sets the time zone used when parsing timestamp values that do not have
+          # specific time zone information (e.g. `2024-04-20 12:34:56`). The expected
+          # format is an IANA timezone string (e.g. `America/Los_Angeles`).
+          #
+          # @param [String, nil] time_zone The IANA time zone name. `nil` to unset.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.time_zone = "America/Los_Angeles"
+          #   end
+          #
+          #   external_data.time_zone #=> "America/Los_Angeles"
+          #
+          def time_zone= time_zone
+            frozen_check!
+            @gapi.time_zone = time_zone
+          end
+
+          ##
+          # Format used to parse TIME values. Supports C-style and SQL-style values.
+          #
+          # @return [String, nil] The time format pattern. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.time_format = "%H:%M:%S"
+          #   end
+          #
+          #   external_data.time_format #=> "%H:%M:%S"
+          #
+          def time_format
+            @gapi.time_format
+          end
+
+          ##
+          # Sets the format used to parse TIME values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] time_format The time format pattern. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.time_format = "%H:%M:%S"
+          #   end
+          #
+          #   external_data.time_format #=> "%H:%M:%S"
+          #
+          def time_format= time_format
+            frozen_check!
+            @gapi.time_format = time_format
+          end
+
+          ##
+          # Format used to parse TIMESTAMP values. Supports C-style and SQL-style
+          # values.
+          #
+          # @return [String, nil] The timestamp format pattern. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.timestamp_format = "%Y-%m-%d %H:%M:%S.%f %z"
+          #   end
+          #
+          #   external_data.timestamp_format #=> "%Y-%m-%d %H:%M:%S.%f %z"
+          #
+          def timestamp_format
+            @gapi.timestamp_format
+          end
+
+          ##
+          # Sets the format used to parse TIMESTAMP values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] timestamp_format The timestamp format pattern. `nil` to unset.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.timestamp_format = "%Y-%m-%d %H:%M:%S.%f %z"
+          #   end
+          #
+          #   external_data.timestamp_format #=> "%Y-%m-%d %H:%M:%S.%f %z"
+          #
+          def timestamp_format= timestamp_format
+            frozen_check!
+            @gapi.timestamp_format = timestamp_format
+          end
+
+          ##
+          # Format used to parse DATETIME values. Supports C-style and SQL-style
+          # values.
+          #
+          # @return [String, nil] The datetime format pattern. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.datetime_format = "%Y-%m-%d %H:%M:%S"
+          #   end
+          #
+          #   external_data.datetime_format #=> "%Y-%m-%d %H:%M:%S"
+          #
+          def datetime_format
+            @gapi.datetime_format
+          end
+
+          ##
+          # Sets the format used to parse DATETIME values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] datetime_format The datetime format pattern. `nil` to unset.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.datetime_format = "%Y-%m-%d %H:%M:%S"
+          #   end
+          #
+          #   external_data.datetime_format #=> "%Y-%m-%d %H:%M:%S"
+          #
+          def datetime_format= datetime_format
+            frozen_check!
+            @gapi.datetime_format = datetime_format
+          end
+
+          ##
+          # Format used to parse DATE values. Supports C-style and SQL-style
+          # values.
+          #
+          # @return [String, nil] The date format pattern. `nil` if not set.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.date_format = "%Y-%m-%d"
+          #   end
+          #
+          #   external_data.date_format #=> "%Y-%m-%d"
+          #
+          def date_format
+            @gapi.date_format
+          end
+
+          ##
+          # Sets the format used to parse DATE values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] date_format The date format pattern. `nil` to unset.
+          #
+          # @example
+          #   require "google/cloud/bigquery"
+          #
+          #   bigquery = Google::Cloud::Bigquery.new
+          #
+          #   external_data = bigquery.external "gs://bucket/path/to/data.csv" do |ext|
+          #     ext.date_format = "%Y-%m-%d"
+          #   end
+          #
+          #   external_data.date_format #=> "%Y-%m-%d"
+          #
+          def date_format= date_format
+            frozen_check!
+            @gapi.date_format = date_format
+          end
+
           ##
           # @private Google API Client object.
           def to_gapi

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

@@ -652,6 +652,90 @@ module Google
           @gapi.configuration.load.clustering.fields if clustering?
         end
 
+        ##
+        # Format used to parse DATE values. Supports C-style and SQL-style
+        # values.
+        #
+        # @return [String, nil] The date format pattern, such as
+        #   `%Y-%m-%d`. `nil` if not set.
+        def date_format
+          @gapi.configuration.load.date_format
+        end
+
+        ##
+        # Format used to parse DATETIME values. Supports C-style and SQL-style
+        # values.
+        #
+        # @return [String, nil] The datetime format pattern, such as
+        #   `%Y-%m-%d %H:%M:%S`. `nil` if not set.
+        def datetime_format
+          @gapi.configuration.load.datetime_format
+        end
+
+        ##
+        # Format used to parse TIME values. Supports C-style and SQL-style
+        # values.
+        #
+        # @return [String, nil] The time format pattern, such as
+        #   `%H:%M:%S`. `nil` if not set.
+        def time_format
+          @gapi.configuration.load.time_format
+        end
+
+        ##
+        # Format used to parse TIMESTAMP values. Supports C-style and SQL-style
+        # values.
+        #
+        # @return [String, nil] The timestamp format pattern, such as
+        #   `%Y-%m-%d %H:%M:%S.%f %z`. `nil` if not set.
+        def timestamp_format
+          @gapi.configuration.load.timestamp_format
+        end
+
+        ##
+        # 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.
+        #
+        # @return [Array<String>] The array of null marker strings.
+        def null_markers
+          @gapi.configuration.load.null_markers || []
+        end
+
+        ##
+        # 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.
+        #
+        # @return [String, nil] The source column match strategy, such as
+        #   `POSITION`. `nil` if not set.
+        def source_column_match
+          @gapi.configuration.load.source_column_match
+        end
+
+        ##
+        # Time zone used when parsing timestamp values that do not have specific
+        # time zone information (e.g. `2024-04-20 12:34:56`). The expected format
+        # is an IANA timezone string (e.g. `America/Los_Angeles`).
+        #
+        # @return [String, nil] The IANA time zone name, such as
+        #   `America/Los_Angeles`. `nil` if not set.
+        def time_zone
+          @gapi.configuration.load.time_zone
+        end
+
         ##
         # Yielded to a block to accumulate changes for a patch request.
         class Updater < LoadJob
@@ -2585,6 +2669,89 @@ module Google
             @gapi.configuration.load.clustering.fields = fields
           end
 
+          ##
+          # Sets the format used to parse DATE values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] date_format The date format pattern, such as
+          #   `%Y-%m-%d`. `nil` to unset.
+          def date_format= date_format
+            @gapi.configuration.load.update! date_format: date_format
+          end
+
+          ##
+          # Sets the format used to parse DATETIME values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] datetime_format The datetime format pattern, such as
+          #   `%Y-%m-%d %H:%M:%S`. `nil` to unset.
+          def datetime_format= datetime_format
+            @gapi.configuration.load.update! datetime_format: datetime_format
+          end
+
+          ##
+          # Sets the format used to parse TIME values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] time_format The time format pattern, such as
+          #   `%H:%M:%S`. `nil` to unset.
+          def time_format= time_format
+            @gapi.configuration.load.update! time_format: time_format
+          end
+
+          ##
+          # Sets the format used to parse TIMESTAMP values. Supports C-style and SQL-style
+          # values.
+          #
+          # @param [String, nil] timestamp_format The timestamp format pattern, such as
+          #   `%Y-%m-%d %H:%M:%S.%f %z`. `nil` to unset.
+          def timestamp_format= timestamp_format
+            @gapi.configuration.load.update! timestamp_format: timestamp_format
+          end
+
+          ##
+          # Sets the 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 [Array<String>] null_markers The array of null marker strings.
+          def null_markers= null_markers
+            @gapi.configuration.load.update! null_markers: null_markers
+          end
+
+          # Sets 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. Assumes columns are ordered the
+          #   same way as the schema.
+          # - `NAME`: matches by name. Reads the header row as column names and
+          #   reorders columns to match the schema.
+          #
+          # @param [String, nil] source_column_match The new source column match value.
+          #   `nil` to unset.
+          def source_column_match= source_column_match
+            @gapi.configuration.load.update! source_column_match: source_column_match
+          end
+
+          ##
+          # Sets the time zone used when parsing timestamp values that do not have
+          # specific time zone information (e.g. `2024-04-20 12:34:56`). The expected
+          # format is an IANA timezone string (e.g. `America/Los_Angeles`).
+          #
+          # @param [String, nil] time_zone The IANA time zone name, such as
+          #   `America/Los_Angeles`. `nil` to unset.
+          def time_zone= time_zone
+            @gapi.configuration.load.update! time_zone: time_zone
+          end
+
           def cancel
             raise "not implemented in #{self.class}"
           end

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

@@ -783,6 +783,9 @@ 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.
+        #
         # @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 +932,7 @@ module Google
                   standard_sql: nil,
                   legacy_sql: nil,
                   session_id: nil,
+                  format_options_use_int64_timestamp: true,
                   &block
           job = query_job query,
                           params: params,
@@ -953,7 +957,7 @@ module Google
             end
           end
 
-          job.data max: max
+          job.data max: max, format_options_use_int64_timestamp: format_options_use_int64_timestamp
         end
 
         ##
@@ -1378,6 +1382,20 @@ 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.
         #
         # @return [Google::Cloud::Bigquery::Dataset, nil] Returns `nil` if the
         #   dataset does not exist.
@@ -1405,12 +1423,12 @@ 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
           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.from_gapi gapi, service, access_policy_version: access_policy_version
         rescue Google::Cloud::NotFoundError
           nil
         end
@@ -1429,6 +1447,20 @@ 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.
         # @yield [access] a block for setting rules
         # @yieldparam [Google::Cloud::Bigquery::Dataset] access the object
         #   accepting rules
@@ -1461,7 +1493,7 @@ module Google
         #   end
         #
         def create_dataset dataset_id, name: nil, description: nil,
-                           expiration: nil, location: nil
+                           expiration: nil, location: nil, access_policy_version: nil
           ensure_service!
 
           new_ds = Google::Apis::BigqueryV2::Dataset.new(
@@ -1484,8 +1516,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
         end
 
         ##