chore: format
Joel
1 year ago
parent
95da3a4cf1
commit
c23fe3b67f
@ -0,0 +1,132 @@
|
|||||||
|
import type { NodeTracing } from '@/types/workflow'
|
||||||
|
|
||||||
|
type TracingNodeProps = {
|
||||||
|
id: string
|
||||||
|
uniqueId: string
|
||||||
|
isParallel: boolean
|
||||||
|
data: NodeTracing | null
|
||||||
|
children: TracingNodeProps[]
|
||||||
|
parallelTitle?: string
|
||||||
|
branchTitle?: string
|
||||||
|
hideNodeInfo?: boolean
|
||||||
|
hideNodeProcessDetail?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildLogTree(nodes: NodeTracing[], t: (key: string) => string): TracingNodeProps[] {
|
||||||
|
const rootNodes: TracingNodeProps[] = []
|
||||||
|
const parallelStacks: { [key: string]: TracingNodeProps } = {}
|
||||||
|
const levelCounts: { [key: string]: number } = {}
|
||||||
|
const parallelChildCounts: { [key: string]: Set<string> } = {}
|
||||||
|
let uniqueIdCounter = 0
|
||||||
|
const getUniqueId = () => {
|
||||||
|
uniqueIdCounter++
|
||||||
|
return `unique-${uniqueIdCounter}`
|
||||||
|
}
|
||||||
|
|
||||||
|
const getParallelTitle = (parentId: string | null): string => {
|
||||||
|
const levelKey = parentId || 'root'
|
||||||
|
if (!levelCounts[levelKey])
|
||||||
|
levelCounts[levelKey] = 0
|
||||||
|
|
||||||
|
levelCounts[levelKey]++
|
||||||
|
|
||||||
|
const parentTitle = parentId ? parallelStacks[parentId]?.parallelTitle : ''
|
||||||
|
const levelNumber = parentTitle ? Number.parseInt(parentTitle.split('-')[1]) + 1 : 1
|
||||||
|
const letter = parallelChildCounts[levelKey]?.size > 1 ? String.fromCharCode(64 + levelCounts[levelKey]) : ''
|
||||||
|
return `${t('workflow.common.parallel')}-${levelNumber}${letter}`
|
||||||
|
}
|
||||||
|
|
||||||
|
const getBranchTitle = (parentId: string | null, branchNum: number): string => {
|
||||||
|
const levelKey = parentId || 'root'
|
||||||
|
const parentTitle = parentId ? parallelStacks[parentId]?.parallelTitle : ''
|
||||||
|
const levelNumber = parentTitle ? Number.parseInt(parentTitle.split('-')[1]) + 1 : 1
|
||||||
|
const letter = parallelChildCounts[levelKey]?.size > 1 ? String.fromCharCode(64 + levelCounts[levelKey]) : ''
|
||||||
|
const branchLetter = String.fromCharCode(64 + branchNum)
|
||||||
|
return `${t('workflow.common.branch')}-${levelNumber}${letter}-${branchLetter}`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Count parallel children (for figuring out if we need to use letters)
|
||||||
|
for (const node of nodes) {
|
||||||
|
const parent_parallel_id = node.parent_parallel_id ?? node.execution_metadata?.parent_parallel_id ?? null
|
||||||
|
const parallel_id = node.parallel_id ?? node.execution_metadata?.parallel_id ?? null
|
||||||
|
|
||||||
|
if (parallel_id) {
|
||||||
|
const parentKey = parent_parallel_id || 'root'
|
||||||
|
if (!parallelChildCounts[parentKey])
|
||||||
|
parallelChildCounts[parentKey] = new Set()
|
||||||
|
|
||||||
|
parallelChildCounts[parentKey].add(parallel_id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const node of nodes) {
|
||||||
|
const parallel_id = node.parallel_id ?? node.execution_metadata?.parallel_id ?? null
|
||||||
|
const parent_parallel_id = node.parent_parallel_id ?? node.execution_metadata?.parent_parallel_id ?? null
|
||||||
|
const parallel_start_node_id = node.parallel_start_node_id ?? node.execution_metadata?.parallel_start_node_id ?? null
|
||||||
|
const parent_parallel_start_node_id = node.parent_parallel_start_node_id ?? node.execution_metadata?.parent_parallel_start_node_id ?? null
|
||||||
|
|
||||||
|
if (!parallel_id || node.node_type === BlockEnum.End) {
|
||||||
|
rootNodes.push({
|
||||||
|
id: node.id,
|
||||||
|
uniqueId: getUniqueId(),
|
||||||
|
isParallel: false,
|
||||||
|
data: node,
|
||||||
|
children: [],
|
||||||
|
})
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
if (!parallelStacks[parallel_id]) {
|
||||||
|
const newParallelGroup: TracingNodeProps = {
|
||||||
|
id: parallel_id,
|
||||||
|
uniqueId: getUniqueId(),
|
||||||
|
isParallel: true,
|
||||||
|
data: null,
|
||||||
|
children: [],
|
||||||
|
parallelTitle: '',
|
||||||
|
}
|
||||||
|
parallelStacks[parallel_id] = newParallelGroup
|
||||||
|
|
||||||
|
if (parent_parallel_id && parallelStacks[parent_parallel_id]) {
|
||||||
|
const sameBranchIndex = parallelStacks[parent_parallel_id].children.findLastIndex(c =>
|
||||||
|
c.data?.execution_metadata?.parallel_start_node_id === parent_parallel_start_node_id || c.data?.parallel_start_node_id === parent_parallel_start_node_id,
|
||||||
|
)
|
||||||
|
parallelStacks[parent_parallel_id].children.splice(sameBranchIndex + 1, 0, newParallelGroup)
|
||||||
|
newParallelGroup.parallelTitle = getParallelTitle(parent_parallel_id)
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
newParallelGroup.parallelTitle = getParallelTitle(parent_parallel_id)
|
||||||
|
rootNodes.push(newParallelGroup)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const branchTitle = parallel_start_node_id === node.node_id ? getBranchTitle(parent_parallel_id, parallelStacks[parallel_id].children.length + 1) : ''
|
||||||
|
if (branchTitle) {
|
||||||
|
parallelStacks[parallel_id].children.push({
|
||||||
|
id: node.id,
|
||||||
|
uniqueId: getUniqueId(),
|
||||||
|
isParallel: false,
|
||||||
|
data: node,
|
||||||
|
children: [],
|
||||||
|
branchTitle,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
let sameBranchIndex = parallelStacks[parallel_id].children.findLastIndex(c =>
|
||||||
|
c.data?.execution_metadata?.parallel_start_node_id === parallel_start_node_id || c.data?.parallel_start_node_id === parallel_start_node_id,
|
||||||
|
)
|
||||||
|
if (parallelStacks[parallel_id].children[sameBranchIndex + 1]?.isParallel)
|
||||||
|
sameBranchIndex++
|
||||||
|
|
||||||
|
parallelStacks[parallel_id].children.splice(sameBranchIndex + 1, 0, {
|
||||||
|
id: node.id,
|
||||||
|
uniqueId: getUniqueId(),
|
||||||
|
isParallel: false,
|
||||||
|
data: node,
|
||||||
|
children: [],
|
||||||
|
branchTitle,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return rootNodes
|
||||||
|
}
|
||||||
@ -0,0 +1,127 @@
|
|||||||
|
import { BlockEnum } from '@/app/components/workflow/types'
|
||||||
|
import type { NodeTracing } from '@/types/workflow'
|
||||||
|
|
||||||
|
function addTitle({
|
||||||
|
list, level, parallelNumRecord,
|
||||||
|
}: {
|
||||||
|
list: NodeTracing[], level: number, parallelNumRecord: Record<string, number>
|
||||||
|
}, t: any) {
|
||||||
|
let branchIndex = 0
|
||||||
|
list.forEach((node) => {
|
||||||
|
const parallel_id = node.parallel_id ?? node.execution_metadata?.parallel_id ?? null
|
||||||
|
const parallel_start_node_id = node.parallel_start_node_id ?? node.execution_metadata?.parallel_start_node_id ?? null
|
||||||
|
|
||||||
|
const isNotInParallel = !parallel_id || node.node_type === BlockEnum.End
|
||||||
|
if (isNotInParallel)
|
||||||
|
return
|
||||||
|
|
||||||
|
const isParallelStartNode = node.parallelDetail?.isParallelStartNode
|
||||||
|
if (isParallelStartNode)
|
||||||
|
parallelNumRecord.num++
|
||||||
|
|
||||||
|
const letter = parallelNumRecord.num > 1 ? String.fromCharCode(64 + level) : ''
|
||||||
|
const parallelLevelInfo = `${parallelNumRecord.num}${letter}`
|
||||||
|
|
||||||
|
if (isParallelStartNode) {
|
||||||
|
node.parallelDetail!.isParallelStartNode = true
|
||||||
|
node.parallelDetail!.parallelTitle = `${t('workflow.common.parallel')}-${parallelLevelInfo}`
|
||||||
|
}
|
||||||
|
|
||||||
|
const isBrachStartNode = parallel_start_node_id === node.node_id
|
||||||
|
if (isBrachStartNode) {
|
||||||
|
branchIndex++
|
||||||
|
const branchLetter = String.fromCharCode(64 + branchIndex)
|
||||||
|
if (!node.parallelDetail) {
|
||||||
|
node.parallelDetail = {
|
||||||
|
branchTitle: '',
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
node.parallelDetail!.branchTitle = `${t('workflow.common.branch')}-${parallelLevelInfo}-${branchLetter}`
|
||||||
|
}
|
||||||
|
|
||||||
|
if (node.parallelDetail?.children && node.parallelDetail.children.length > 0) {
|
||||||
|
addTitle({
|
||||||
|
list: node.parallelDetail.children,
|
||||||
|
level: level + 1,
|
||||||
|
parallelNumRecord,
|
||||||
|
}, t)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// list => group by parallel_id(parallel tree).
|
||||||
|
const format = (list: NodeTracing[], t: any): NodeTracing[] => {
|
||||||
|
const result: NodeTracing[] = [...list]
|
||||||
|
const parallelFirstNodeMap: Record<string, string> = {}
|
||||||
|
// list to tree by parent_parallel_start_node_id and parallel_start_node_id
|
||||||
|
result.forEach((node) => {
|
||||||
|
const parallel_id = node.parallel_id ?? node.execution_metadata?.parallel_id ?? null
|
||||||
|
const parent_parallel_id = node.parent_parallel_id ?? node.execution_metadata?.parent_parallel_id ?? null
|
||||||
|
const parent_parallel_start_node_id = node.parent_parallel_start_node_id ?? node.execution_metadata?.parent_parallel_start_node_id ?? null
|
||||||
|
const isNotInParallel = !parallel_id || node.node_type === BlockEnum.End
|
||||||
|
if (isNotInParallel)
|
||||||
|
return
|
||||||
|
|
||||||
|
const isParallelStartNode = !parallelFirstNodeMap[parallel_id]
|
||||||
|
if (isParallelStartNode) {
|
||||||
|
const selfNode = { ...node }
|
||||||
|
node.parallelDetail = {
|
||||||
|
isParallelStartNode: true,
|
||||||
|
children: [selfNode],
|
||||||
|
}
|
||||||
|
parallelFirstNodeMap[parallel_id] = node.node_id
|
||||||
|
const isRootLevel = !parent_parallel_id
|
||||||
|
if (isRootLevel)
|
||||||
|
return
|
||||||
|
|
||||||
|
const parentParallelStartNode = result.find(item => item.node_id === parent_parallel_start_node_id)
|
||||||
|
// append to parent parallel start node
|
||||||
|
if (parentParallelStartNode) {
|
||||||
|
if (!parentParallelStartNode?.parallelDetail) {
|
||||||
|
parentParallelStartNode!.parallelDetail = {
|
||||||
|
children: [],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (parentParallelStartNode!.parallelDetail.children)
|
||||||
|
parentParallelStartNode!.parallelDetail.children.push(node)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// append to parallel start node
|
||||||
|
const parallelStartNode = result.find(item => item.node_id === parallelFirstNodeMap[parallel_id])
|
||||||
|
if (parallelStartNode && parallelStartNode.parallelDetail && parallelStartNode!.parallelDetail!.children)
|
||||||
|
parallelStartNode!.parallelDetail!.children.push(node)
|
||||||
|
})
|
||||||
|
|
||||||
|
const filteredInParallelSubNodes = result.filter((node) => {
|
||||||
|
const parallel_id = node.parallel_id ?? node.execution_metadata?.parallel_id ?? null
|
||||||
|
const isNotInParallel = !parallel_id || node.node_type === BlockEnum.End
|
||||||
|
if (isNotInParallel)
|
||||||
|
return true
|
||||||
|
|
||||||
|
const parent_parallel_id = node.parent_parallel_id ?? node.execution_metadata?.parent_parallel_id ?? null
|
||||||
|
|
||||||
|
if (parent_parallel_id)
|
||||||
|
return false
|
||||||
|
|
||||||
|
const isParallelStartNode = node.parallelDetail?.isParallelStartNode
|
||||||
|
if (!isParallelStartNode)
|
||||||
|
return false
|
||||||
|
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
const parallelNumRecord: Record<string, number> = {
|
||||||
|
num: 0,
|
||||||
|
}
|
||||||
|
|
||||||
|
addTitle({
|
||||||
|
list: filteredInParallelSubNodes,
|
||||||
|
level: 1,
|
||||||
|
parallelNumRecord,
|
||||||
|
}, t)
|
||||||
|
|
||||||
|
return filteredInParallelSubNodes
|
||||||
|
}
|
||||||
|
export default format
|
||||||
Loading…
Reference in New Issue