chore(samples): added samples for Regional endpoint (GoogleCloudPlatform#9479)

shweta345 · nicain · Mariatta · web-flow · commit 203bf2e7f719 · 2023-04-12T10:35:33.000-07:00
* Added code sample and test for regional endpoints


---------

Co-authored-by: nicain &lt;nicholascain@google.com&gt;
Co-authored-by: Mariatta Wijaya &lt;Mariatta@users.noreply.github.com&gt;
diff --git a/datastore/cloud-client/snippets.py b/datastore/cloud-client/snippets.py
@@ -16,6 +16,7 @@
 import datetime
 from pprint import pprint
 
+from google.api_core.client_options import ClientOptions
 import google.cloud.exceptions
 from google.cloud import datastore  # noqa: I100
 
@@ -883,6 +884,21 @@ def property_by_kind_run_query(client):
     return representations_by_property
 
 
+def regional_endpoint():
+    # [START datastore_regional_endpoints]
+    ENDPOINT = "https://datastore.googleapis.com"
+    client_options = ClientOptions(api_endpoint=ENDPOINT)
+    client = datastore.Client(client_options=client_options)
+
+    query = client.query(kind="Task")
+    results = list(query.fetch())
+    for r in results:
+        print(r)
+    # [END datastore_regional_endpoints]
+
+    return client
+
+
 def eventual_consistent_query(client):
     # [START datastore_eventual_consistent_query]
     query = client.query(kind="Task")
diff --git a/datastore/cloud-client/snippets_test.py b/datastore/cloud-client/snippets_test.py
@@ -301,3 +301,8 @@ def test_property_by_kind_run_query(self, client):
     @backoff.on_exception(backoff.expo, AssertionError, max_time=240)
     def test_index_merge_queries(self, client):
         snippets.index_merge_queries(client)
+
+    @backoff.on_exception(backoff.expo, AssertionError, max_time=240)
+    def test_regional_endpoint(self, client):
+        client = snippets.regional_endpoint()
+        assert client is not None
diff --git a/firestore/cloud-client/snippets.py b/firestore/cloud-client/snippets.py
@@ -16,6 +16,7 @@
 import threading
 from time import sleep
 
+from google.api_core.client_options import ClientOptions
 from google.cloud import firestore
 
 
@@ -1009,3 +1010,15 @@ def create_and_build_bundle():
     # [END firestore_create_and_build_bundle]
 
     return bundle, bundle_buffer
+
+
+def regional_endpoint():
+    # [START firestore_regional_endpoint]
+    ENDPOINT = "nam5-firestore.googleapis.com"
+    client_options = ClientOptions(api_endpoint=ENDPOINT)
+    db = firestore.Client(client_options=client_options)
+
+    cities_query = db.collection(u'cities').limit(2).get()
+    for r in cities_query:
+        print(r)
+    # [END firestore_regional_endpoint]
diff --git a/firestore/cloud-client/snippets_test.py b/firestore/cloud-client/snippets_test.py
@@ -330,3 +330,7 @@ def test_list_document_subcollections():
 def test_create_and_build_bundle():
     bundle, buffer = snippets.create_and_build_bundle()
     assert "latest-stories-query" in bundle.named_queries
+
+
+def test_regional_endpoint():
+    snippets.regional_endpoint()
