google-cloud-datastore-v1 0.6.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5756bbbfdf9b930addfb17ff508cd1ff58e698c1381a922a2a1d06c442cc3499
-  data.tar.gz: '0087211fa2c48a1c58801db241b387eb29543cabe28c5eca2b290ab96c0370cd'
+  metadata.gz: c45c26b4609092449a125a9c82ea164e7a8cc8ae73dc4b221879fe2aec45d13e
+  data.tar.gz: d0949170d61b922ecde9f8ceda24e9d169afc6949b344e91c9d536d04aa50099
 SHA512:
-  metadata.gz: df7ce1dd9962a3e286eab555739ccbd3223b937b53d98f0f68ed5a3b2b3d2e163832f7298e8202e08c961b18788a1de3e7ce8ceec3576009e7ec3522576fedf0
-  data.tar.gz: e37399f05e9dae3dbec09a7c9f3d4d56cc525d76cda7ccf42b9da787246e387348f034bc5496e3c446c14a325eeba256adb01c4734fe6e85cea999d27c057452
+  metadata.gz: 7eee514105c9d4a2e2b98d5ff9948059d4c1b52f2770e2ad57843945f7dc89b8d61b383845125d7c1a070ddc2ae51c935db270237a90f7965657da0906698a94
+  data.tar.gz: ea3a3d81f08b5b8d7232405bae7e7f9ae53a01842121e6363f80e030749052bc186a3ea15fe07d076272541a392f67c753b7e3ab24fd27055e3fdbeecdcc1d67

data/AUTHENTICATION.md

@@ -114,7 +114,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/datastore/v1/datastore/client.rb

@@ -77,6 +77,11 @@ module Google
                   initial_delay: 0.1, max_delay: 60.0, multiplier: 1.3, retry_codes: [14, 4]
                 }
 
+                default_config.rpcs.run_aggregation_query.timeout = 60.0
+                default_config.rpcs.run_aggregation_query.retry_policy = {
+                  initial_delay: 0.1, max_delay: 60.0, multiplier: 1.3, retry_codes: [14, 4]
+                }
+
                 default_config.rpcs.begin_transaction.timeout = 60.0
 
                 default_config.rpcs.commit.timeout = 60.0
@@ -183,13 +188,18 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload lookup(project_id: nil, read_options: nil, keys: nil)
+            # @overload lookup(project_id: nil, database_id: nil, read_options: nil, keys: nil)
             #   Pass arguments to `lookup` 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 project_id [::String]
             #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param read_options [::Google::Cloud::Datastore::V1::ReadOptions, ::Hash]
             #     The options for this lookup request.
             #   @param keys [::Array<::Google::Cloud::Datastore::V1::Key, ::Hash>]
@@ -236,11 +246,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.lookup.timeout,
@@ -272,13 +285,18 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload run_query(project_id: nil, partition_id: nil, read_options: nil, query: nil, gql_query: nil)
+            # @overload run_query(project_id: nil, database_id: nil, partition_id: nil, read_options: nil, query: nil, gql_query: nil)
             #   Pass arguments to `run_query` 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 project_id [::String]
             #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param partition_id [::Google::Cloud::Datastore::V1::PartitionId, ::Hash]
             #     Entities are partitioned into subsets, identified by a partition ID.
             #     Queries are scoped to a single partition.
@@ -289,7 +307,7 @@ module Google
             #   @param query [::Google::Cloud::Datastore::V1::Query, ::Hash]
             #     The query to run.
             #   @param gql_query [::Google::Cloud::Datastore::V1::GqlQuery, ::Hash]
-            #     The GQL query to run.
+            #     The GQL query to run. This query must be a non-aggregation query.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Google::Cloud::Datastore::V1::RunQueryResponse]
@@ -332,11 +350,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.run_query.timeout,
@@ -355,6 +376,110 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Runs an aggregation query.
+            #
+            # @overload run_aggregation_query(request, options = nil)
+            #   Pass arguments to `run_aggregation_query` via a request object, either of type
+            #   {::Google::Cloud::Datastore::V1::RunAggregationQueryRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::Datastore::V1::RunAggregationQueryRequest, ::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 run_aggregation_query(project_id: nil, database_id: nil, partition_id: nil, read_options: nil, aggregation_query: nil, gql_query: nil)
+            #   Pass arguments to `run_aggregation_query` 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 project_id [::String]
+            #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
+            #   @param partition_id [::Google::Cloud::Datastore::V1::PartitionId, ::Hash]
+            #     Entities are partitioned into subsets, identified by a partition ID.
+            #     Queries are scoped to a single partition.
+            #     This partition ID is normalized with the standard default context
+            #     partition ID.
+            #   @param read_options [::Google::Cloud::Datastore::V1::ReadOptions, ::Hash]
+            #     The options for this query.
+            #   @param aggregation_query [::Google::Cloud::Datastore::V1::AggregationQuery, ::Hash]
+            #     The query to run.
+            #   @param gql_query [::Google::Cloud::Datastore::V1::GqlQuery, ::Hash]
+            #     The GQL query to run. This query must be an aggregation query.
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Cloud::Datastore::V1::RunAggregationQueryResponse]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Cloud::Datastore::V1::RunAggregationQueryResponse]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC is aborted.
+            #
+            # @example Basic example
+            #   require "google/cloud/datastore/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::Datastore::V1::Datastore::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::Datastore::V1::RunAggregationQueryRequest.new
+            #
+            #   # Call the run_aggregation_query method.
+            #   result = client.run_aggregation_query request
+            #
+            #   # The returned object is of type Google::Cloud::Datastore::V1::RunAggregationQueryResponse.
+            #   p result
+            #
+            def run_aggregation_query request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Datastore::V1::RunAggregationQueryRequest
+
+              # 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.run_aggregation_query.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::Datastore::V1::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.project_id && !request.project_id.empty?
+                header_params["project_id"] = request.project_id
+              end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
+
+              request_params_header = URI.encode_www_form header_params
+              metadata[:"x-goog-request-params"] ||= request_params_header
+
+              options.apply_defaults timeout:      @config.rpcs.run_aggregation_query.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.run_aggregation_query.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @datastore_stub.call_rpc :run_aggregation_query, request, options: options do |response, operation|
+                yield response, operation if block_given?
+                return response
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Begins a new transaction.
             #
@@ -368,13 +493,18 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload begin_transaction(project_id: nil, transaction_options: nil)
+            # @overload begin_transaction(project_id: nil, database_id: nil, transaction_options: nil)
             #   Pass arguments to `begin_transaction` 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 project_id [::String]
             #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param transaction_options [::Google::Cloud::Datastore::V1::TransactionOptions, ::Hash]
             #     Options for a new transaction.
             #
@@ -419,11 +549,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.begin_transaction.timeout,
@@ -456,13 +589,18 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload commit(project_id: nil, mode: nil, transaction: nil, mutations: nil)
+            # @overload commit(project_id: nil, database_id: nil, mode: nil, transaction: nil, mutations: nil)
             #   Pass arguments to `commit` 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 project_id [::String]
             #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param mode [::Google::Cloud::Datastore::V1::CommitRequest::Mode]
             #     The type of commit to perform. Defaults to `TRANSACTIONAL`.
             #   @param transaction [::String]
@@ -525,11 +663,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.commit.timeout,
@@ -561,13 +702,18 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload rollback(project_id: nil, transaction: nil)
+            # @overload rollback(project_id: nil, database_id: nil, transaction: nil)
             #   Pass arguments to `rollback` 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 project_id [::String]
             #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param transaction [::String]
             #     Required. The transaction identifier, returned by a call to
             #     {::Google::Cloud::Datastore::V1::Datastore::Client#begin_transaction Datastore.BeginTransaction}.
@@ -613,11 +759,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.rollback.timeout,
@@ -650,16 +799,21 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload allocate_ids(project_id: nil, keys: nil)
+            # @overload allocate_ids(project_id: nil, database_id: nil, keys: nil)
             #   Pass arguments to `allocate_ids` 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 project_id [::String]
             #     Required. The ID of the project against which to make the request.
+            #   @param database_id [::String]
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param keys [::Array<::Google::Cloud::Datastore::V1::Key, ::Hash>]
-            #     Required. A list of keys with incomplete key paths for which to allocate IDs.
-            #     No key may be reserved/read-only.
+            #     Required. A list of keys with incomplete key paths for which to allocate
+            #     IDs. No key may be reserved/read-only.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Google::Cloud::Datastore::V1::AllocateIdsResponse]
@@ -702,11 +856,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.allocate_ids.timeout,
@@ -747,10 +904,13 @@ module Google
             #   @param project_id [::String]
             #     Required. The ID of the project against which to make the request.
             #   @param database_id [::String]
-            #     If not empty, the ID of the database against which to make the request.
+            #     The ID of the database against which to make the request.
+            #
+            #     '(default)' is not allowed; please use empty string '' to refer the default
+            #     database.
             #   @param keys [::Array<::Google::Cloud::Datastore::V1::Key, ::Hash>]
-            #     Required. A list of keys with complete key paths whose numeric IDs should not be
-            #     auto-allocated.
+            #     Required. A list of keys with complete key paths whose numeric IDs should
+            #     not be auto-allocated.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Google::Cloud::Datastore::V1::ReserveIdsResponse]
@@ -793,11 +953,14 @@ module Google
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
-              if request.project_id
+              if request.project_id && !request.project_id.empty?
                 header_params["project_id"] = request.project_id
               end
+              if request.database_id && !request.database_id.empty?
+                header_params["database_id"] = request.database_id
+              end
 
-              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              request_params_header = URI.encode_www_form header_params
               metadata[:"x-goog-request-params"] ||= request_params_header
 
               options.apply_defaults timeout:      @config.rpcs.reserve_ids.timeout,
@@ -962,6 +1125,11 @@ module Google
                 #
                 attr_reader :run_query
                 ##
+                # RPC-specific configuration for `run_aggregation_query`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :run_aggregation_query
+                ##
                 # RPC-specific configuration for `begin_transaction`
                 # @return [::Gapic::Config::Method]
                 #
@@ -993,6 +1161,8 @@ module Google
                   @lookup = ::Gapic::Config::Method.new lookup_config
                   run_query_config = parent_rpcs.run_query if parent_rpcs.respond_to? :run_query
                   @run_query = ::Gapic::Config::Method.new run_query_config
+                  run_aggregation_query_config = parent_rpcs.run_aggregation_query if parent_rpcs.respond_to? :run_aggregation_query
+                  @run_aggregation_query = ::Gapic::Config::Method.new run_aggregation_query_config
                   begin_transaction_config = parent_rpcs.begin_transaction if parent_rpcs.respond_to? :begin_transaction
                   @begin_transaction = ::Gapic::Config::Method.new begin_transaction_config
                   commit_config = parent_rpcs.commit if parent_rpcs.respond_to? :commit

data/lib/google/cloud/datastore/v1/datastore.rb

@@ -37,7 +37,7 @@ module Google
         # input keys sets the project ID (if not already set) to the project ID from
         # the request.
         #
-        # To load this service and instantiate a client:
+        # @example Load this service and instantiate a gRPC client
         #
         #     require "google/cloud/datastore/v1/datastore"
         #     client = ::Google::Cloud::Datastore::V1::Datastore::Client.new

data/lib/google/cloud/datastore/v1/version.rb

@@ -21,7 +21,7 @@ module Google
   module Cloud
     module Datastore
       module V1
-        VERSION = "0.6.0"
+        VERSION = "0.8.0"
       end
     end
   end

data/lib/google/cloud/datastore/v1.rb

@@ -23,9 +23,9 @@ module Google
   module Cloud
     module Datastore
       ##
-      # 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/datastore/v1"
       #     client = ::Google::Cloud::Datastore::V1::Datastore::Client.new

data/lib/google/datastore/v1/aggregation_result_pb.rb

@@ -0,0 +1,32 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/datastore/v1/aggregation_result.proto
+
+require 'google/protobuf'
+
+require 'google/datastore/v1/entity_pb'
+require 'google/datastore/v1/query_pb'
+require 'google/protobuf/timestamp_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/datastore/v1/aggregation_result.proto", :syntax => :proto3) do
+    add_message "google.datastore.v1.AggregationResult" do
+      map :aggregate_properties, :string, :message, 2, "google.datastore.v1.Value"
+    end
+    add_message "google.datastore.v1.AggregationResultBatch" do
+      repeated :aggregation_results, :message, 1, "google.datastore.v1.AggregationResult"
+      optional :more_results, :enum, 2, "google.datastore.v1.QueryResultBatch.MoreResultsType"
+      optional :read_time, :message, 3, "google.protobuf.Timestamp"
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module Datastore
+      module V1
+        AggregationResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.AggregationResult").msgclass
+        AggregationResultBatch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.AggregationResultBatch").msgclass
+      end
+    end
+  end
+end

data/lib/google/datastore/v1/datastore_pb.rb

@@ -6,6 +6,8 @@ require 'google/protobuf'
 require 'google/api/annotations_pb'
 require 'google/api/client_pb'
 require 'google/api/field_behavior_pb'
+require 'google/api/routing_pb'
+require 'google/datastore/v1/aggregation_result_pb'
 require 'google/datastore/v1/entity_pb'
 require 'google/datastore/v1/query_pb'
 require 'google/protobuf/timestamp_pb'
@@ -14,6 +16,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/datastore/v1/datastore.proto", :syntax => :proto3) do
     add_message "google.datastore.v1.LookupRequest" do
       optional :project_id, :string, 8
+      optional :database_id, :string, 9
       optional :read_options, :message, 1, "google.datastore.v1.ReadOptions"
       repeated :keys, :message, 3, "google.datastore.v1.Key"
     end
@@ -25,6 +28,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "google.datastore.v1.RunQueryRequest" do
       optional :project_id, :string, 8
+      optional :database_id, :string, 9
       optional :partition_id, :message, 2, "google.datastore.v1.PartitionId"
       optional :read_options, :message, 1, "google.datastore.v1.ReadOptions"
       oneof :query_type do
@@ -36,8 +40,23 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :batch, :message, 1, "google.datastore.v1.QueryResultBatch"
       optional :query, :message, 2, "google.datastore.v1.Query"
     end
+    add_message "google.datastore.v1.RunAggregationQueryRequest" do
+      optional :project_id, :string, 8
+      optional :database_id, :string, 9
+      optional :partition_id, :message, 2, "google.datastore.v1.PartitionId"
+      optional :read_options, :message, 1, "google.datastore.v1.ReadOptions"
+      oneof :query_type do
+        optional :aggregation_query, :message, 3, "google.datastore.v1.AggregationQuery"
+        optional :gql_query, :message, 7, "google.datastore.v1.GqlQuery"
+      end
+    end
+    add_message "google.datastore.v1.RunAggregationQueryResponse" do
+      optional :batch, :message, 1, "google.datastore.v1.AggregationResultBatch"
+      optional :query, :message, 2, "google.datastore.v1.AggregationQuery"
+    end
     add_message "google.datastore.v1.BeginTransactionRequest" do
       optional :project_id, :string, 8
+      optional :database_id, :string, 9
       optional :transaction_options, :message, 10, "google.datastore.v1.TransactionOptions"
     end
     add_message "google.datastore.v1.BeginTransactionResponse" do
@@ -45,12 +64,14 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "google.datastore.v1.RollbackRequest" do
       optional :project_id, :string, 8
+      optional :database_id, :string, 9
       optional :transaction, :bytes, 1
     end
     add_message "google.datastore.v1.RollbackResponse" do
     end
     add_message "google.datastore.v1.CommitRequest" do
       optional :project_id, :string, 8
+      optional :database_id, :string, 9
       optional :mode, :enum, 5, "google.datastore.v1.CommitRequest.Mode"
       repeated :mutations, :message, 6, "google.datastore.v1.Mutation"
       oneof :transaction_selector do
@@ -69,6 +90,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "google.datastore.v1.AllocateIdsRequest" do
       optional :project_id, :string, 8
+      optional :database_id, :string, 9
       repeated :keys, :message, 1, "google.datastore.v1.Key"
     end
     add_message "google.datastore.v1.AllocateIdsResponse" do
@@ -134,6 +156,8 @@ module Google
         LookupResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.LookupResponse").msgclass
         RunQueryRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.RunQueryRequest").msgclass
         RunQueryResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.RunQueryResponse").msgclass
+        RunAggregationQueryRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.RunAggregationQueryRequest").msgclass
+        RunAggregationQueryResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.RunAggregationQueryResponse").msgclass
         BeginTransactionRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.BeginTransactionRequest").msgclass
         BeginTransactionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.BeginTransactionResponse").msgclass
         RollbackRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.RollbackRequest").msgclass

data/lib/google/datastore/v1/datastore_services_pb.rb

@@ -43,6 +43,8 @@ module Google
             rpc :Lookup, ::Google::Cloud::Datastore::V1::LookupRequest, ::Google::Cloud::Datastore::V1::LookupResponse
             # Queries for entities.
             rpc :RunQuery, ::Google::Cloud::Datastore::V1::RunQueryRequest, ::Google::Cloud::Datastore::V1::RunQueryResponse
+            # Runs an aggregation query.
+            rpc :RunAggregationQuery, ::Google::Cloud::Datastore::V1::RunAggregationQueryRequest, ::Google::Cloud::Datastore::V1::RunAggregationQueryResponse
             # Begins a new transaction.
             rpc :BeginTransaction, ::Google::Cloud::Datastore::V1::BeginTransactionRequest, ::Google::Cloud::Datastore::V1::BeginTransactionResponse
             # Commits a transaction, optionally creating, deleting or modifying some

data/lib/google/datastore/v1/entity_pb.rb

@@ -11,6 +11,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/datastore/v1/entity.proto", :syntax => :proto3) do
     add_message "google.datastore.v1.PartitionId" do
       optional :project_id, :string, 2
+      optional :database_id, :string, 3
       optional :namespace_id, :string, 4
     end
     add_message "google.datastore.v1.Key" do

data/lib/google/datastore/v1/query_pb.rb

@@ -3,6 +3,7 @@
 
 require 'google/protobuf'
 
+require 'google/api/field_behavior_pb'
 require 'google/datastore/v1/entity_pb'
 require 'google/protobuf/timestamp_pb'
 require 'google/protobuf/wrappers_pb'
@@ -32,6 +33,21 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :offset, :int32, 10
       optional :limit, :message, 12, "google.protobuf.Int32Value"
     end
+    add_message "google.datastore.v1.AggregationQuery" do
+      repeated :aggregations, :message, 3, "google.datastore.v1.AggregationQuery.Aggregation"
+      oneof :query_type do
+        optional :nested_query, :message, 1, "google.datastore.v1.Query"
+      end
+    end
+    add_message "google.datastore.v1.AggregationQuery.Aggregation" do
+      optional :alias, :string, 7
+      oneof :operator do
+        optional :count, :message, 1, "google.datastore.v1.AggregationQuery.Aggregation.Count"
+      end
+    end
+    add_message "google.datastore.v1.AggregationQuery.Aggregation.Count" do
+      optional :up_to, :message, 1, "google.protobuf.Int64Value"
+    end
     add_message "google.datastore.v1.KindExpression" do
       optional :name, :string, 1
     end
@@ -120,6 +136,9 @@ module Google
         EntityResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.EntityResult").msgclass
         EntityResult::ResultType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.EntityResult.ResultType").enummodule
         Query = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.Query").msgclass
+        AggregationQuery = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.AggregationQuery").msgclass
+        AggregationQuery::Aggregation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.AggregationQuery.Aggregation").msgclass
+        AggregationQuery::Aggregation::Count = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.AggregationQuery.Aggregation.Count").msgclass
         KindExpression = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.KindExpression").msgclass
         PropertyReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.PropertyReference").msgclass
         Projection = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.datastore.v1.Projection").msgclass