feat: add dynamic routing header annotation to DatastoreV1

Google APIs · copybara-github · commit c4ebfff4f512 · 2023-01-12T09:46:54.000-08:00
PiperOrigin-RevId: 501591105
diff --git a/google/datastore/v1/BUILD.bazel b/google/datastore/v1/BUILD.bazel
@@ -30,6 +30,7 @@ proto_library(
         "//google/api:annotations_proto",
         "//google/api:client_proto",
         "//google/api:field_behavior_proto",
+        "//google/api:routing_proto",
         "//google/type:latlng_proto",
         "@com_google_protobuf//:struct_proto",
         "@com_google_protobuf//:timestamp_proto",
diff --git a/google/datastore/v1/datastore.proto b/google/datastore/v1/datastore.proto
@@ -19,6 +19,7 @@ package google.datastore.v1;
 import "google/api/annotations.proto";
 import "google/api/client.proto";
 import "google/api/field_behavior.proto";
+import "google/api/routing.proto";
 import "google/datastore/v1/aggregation_result.proto";
 import "google/datastore/v1/entity.proto";
 import "google/datastore/v1/query.proto";
@@ -51,6 +52,10 @@ service Datastore {
       post: "/v1/projects/{project_id}:lookup"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
     option (google.api.method_signature) = "project_id,read_options,keys";
   }
 
@@ -60,22 +65,36 @@ service Datastore {
       post: "/v1/projects/{project_id}:runQuery"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
   }
 
   // Runs an aggregation query.
-  rpc RunAggregationQuery(RunAggregationQueryRequest) returns (RunAggregationQueryResponse) {
+  rpc RunAggregationQuery(RunAggregationQueryRequest)
+      returns (RunAggregationQueryResponse) {
     option (google.api.http) = {
       post: "/v1/projects/{project_id}:runAggregationQuery"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
   }
 
   // Begins a new transaction.
-  rpc BeginTransaction(BeginTransactionRequest) returns (BeginTransactionResponse) {
+  rpc BeginTransaction(BeginTransactionRequest)
+      returns (BeginTransactionResponse) {
     option (google.api.http) = {
       post: "/v1/projects/{project_id}:beginTransaction"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
     option (google.api.method_signature) = "project_id";
   }
 
@@ -86,7 +105,12 @@ service Datastore {
       post: "/v1/projects/{project_id}:commit"
       body: "*"
     };
-    option (google.api.method_signature) = "project_id,mode,transaction,mutations";
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
+    option (google.api.method_signature) =
+        "project_id,mode,transaction,mutations";
     option (google.api.method_signature) = "project_id,mode,mutations";
   }
 
@@ -96,6 +120,10 @@ service Datastore {
       post: "/v1/projects/{project_id}:rollback"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
     option (google.api.method_signature) = "project_id,transaction";
   }
 
@@ -106,6 +134,10 @@ service Datastore {
       post: "/v1/projects/{project_id}:allocateIds"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
     option (google.api.method_signature) = "project_id,keys";
   }
 
@@ -116,6 +148,10 @@ service Datastore {
       post: "/v1/projects/{project_id}:reserveIds"
       body: "*"
     };
+    option (google.api.routing) = {
+      routing_parameters { field: "project_id" }
+      routing_parameters { field: "database_id" }
+    };
     option (google.api.method_signature) = "project_id,keys";
   }
 }
@@ -189,7 +225,8 @@ message RunQueryRequest {
   }
 }
 
-// The response for [Datastore.RunQuery][google.datastore.v1.Datastore.RunQuery].
+// The response for
+// [Datastore.RunQuery][google.datastore.v1.Datastore.RunQuery].
 message RunQueryResponse {
   // A batch of query results (always present).
   QueryResultBatch batch = 1;
@@ -198,7 +235,8 @@ message RunQueryResponse {
   Query query = 2;
 }
 
-// The request for [Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
+// The request for
+// [Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
 message RunAggregationQueryRequest {
   // Required. The ID of the project against which to make the request.
   string project_id = 8 [(google.api.field_behavior) = REQUIRED];
@@ -228,7 +266,8 @@ message RunAggregationQueryRequest {
   }
 }
 
-// The response for [Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
+// The response for
+// [Datastore.RunAggregationQuery][google.datastore.v1.Datastore.RunAggregationQuery].
 message RunAggregationQueryResponse {
   // A batch of aggregation results. Always present.
   AggregationResultBatch batch = 1;
@@ -237,7 +276,8 @@ message RunAggregationQueryResponse {
   AggregationQuery query = 2;
 }
 
-// The request for [Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
+// The request for
+// [Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
 message BeginTransactionRequest {
   // Required. The ID of the project against which to make the request.
   string project_id = 8 [(google.api.field_behavior) = REQUIRED];
@@ -252,7 +292,8 @@ message BeginTransactionRequest {
   TransactionOptions transaction_options = 10;
 }
 
-// The response for [Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
+// The response for
+// [Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction].
 message BeginTransactionResponse {
   // The transaction identifier (always present).
   bytes transaction = 1;
@@ -274,11 +315,10 @@ message RollbackRequest {
   bytes transaction = 1 [(google.api.field_behavior) = REQUIRED];
 }
 
-// The response for [Datastore.Rollback][google.datastore.v1.Datastore.Rollback].
-// (an empty message).
-message RollbackResponse {
-
-}
+// The response for
+// [Datastore.Rollback][google.datastore.v1.Datastore.Rollback]. (an empty
+// message).
+message RollbackResponse {}
 
 // The request for [Datastore.Commit][google.datastore.v1.Datastore.Commit].
 message CommitRequest {
@@ -346,7 +386,8 @@ message CommitResponse {
   google.protobuf.Timestamp commit_time = 8;
 }
 
-// The request for [Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
+// The request for
+// [Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
 message AllocateIdsRequest {
   // Required. The ID of the project against which to make the request.
   string project_id = 8 [(google.api.field_behavior) = REQUIRED];
@@ -357,19 +398,21 @@ message AllocateIdsRequest {
   // database.
   string database_id = 9;
 
-  // 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.
   repeated Key keys = 1 [(google.api.field_behavior) = REQUIRED];
 }
 
-// The response for [Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
+// The response for
+// [Datastore.AllocateIds][google.datastore.v1.Datastore.AllocateIds].
 message AllocateIdsResponse {
   // The keys specified in the request (in the same order), each with
   // its key path completed with a newly allocated ID.
   repeated Key keys = 1;
 }
 
-// The request for [Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
+// The request for
+// [Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
 message ReserveIdsRequest {
   // Required. The ID of the project against which to make the request.
   string project_id = 8 [(google.api.field_behavior) = REQUIRED];
@@ -380,15 +423,14 @@ message ReserveIdsRequest {
   // database.
   string database_id = 9;
 
-  // 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.
   repeated Key keys = 1 [(google.api.field_behavior) = REQUIRED];
 }
 
-// The response for [Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
-message ReserveIdsResponse {
-
-}
+// The response for
+// [Datastore.ReserveIds][google.datastore.v1.Datastore.ReserveIds].
+message ReserveIdsResponse {}
 
 // A mutation to apply to an entity.
 message Mutation {
@@ -501,8 +543,10 @@ message ReadOptions {
 // Options for beginning a new transaction.
 //
 // Transactions can be created explicitly with calls to
-// [Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction] or implicitly by setting
-// [ReadOptions.new_transaction][google.datastore.v1.ReadOptions.new_transaction] in read requests.
+// [Datastore.BeginTransaction][google.datastore.v1.Datastore.BeginTransaction]
+// or implicitly by setting
+// [ReadOptions.new_transaction][google.datastore.v1.ReadOptions.new_transaction]
+// in read requests.
 message TransactionOptions {
   // Options specific to read / write transactions.
   message ReadWrite {
