refactor(core): Delete more redundant code across migrations (Part 1) (no-changelog) (#6691)

packages/cli/src/databases/migrations/common/1658930531669-AddNodeIds.ts

@@ -0,0 +1,44 @@
+import type { INode } from 'n8n-workflow';
+import type { WorkflowEntity } from '@db/entities/WorkflowEntity';
+import type { MigrationContext, ReversibleMigration } from '@db/types';
+import { v4 as uuid } from 'uuid';
+
+type Workflow = Pick<WorkflowEntity, 'id'> & { nodes: string | INode[] };
+
+export class AddNodeIds1658930531669 implements ReversibleMigration {
+	async up({ escape, runQuery, runInBatches, parseJson }: MigrationContext) {
+		const tableName = escape.tableName('workflow_entity');
+		const workflowsQuery = `SELECT id, nodes FROM ${tableName}`;
+		await runInBatches<Workflow>(workflowsQuery, async (workflows) => {
+			workflows.forEach(async (workflow) => {
+				const nodes = parseJson(workflow.nodes);
+				nodes.forEach((node: INode) => {
+					if (!node.id) {
+						node.id = uuid();
+					}
+				});
+
+				await runQuery(
+					`UPDATE ${tableName} SET nodes = :nodes WHERE id = :id`,
+					{ nodes: JSON.stringify(nodes) },
+					{ id: workflow.id },
+				);
+			});
+		});
+	}
+
+	async down({ escape, runQuery, runInBatches, parseJson }: MigrationContext) {
+		const tableName = escape.tableName('workflow_entity');
+		const workflowsQuery = `SELECT id, nodes FROM ${tableName}`;
+		await runInBatches<Workflow>(workflowsQuery, async (workflows) => {
+			workflows.forEach(async (workflow) => {
+				const nodes = parseJson(workflow.nodes).map(({ id, ...rest }) => rest);
+				await runQuery(
+					`UPDATE ${tableName} SET nodes = :nodes WHERE id = :id`,
+					{ nodes: JSON.stringify(nodes) },
+					{ id: workflow.id },
+				);
+			});
+		});
+	}
+}