feat: Add QueryCancellationAction message in executor protos

Google APIs · copybara-github · commit e3a34df5b36e · 2024-05-13T23:19:22.000-07:00
PiperOrigin-RevId: 633451546
diff --git a/google/spanner/executor/v1/cloud_executor.proto b/google/spanner/executor/v1/cloud_executor.proto
@@ -132,6 +132,9 @@ message SpannerAction {
 
     // Action to execute change stream query.
     ExecuteChangeStreamQuery execute_change_stream_query = 50;
+
+    // Query cancellation action for testing the cancellation of a query.
+    QueryCancellationAction query_cancellation = 51;
   }
 }
 
@@ -1048,6 +1051,16 @@ message GetOperationAction {
   string operation = 1;
 }
 
+// Query cancellation action defines the long running query and the cancel query
+// format depening on the Cloud database dialect.
+message QueryCancellationAction {
+  // Long running query.
+  string long_running_sql = 1;
+
+  // Format of the cancel query for the cloud database dialect.
+  string cancel_query = 2;
+}
+
 // Action that cancels an operation.
 message CancelOperationAction {
   // The name of the operation resource to be cancelled.
