diff --git a/src/views/iot/deviceParamAnalysis/index.vue b/src/views/iot/deviceParamAnalysis/index.vue index 0353ca11..45dc7b84 100644 --- a/src/views/iot/deviceParamAnalysis/index.vue +++ b/src/views/iot/deviceParamAnalysis/index.vue @@ -78,6 +78,7 @@ import dayjs from 'dayjs' import { Echart as EChart } from '@/components/Echart' import { OrganizationApi } from '@/api/mes/organization' import { DeviceModelAttributeApi } from '@/api/iot/devicemodelattribute' +import { handleTree } from '@/utils/tree' defineOptions({ name: 'DeviceParamAnalysis' }) @@ -110,7 +111,9 @@ type ApiTreeEquipment = { type ApiTreeOrg = { id: number name: string + parentId?: number | null equipments?: ApiTreeEquipment[] + children?: ApiTreeOrg[] } const message = useMessage() @@ -215,37 +218,47 @@ const handleKeywordChange = () => { } const buildTreeFromApi = (orgs: ApiTreeOrg[]): DeviceTreeNode[] => { - return ( - orgs?.map((org) => { - const equipments = Array.isArray(org?.equipments) ? org.equipments : [] - const equipmentNodes: DeviceTreeNode[] = equipments.map((eq) => { - const params = Array.isArray(eq?.parameters) ? eq.parameters : [] - const paramNodes: DeviceTreeNode[] = params.map((p) => ({ - id: `param-${eq.id}-${p.id}`, - label: p?.name ?? p?.code ?? String(p?.id ?? ''), - type: 'param', - deviceId: Number(eq.id), - modelId: Number(p.id), - paramKey: p?.code ? String(p.code) : undefined, - unit: p?.unit ? String(p.unit) : undefined - })) - - return { - id: `equipment-${org.id}-${eq.id}`, - label: eq?.name ?? String(eq?.id ?? ''), - type: 'device', - children: paramNodes.length ? paramNodes : undefined - } - }) + const normalizedOrgs: ApiTreeOrg[] = (Array.isArray(orgs) ? orgs : []).map((o) => ({ + ...o, + parentId: typeof o?.parentId === 'number' ? o.parentId : Number(o?.parentId ?? 0) || 0 + })) + + const orgTree = handleTree(normalizedOrgs, 'id', 'parentId', 'children') as ApiTreeOrg[] + + const toOrgNode = (org: ApiTreeOrg): DeviceTreeNode => { + const orgChildren = Array.isArray(org?.children) ? org.children.map(toOrgNode) : [] + + const equipments = Array.isArray(org?.equipments) ? org.equipments : [] + const equipmentNodes: DeviceTreeNode[] = equipments.map((eq) => { + const params = Array.isArray(eq?.parameters) ? eq.parameters : [] + const paramNodes: DeviceTreeNode[] = params.map((p) => ({ + id: `param-${eq.id}-${p.id}`, + label: p?.name ?? p?.code ?? String(p?.id ?? ''), + type: 'param', + deviceId: Number(eq.id), + modelId: Number(p.id), + paramKey: p?.code ? String(p.code) : undefined, + unit: p?.unit ? String(p.unit) : undefined + })) return { - id: `org-${org.id}`, - label: org?.name ?? String(org?.id ?? ''), + id: `equipment-${org.id}-${eq.id}`, + label: eq?.name ?? String(eq?.id ?? ''), type: 'device', - children: equipmentNodes.length ? equipmentNodes : undefined + children: paramNodes.length ? paramNodes : undefined } - }) ?? [] - ) + }) + + const children = [...orgChildren, ...equipmentNodes] + return { + id: `org-${org.id}`, + label: org?.name ?? String(org?.id ?? ''), + type: 'device', + children: children.length ? children : undefined + } + } + + return Array.isArray(orgTree) ? orgTree.map(toOrgNode) : [] } const loadTree = async () => {