Skip to content

Commit 1043ba3

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: add dynamic routing header annotation to DatastoreV1 (#400)
* feat: add dynamic routing header annotation to DatastoreV1 PiperOrigin-RevId: 501591105 Source-Link: googleapis/googleapis@c4ebfff Source-Link: googleapis/googleapis-gen@e2d6011 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZTJkNjAxMWRmMzVjM2I3YjU0ZmYzOTA5OGJlNzc5YzEwNTQwOGIxYiJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
1 parent 9c9347b commit 1043ba3

File tree

3 files changed

+190
-308
lines changed

3 files changed

+190
-308
lines changed

‎google/cloud/datastore_v1/services/datastore/async_client.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -397,7 +397,7 @@ async def sample_run_query():
397397
Returns:
398398
google.cloud.datastore_v1.types.RunQueryResponse:
399399
The response for
400-
[Datastore.RunQuery][google.datastore.v1.Datastore.RunQuery].
400+
[Datastore.RunQuery][google.datastore.v1.Datastore.RunQuery].
401401
402402
"""
403403
# Create or coerce a protobuf request object.
@@ -489,7 +489,7 @@ async def sample_run_aggregation_query():
489489
Returns:
490490
google.cloud.datastore_v1.types.RunAggregationQueryResponse:
491491
The response for
492-
[Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
492+
[Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
493493
494494
"""
495495
# Create or coerce a protobuf request object.
@@ -589,7 +589,7 @@ async def sample_begin_transaction():
589589
Returns:
590590
google.cloud.datastore_v1.types.BeginTransactionResponse:
591591
The response for
592-
[Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
592+
[Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
593593
594594
"""
595595
# Create or coerce a protobuf request object.
@@ -852,7 +852,8 @@ async def sample_rollback():
852852
853853
Returns:
854854
google.cloud.datastore_v1.types.RollbackResponse:
855-
The response for [Datastore.Rollback][google.datastore.v1.Datastore.Rollback].
855+
The response for
856+
[Datastore.Rollback][google.datastore.v1.Datastore.Rollback].
856857
(an empty message).
857858
858859
"""
@@ -970,7 +971,7 @@ async def sample_allocate_ids():
970971
Returns:
971972
google.cloud.datastore_v1.types.AllocateIdsResponse:
972973
The response for
973-
[Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
974+
[Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
974975
975976
"""
976977
# Create or coerce a protobuf request object.
@@ -1086,7 +1087,7 @@ async def sample_reserve_ids():
10861087
Returns:
10871088
google.cloud.datastore_v1.types.ReserveIdsResponse:
10881089
The response for
1089-
[Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
1090+
[Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
10901091
10911092
"""
10921093
# Create or coerce a protobuf request object.

‎google/cloud/datastore_v1/services/datastore/client.py

Lines changed: 103 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -530,13 +530,18 @@ def sample_lookup():
530530
# and friendly error handling.
531531
rpc = self._transport._wrapped_methods[self._transport.lookup]
532532

533-
# Certain fields should be provided within the metadata header;
534-
# add these here.
535-
metadata = tuple(metadata) + (
536-
gapic_v1.routing_header.to_grpc_metadata(
537-
(("project_id", request.project_id),)
538-
),
539-
)
533+
header_params = {}
534+
535+
if request.project_id:
536+
header_params["project_id"] = request.project_id
537+
538+
if request.database_id:
539+
header_params["database_id"] = request.database_id
540+
541+
if header_params:
542+
metadata = tuple(metadata) + (
543+
gapic_v1.routing_header.to_grpc_metadata(header_params),
544+
)
540545

541546
# Send the request.
542547
response = rpc(
@@ -598,7 +603,7 @@ def sample_run_query():
598603
Returns:
599604
google.cloud.datastore_v1.types.RunQueryResponse:
600605
The response for
601-
[Datastore.RunQuery][google.datastore.v1.Datastore.RunQuery].
606+
[Datastore.RunQuery][google.datastore.v1.Datastore.RunQuery].
602607
603608
"""
604609
# Create or coerce a protobuf request object.
@@ -613,13 +618,18 @@ def sample_run_query():
613618
# and friendly error handling.
614619
rpc = self._transport._wrapped_methods[self._transport.run_query]
615620

616-
# Certain fields should be provided within the metadata header;
617-
# add these here.
618-
metadata = tuple(metadata) + (
619-
gapic_v1.routing_header.to_grpc_metadata(
620-
(("project_id", request.project_id),)
621-
),
622-
)
621+
header_params = {}
622+
623+
if request.project_id:
624+
header_params["project_id"] = request.project_id
625+
626+
if request.database_id:
627+
header_params["database_id"] = request.database_id
628+
629+
if header_params:
630+
metadata = tuple(metadata) + (
631+
gapic_v1.routing_header.to_grpc_metadata(header_params),
632+
)
623633

624634
# Send the request.
625635
response = rpc(
@@ -681,7 +691,7 @@ def sample_run_aggregation_query():
681691
Returns:
682692
google.cloud.datastore_v1.types.RunAggregationQueryResponse:
683693
The response for
684-
[Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
694+
[Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
685695
686696
"""
687697
# Create or coerce a protobuf request object.
@@ -696,13 +706,18 @@ def sample_run_aggregation_query():
696706
# and friendly error handling.
697707
rpc = self._transport._wrapped_methods[self._transport.run_aggregation_query]
698708

699-
# Certain fields should be provided within the metadata header;
700-
# add these here.
701-
metadata = tuple(metadata) + (
702-
gapic_v1.routing_header.to_grpc_metadata(
703-
(("project_id", request.project_id),)
704-
),
705-
)
709+
header_params = {}
710+
711+
if request.project_id:
712+
header_params["project_id"] = request.project_id
713+
714+
if request.database_id:
715+
header_params["database_id"] = request.database_id
716+
717+
if header_params:
718+
metadata = tuple(metadata) + (
719+
gapic_v1.routing_header.to_grpc_metadata(header_params),
720+
)
706721

707722
# Send the request.
708723
response = rpc(
@@ -772,7 +787,7 @@ def sample_begin_transaction():
772787
Returns:
773788
google.cloud.datastore_v1.types.BeginTransactionResponse:
774789
The response for
775-
[Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
790+
[Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
776791
777792
"""
778793
# Create or coerce a protobuf request object.
@@ -800,13 +815,18 @@ def sample_begin_transaction():
800815
# and friendly error handling.
801816
rpc = self._transport._wrapped_methods[self._transport.begin_transaction]
802817

803-
# Certain fields should be provided within the metadata header;
804-
# add these here.
805-
metadata = tuple(metadata) + (
806-
gapic_v1.routing_header.to_grpc_metadata(
807-
(("project_id", request.project_id),)
808-
),
809-
)
818+
header_params = {}
819+
820+
if request.project_id:
821+
header_params["project_id"] = request.project_id
822+
823+
if request.database_id:
824+
header_params["database_id"] = request.database_id
825+
826+
if header_params:
827+
metadata = tuple(metadata) + (
828+
gapic_v1.routing_header.to_grpc_metadata(header_params),
829+
)
810830

811831
# Send the request.
812832
response = rpc(
@@ -950,13 +970,18 @@ def sample_commit():
950970
# and friendly error handling.
951971
rpc = self._transport._wrapped_methods[self._transport.commit]
952972

953-
# Certain fields should be provided within the metadata header;
954-
# add these here.
955-
metadata = tuple(metadata) + (
956-
gapic_v1.routing_header.to_grpc_metadata(
957-
(("project_id", request.project_id),)
958-
),
959-
)
973+
header_params = {}
974+
975+
if request.project_id:
976+
header_params["project_id"] = request.project_id
977+
978+
if request.database_id:
979+
header_params["database_id"] = request.database_id
980+
981+
if header_params:
982+
metadata = tuple(metadata) + (
983+
gapic_v1.routing_header.to_grpc_metadata(header_params),
984+
)
960985

961986
# Send the request.
962987
response = rpc(
@@ -1035,7 +1060,8 @@ def sample_rollback():
10351060
10361061
Returns:
10371062
google.cloud.datastore_v1.types.RollbackResponse:
1038-
The response for [Datastore.Rollback][google.datastore.v1.Datastore.Rollback].
1063+
The response for
1064+
[Datastore.Rollback][google.datastore.v1.Datastore.Rollback].
10391065
(an empty message).
10401066
10411067
"""
@@ -1066,13 +1092,18 @@ def sample_rollback():
10661092
# and friendly error handling.
10671093
rpc = self._transport._wrapped_methods[self._transport.rollback]
10681094

1069-
# Certain fields should be provided within the metadata header;
1070-
# add these here.
1071-
metadata = tuple(metadata) + (
1072-
gapic_v1.routing_header.to_grpc_metadata(
1073-
(("project_id", request.project_id),)
1074-
),
1075-
)
1095+
header_params = {}
1096+
1097+
if request.project_id:
1098+
header_params["project_id"] = request.project_id
1099+
1100+
if request.database_id:
1101+
header_params["database_id"] = request.database_id
1102+
1103+
if header_params:
1104+
metadata = tuple(metadata) + (
1105+
gapic_v1.routing_header.to_grpc_metadata(header_params),
1106+
)
10761107

10771108
# Send the request.
10781109
response = rpc(
@@ -1153,7 +1184,7 @@ def sample_allocate_ids():
11531184
Returns:
11541185
google.cloud.datastore_v1.types.AllocateIdsResponse:
11551186
The response for
1156-
[Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
1187+
[Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
11571188
11581189
"""
11591190
# Create or coerce a protobuf request object.
@@ -1183,13 +1214,18 @@ def sample_allocate_ids():
11831214
# and friendly error handling.
11841215
rpc = self._transport._wrapped_methods[self._transport.allocate_ids]
11851216

1186-
# Certain fields should be provided within the metadata header;
1187-
# add these here.
1188-
metadata = tuple(metadata) + (
1189-
gapic_v1.routing_header.to_grpc_metadata(
1190-
(("project_id", request.project_id),)
1191-
),
1192-
)
1217+
header_params = {}
1218+
1219+
if request.project_id:
1220+
header_params["project_id"] = request.project_id
1221+
1222+
if request.database_id:
1223+
header_params["database_id"] = request.database_id
1224+
1225+
if header_params:
1226+
metadata = tuple(metadata) + (
1227+
gapic_v1.routing_header.to_grpc_metadata(header_params),
1228+
)
11931229

11941230
# Send the request.
11951231
response = rpc(
@@ -1269,7 +1305,7 @@ def sample_reserve_ids():
12691305
Returns:
12701306
google.cloud.datastore_v1.types.ReserveIdsResponse:
12711307
The response for
1272-
[Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
1308+
[Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
12731309
12741310
"""
12751311
# Create or coerce a protobuf request object.
@@ -1299,13 +1335,18 @@ def sample_reserve_ids():
12991335
# and friendly error handling.
13001336
rpc = self._transport._wrapped_methods[self._transport.reserve_ids]
13011337

1302-
# Certain fields should be provided within the metadata header;
1303-
# add these here.
1304-
metadata = tuple(metadata) + (
1305-
gapic_v1.routing_header.to_grpc_metadata(
1306-
(("project_id", request.project_id),)
1307-
),
1308-
)
1338+
header_params = {}
1339+
1340+
if request.project_id:
1341+
header_params["project_id"] = request.project_id
1342+
1343+
if request.database_id:
1344+
header_params["database_id"] = request.database_id
1345+
1346+
if header_params:
1347+
metadata = tuple(metadata) + (
1348+
gapic_v1.routing_header.to_grpc_metadata(header_params),
1349+
)
13091350

13101351
# Send the request.
13111352
response = rpc(

0 commit comments

Comments
 (0)