n8n-enterprise-unlocked/packages/core/src/execution-engine/partial-execution-utils/clean-run-data.ts

import type { INode, IRunData } from 'n8n-workflow';

import type { DirectedGraph } from './directed-graph';

/**
 * Returns new run data that does not contain data for any node that is a child
 * of any start node.
 * This does not mutate the `runData` being passed in.
 */
export function cleanRunData(
	runData: IRunData,
	graph: DirectedGraph,
	startNodes: Set<INode>,
): IRunData {
	const newRunData: IRunData = { ...runData };

	for (const startNode of startNodes) {
		delete newRunData[startNode.name];
		const children = graph.getChildren(startNode);

		for (const child of children) {
			delete newRunData[child.name];
		}
	}

	// Remove run data for all nodes that are not part of the subgraph
	for (const nodeName of Object.keys(newRunData)) {
		if (!graph.hasNode(nodeName)) {
			// remove run data for node that is not part of the graph
			delete newRunData[nodeName];
		}
	}

	return newRunData;
}