Add proto

Signed-off-by: PengFei Li <[email protected]>

gensrc/proto/internal_service.proto

@@ -661,6 +661,15 @@ message PStreamLoadResponse {
     optional string json_result = 1;
 }
 
+message PUpdateTransactionStateRequest {
+    repeated TransactionStatePB states = 1;
+}
+
+// The update result for each transaction state
+message PUpdateTransactionStateResponse {
+    repeated StatusPB results = 1;
+}
+
 service PInternalService {
     rpc transmit_data(PTransmitDataParams) returns (PTransmitDataResult);
     rpc exec_plan_fragment(PExecPlanFragmentRequest) returns (PExecPlanFragmentResult);
@@ -704,6 +713,8 @@ service PInternalService {
     rpc process_dictionary_cache(starrocks.PProcessDictionaryCacheRequest) returns (starrocks.PProcessDictionaryCacheResult);
 
     rpc fetch_arrow_schema(PFetchArrowSchemaRequest) returns (PFetchArrowSchemaResult);
-    
+
     rpc stream_load(PStreamLoadRequest) returns (PStreamLoadResponse);
+
+    rpc update_transaction_state(PUpdateTransactionStateRequest) returns (PUpdateTransactionStateResponse);
 };

gensrc/proto/types.proto

@@ -205,3 +205,18 @@ message PersistentIndexSstableMetaPB {
     // sstables are ordered with the smaller version on the left.
     repeated PersistentIndexSstablePB sstables = 1;
 }
+
+enum TransactionStatusPB {
+    TRANS_UNKNOWN = 0;
+    TRANS_PREPARE = 1;
+    TRANS_COMMITTED = 2;
+    TRANS_VISIBLE = 3;
+    TRANS_ABORTED = 4;
+    TRANS_PREPARED = 5;
+}
+
+message TransactionStatePB {
+    optional int64 txn_id = 1;
+    optional TransactionStatusPB status = 2;
+    optional string reason = 3;
+}