protos

connector/connect/common/src/main/protobuf/spark/connect/base.proto

@@ -281,6 +281,12 @@ message ExecutePlanRequest {
   // server side.
   UserContext user_context = 2;
 
+  // (Optional)
+  // Provide an id for this request. If not provided, it will be generated by the server.
+  // It is returned in every ExecutePlanResponse.operation_id of the ExecutePlan response stream.
+  // The id must be an UUID string of the format `00112233-4455-6677-8899-aabbccddeeff`
+  optional string operation_id = 6;
+
   // (Required) The logical plan to be executed / analyzed.
   Plan plan = 3;
 
@@ -299,13 +305,23 @@ message ExecutePlanRequest {
       google.protobuf.Any extension = 999;
     }
   }
+
+  // Tags to tag the given execution with.
+  // Used by Interrupt with interrupt.tag.
+  repeated string tags = 7;
 }
 
 // The response of a query, can be one or more for each request. Responses belonging to the
 // same input query, carry the same `session_id`.
 message ExecutePlanResponse {
   string session_id = 1;
 
+  // Identifies the ExecutePlan execution.
+  // If set by the client in ExecutePlanRequest.operationId, that value is returned.
+  // Otherwise generated by the server.
+  // It is an UUID string of the format `00112233-4455-6677-8899-aabbccddeeff`
+  string operation_id = 12;
+
   // Union type for the different response messages.
   oneof response_type {
     ArrowBatch arrow_batch = 2;
@@ -616,13 +632,31 @@ message InterruptRequest {
   enum InterruptType {
     INTERRUPT_TYPE_UNSPECIFIED = 0;
 
-    // Interrupt all running executions within session with provided session_id.
+    // Interrupt all running executions within the session with provided the session_id.
     INTERRUPT_TYPE_ALL = 1;
+
+    // Interrupt all running executions within the session with the provided tag.
+    INTERRUPT_TYPE_TAG = 2;
+
+    // Interrupt the running execution within the session with the provided id.
+    INTERRUPT_TYPE_ID = 3;
+  }
+
+  oneof interrupt {
+    // if interrupt_tag == INTERRUPT_TYPE_TAG, interrupt operation with this tag.
+    string operation_tag = 5;
+
+    // if interrupt_tag == INTERRUPT_TYPE_ID, interrupt operation with this operation_id.
+    string operation_id = 6;
   }
 }
 
 message InterruptResponse {
+  // Session id in which the interrupt was running.
   string session_id = 1;
+
+  // Operation ids of the executions which were interrupted.
+  repeated string interrupted_ids = 2;
 }
 
 // Main interface for the SparkConnect service.