adding rollback

matrixorigin · Dec 17, 2024 · 5cc78b6 · 5cc78b6
1 parent 10ef75a
commit 5cc78b6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pkg/vm/engine/tae/db/controller.go b/pkg/vm/engine/tae/db/controller.go
@@ -254,6 +254,15 @@ func (c *Controller) handleToReplayCmd(cmd *controlCmd) {
 		})); err != nil {
 		return
 	}
+	rollbackSteps.Push(struct {
+		fn   func() error
+		desc string
+	}{
+		fn: func() error {
+			return c.db.TxnServer.SwitchTxnHandleStateTo(rpc2.TxnLocalHandle)
+		},
+		desc: "rollback txn handle state",
+	})
 
 	// 4. build logtail tunnel to the new write candidate
 	// TODO