import FlowNodeModel from '../models/FlowNode'; import JobModel from '../models/Job'; import Processor from '../Processor'; import { JOB_STATUS } from '../constants'; function getTargetLength(target) { let length = 0; if (typeof target === 'number') { if (target < 0) { throw new Error('Loop target in number type must be greater than 0'); } length = Math.floor(target); } else { const targets = (Array.isArray(target) ? target : [target]).filter((t) => t != null); length = targets.length; } return length; } export default { async run(node: FlowNodeModel, prevJob: JobModel, processor: Processor) { const [branch] = processor.getBranches(node); const target = processor.getParsedValue(node.config.target, node); const length = getTargetLength(target); if (!branch || !length) { return { status: JOB_STATUS.RESOLVED, result: 0, }; } const job = await processor.saveJob({ status: JOB_STATUS.PENDING, // save loop index result: 0, nodeId: node.id, upstreamId: prevJob?.id ?? null, }); // TODO: add loop scope to stack // processor.stack.push({ // label: node.title, // value: node.id // }); await processor.run(branch, job); return null; }, async resume(node: FlowNodeModel, branchJob, processor: Processor) { const job = processor.findBranchParentJob(branchJob, node) as JobModel; const loop = processor.nodesMap.get(job.nodeId); const [branch] = processor.getBranches(node); const { result, status } = job; // if loop has been done (resolved / rejected), do not care newly executed branch jobs. if (status !== JOB_STATUS.PENDING) { return processor.exit(); } const nextIndex = result + 1; const target = processor.getParsedValue(loop.config.target, node); // branchJob.status === JOB_STATUS.RESOLVED means branchJob is done, try next loop or exit as resolved if (branchJob.status > JOB_STATUS.PENDING) { job.set({ result: nextIndex }); const length = getTargetLength(target); if (nextIndex < length) { await processor.saveJob(job); await processor.run(branch, job); return null; } } // branchJob.status < JOB_STATUS.PENDING means branchJob is rejected, any rejection should cause loop rejected job.set({ status: branchJob.status, }); return job; }, getScope(node, index, processor) { const target = processor.getParsedValue(node.config.target, node); const targets = (Array.isArray(target) ? target : [target]).filter((t) => t != null); const length = getTargetLength(target); const item = typeof target === 'number' ? index : targets[index]; const result = { item, index, length, }; return result; }, };