go/worker/executor: reinsert aborted txs on epoch transition

go/worker/compute/executor/committee/node.go

@@ -442,6 +442,15 @@ func (n *Node) HandleEpochTransitionLocked(epoch *committee.EpochSnapshot) {
 
 	switch {
 	case epoch.IsExecutorMember():
+		if state, ok := n.state.(StateWaitingForFinalize); ok {
+			// In case we are finalizing a batch and epoch transition occurred,
+			// reinsert the failed transactions back into queue.
+			if err := n.scheduler.AppendTxBatch(state.raw); err != nil {
+				n.logger.Warn("failed reinserting aborted transactions in queue",
+					"err", err,
+				)
+			}
+		}
 		n.transitionLocked(StateWaitingForBatch{})
 	default:
 		n.transitionLocked(StateNotReady{})