sql: fix dispatchReadCommittedStmtToExecutionEngine for pausable portals

If we want to preserve retry information then I think we need to save
these fields in the pauseInfo.

Informs: #145377

Release note: None

Author: michae2

pkg/sql/conn_executor_exec.go

@@ -2722,6 +2722,17 @@ func (ex *connExecutor) rollbackSQLTransaction(
 func (ex *connExecutor) dispatchReadCommittedStmtToExecutionEngine(
 	ctx context.Context, p *planner, res RestrictedCommandResult,
 ) error {
+	getPausablePortalInfo := func() *portalPauseInfo {
+		if p != nil && p.pausablePortal != nil {
+			return p.pausablePortal.pauseInfo
+		}
+		return nil
+	}
+	if ppInfo := getPausablePortalInfo(); ppInfo != nil {
+		p.autoRetryStmtReason = ppInfo.dispatchReadCommittedStmtToExecutionEngine.autoRetryStmtReason
+		p.autoRetryStmtCounter = ppInfo.dispatchReadCommittedStmtToExecutionEngine.autoRetryStmtCounter
+	}
+
 	readCommittedSavePointToken, err := ex.state.mu.txn.CreateSavepoint(ctx)
 	if err != nil {
 		return err
@@ -2796,6 +2807,10 @@ func (ex *connExecutor) dispatchReadCommittedStmtToExecutionEngine(
 		}
 		p.autoRetryStmtCounter++
 		p.autoRetryStmtReason = maybeRetriableErr
+		if ppInfo := getPausablePortalInfo(); ppInfo != nil {
+			ppInfo.dispatchReadCommittedStmtToExecutionEngine.autoRetryStmtReason = p.autoRetryStmtReason
+			ppInfo.dispatchReadCommittedStmtToExecutionEngine.autoRetryStmtCounter = p.autoRetryStmtCounter
+		}
 	}
 	return nil
 }

pkg/sql/prepared_stmt.go

@@ -259,6 +259,15 @@ type portalPauseInfo struct {
 		cleanup cleanupFuncStack
 	}
 
+	dispatchReadCommittedStmtToExecutionEngine struct {
+		// autoRetryStmtReason is the planner.autoRetryStmtReason to restore when
+		// resuming a portal.
+		autoRetryStmtReason error
+		// autoRetryStmtCounter is the planner.autoRetryStmtCounter to restore when
+		// resuming a portal.
+		autoRetryStmtCounter int
+	}
+
 	// TODO(sql-session): replace certain fields here with planner.
 	// https://github.com/cockroachdb/cockroach/issues/99625
 	dispatchToExecutionEngine struct {