From 53eb27d92cc1765ed5efecfff9a8cfe222275a95 Mon Sep 17 00:00:00 2001
From: ekexium <ekexium@gmail.com>
Date: Mon, 12 Apr 2021 17:59:17 +0800
Subject: [PATCH] fix: allow txn to rollback after commit failed

Signed-off-by: ekexium <ekexium@gmail.com>
---
 src/transaction/transaction.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/transaction/transaction.rs b/src/transaction/transaction.rs
index 1fe8be4..35d3a8c 100644
--- a/src/transaction/transaction.rs
+++ b/src/transaction/transaction.rs
@@ -538,7 +538,9 @@ impl<PdC: PdClient> Transaction<PdC> {
             let status = self.status.read().await;
             if !matches!(
                 *status,
-                TransactionStatus::StartedRollback | TransactionStatus::Active
+                TransactionStatus::StartedRollback
+                    | TransactionStatus::Active
+                    | TransactionStatus::StartedCommit
             ) {
                 return Err(Error::OperationAfterCommitError);
             }