Skip to content

Commit 203bf2e

Browse files
shweta345nicainMariatta
authored
chore(samples): added samples for Regional endpoint (#9479)
* Added code sample and test for regional endpoints --------- Co-authored-by: nicain <[email protected]> Co-authored-by: Mariatta Wijaya <[email protected]>
1 parent 7073ec1 commit 203bf2e

File tree

4 files changed

+38
-0
lines changed

4 files changed

+38
-0
lines changed

datastore/cloud-client/snippets.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import datetime
1717
from pprint import pprint
1818

19+
from google.api_core.client_options import ClientOptions
1920
import google.cloud.exceptions
2021
from google.cloud import datastore # noqa: I100
2122

@@ -883,6 +884,21 @@ def property_by_kind_run_query(client):
883884
return representations_by_property
884885

885886

887+
def regional_endpoint():
888+
# [START datastore_regional_endpoints]
889+
ENDPOINT = "https://datastore.googleapis.com"
890+
client_options = ClientOptions(api_endpoint=ENDPOINT)
891+
client = datastore.Client(client_options=client_options)
892+
893+
query = client.query(kind="Task")
894+
results = list(query.fetch())
895+
for r in results:
896+
print(r)
897+
# [END datastore_regional_endpoints]
898+
899+
return client
900+
901+
886902
def eventual_consistent_query(client):
887903
# [START datastore_eventual_consistent_query]
888904
query = client.query(kind="Task")

datastore/cloud-client/snippets_test.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,3 +301,8 @@ def test_property_by_kind_run_query(self, client):
301301
@backoff.on_exception(backoff.expo, AssertionError, max_time=240)
302302
def test_index_merge_queries(self, client):
303303
snippets.index_merge_queries(client)
304+
305+
@backoff.on_exception(backoff.expo, AssertionError, max_time=240)
306+
def test_regional_endpoint(self, client):
307+
client = snippets.regional_endpoint()
308+
assert client is not None

firestore/cloud-client/snippets.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import threading
1717
from time import sleep
1818

19+
from google.api_core.client_options import ClientOptions
1920
from google.cloud import firestore
2021

2122

@@ -1009,3 +1010,15 @@ def create_and_build_bundle():
10091010
# [END firestore_create_and_build_bundle]
10101011

10111012
return bundle, bundle_buffer
1013+
1014+
1015+
def regional_endpoint():
1016+
# [START firestore_regional_endpoint]
1017+
ENDPOINT = "nam5-firestore.googleapis.com"
1018+
client_options = ClientOptions(api_endpoint=ENDPOINT)
1019+
db = firestore.Client(client_options=client_options)
1020+
1021+
cities_query = db.collection(u'cities').limit(2).get()
1022+
for r in cities_query:
1023+
print(r)
1024+
# [END firestore_regional_endpoint]

firestore/cloud-client/snippets_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -330,3 +330,7 @@ def test_list_document_subcollections():
330330
def test_create_and_build_bundle():
331331
bundle, buffer = snippets.create_and_build_bundle()
332332
assert "latest-stories-query" in bundle.named_queries
333+
334+
335+
def test_regional_endpoint():
336+
snippets.regional_endpoint()

0 commit comments

Comments
 (0)