From 15c96e52de3034a1859d0464d40a7d7a22ac9c0f Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Tue, 5 Jul 2022 15:08:29 +0800
Subject: [PATCH] fix CompensateErrorBranch

---
 dtmgrpc/workflow/imp.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dtmgrpc/workflow/imp.go b/dtmgrpc/workflow/imp.go
index 3477151..d58a320 100644
--- a/dtmgrpc/workflow/imp.go
+++ b/dtmgrpc/workflow/imp.go
@@ -160,9 +160,10 @@ func (wf *Workflow) callPhase2(branchID string, fn WfPhase2Func) error {
 func (wf *Workflow) recordedDo(fn func(bb *dtmcli.BranchBarrier) *stepResult) *stepResult {
 	sr := wf.recordedDoInner(fn)
 	// if options not enabled, only successful branch need to be compensated
-	if !wf.Options.CompensateErrorBranch && wf.currentRollbackItem != nil && sr.Status == dtmcli.ResultSuccess {
+	if !wf.Options.CompensateErrorBranch && wf.currentRollbackItem != nil && sr.Status == dtmcli.StatusSucceed {
 		wf.failedOps = append(wf.failedOps, *wf.currentRollbackItem)
 	}
+	wf.currentRollbackItem = nil
 	return sr
 }