From 5d790d612bfe9242b7624fe09a8ee5e1672941a5 Mon Sep 17 00:00:00 2001 From: PengFei Li Date: Tue, 14 Jan 2025 18:58:19 +0800 Subject: [PATCH] Add proto Signed-off-by: PengFei Li --- gensrc/proto/internal_service.proto | 13 ++++++++++++- gensrc/proto/types.proto | 15 +++++++++++++++ 2 files changed, 27 insertions(+), 1 deletion(-) diff --git a/gensrc/proto/internal_service.proto b/gensrc/proto/internal_service.proto index 7d3f3b030508f..209e5d47db5af 100644 --- a/gensrc/proto/internal_service.proto +++ b/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); }; diff --git a/gensrc/proto/types.proto b/gensrc/proto/types.proto index b203dcc491042..0d3305ec99337 100644 --- a/gensrc/proto/types.proto +++ b/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; +}