Export commitTS of KVTxn

Signed-off-by: Bin Zhang <[email protected]>
tikv · Nov 11, 2024 · be41b52 · be41b52
1 parent c154447
commit be41b52
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/txnkv/transaction/test_probe.go b/txnkv/transaction/test_probe.go
@@ -38,11 +38,6 @@ func (txn TxnProbe) SetStartTS(ts uint64) {
 	txn.startTS = ts
 }
 
-// GetCommitTS returns the commit ts.
-func (txn TxnProbe) GetCommitTS() uint64 {
-	return txn.commitTS
-}
-
 // GetUnionStore returns transaction's embedded unionstore.
 func (txn TxnProbe) GetUnionStore() *unionstore.KVUnionStore {
 	return txn.us

diff --git a/txnkv/transaction/txn.go b/txnkv/transaction/txn.go
@@ -1700,6 +1700,13 @@ func (txn *KVTxn) StartTS() uint64 {
 	return txn.startTS
 }
 
+func (txn *KVTxn) CommitTS() uint64 {
+	if txn != nil {
+		return txn.commitTS
+	}
+	return 0
+}
+
 // Valid returns if the transaction is valid.
 // A transaction become invalid after commit or rollback.
 func (txn *KVTxn) Valid() bool {