Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: Support the Count aggregation query #368

Merged
merged 24 commits into from
Nov 1, 2022
Merged
Show file tree
Hide file tree
Changes from 20 commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
6 changes: 6 additions & 0 deletions docs/aggregations.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
Aggregations
~~~~~~~~~~~~

.. automodule:: google.cloud.datastore.aggregation
:members:
:show-inheritance:
1 change: 1 addition & 0 deletions docs/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ API Reference
entities
keys
queries
aggregations
transactions
batches
helpers
Expand Down
33 changes: 33 additions & 0 deletions google/cloud/datastore/_http.py
Original file line number Diff line number Diff line change
Expand Up @@ -280,6 +280,39 @@ def run_query(self, request, retry=None, timeout=None):
timeout=timeout,
)

def run_aggregation_query(self, request, retry=None, timeout=None):
"""Perform a ``runAggregationQuery`` request.

:type request: :class:`_datastore_pb2.BeginTransactionRequest` or dict
:param request:
Parameter bundle for API request.

:type retry: :class:`google.api_core.retry.Retry`
:param retry: (Optional) retry policy for the request

:type timeout: float or tuple(float, float)
:param timeout: (Optional) timeout for the request

:rtype: :class:`.datastore_pb2.RunAggregationQueryResponse`
:returns: The returned protobuf response object.
"""
request_pb = _make_request_pb(
request, _datastore_pb2.RunAggregationQueryRequest
)
project_id = request_pb.project_id

return _rpc(
self.client._http,
project_id,
"runAggregationQuery",
self.client._base_url,
self.client._client_info,
request_pb,
_datastore_pb2.RunAggregationQueryResponse,
retry=retry,
timeout=timeout,
)

def begin_transaction(self, request, retry=None, timeout=None):
"""Perform a ``beginTransaction`` request.

Expand Down