admin-sjzx/src/components/tree/pms/pointTree.vue

<template>
    <div class='point-tree' style='height: 100%; width: 100%; display: flex; flex-direction: column'>
        <el-select
            v-model='formData.statisticalType'
            placeholder='请选择'
            style='min-width: unset; padding: 10px 10px 0'
            @change='loadData'
        >
            <el-option
                v-for='item in classificationData'
                :key='item.id'
                :label='item.name'
                :value='item.id'
            ></el-option>
        </el-select>
        <div style='flex: 1; overflow: hidden'>
            <Tree ref='treRef' :data='tree' style='width: 100%; height: 100%' :canExpand='false'  v-bind='$attrs' />
        </div>
    </div>
</template>

<script lang='ts' setup>
import { nextTick, onMounted, ref, useAttrs } from 'vue'
import Tree from '../index.vue'
import { useAdminInfo } from '@/stores/adminInfo'
import { useDictData } from '@/stores/dictData'
import { getTerminalTreeForFive } from '@/api/device-boot/terminalTree'
import { useConfig } from '@/stores/config'

defineOptions({
    name: 'pms/pointTree'
})
const emit = defineEmits(['init'])
const attrs = useAttrs()
const adminInfo = useAdminInfo()
const dictData = useDictData()
const config = useConfig()
const classificationData = dictData.getBasicData('Statistical_Type', ['Report_Type'])
const tree = ref()
const treRef = ref()
const formData = ref({
    deptIndex: adminInfo.$state.deptIndex,
    monitorFlag: 2,
    powerFlag: 2,
    loadType: null,
    manufacturer: null,
    serverName: 'event-boot',
    statisticalType: classificationData[0].id,
    scale: null
})
const loadData = () => {
    let form = JSON.parse(JSON.stringify(formData.value))
    form.statisticalType = classificationData.find((item: any) => item.id == form.statisticalType)
    let nodeKey = ''
    getTerminalTreeForFive(form).then(res => {
        console.log(res)
        res.data.forEach((item: any) => {
            item.icon = 'fa-solid fa-synagogue'
            item.color = config.getColorVal('elementUiPrimary')
            item.children.forEach((item2: any) => {
                item2.icon = 'fa-solid fa-city'
                item.color = config.getColorVal('elementUiPrimary')
                item2.children.forEach((item3: any) => {
                    item3.icon = 'fa-solid fa-building'
                    item3.color = config.getColorVal('elementUiPrimary')
                    item3.children.forEach((item4: any) => {
                        item4.icon = 'fa-solid fa-tower-observation'
                        item4.color = config.getColorVal('elementUiPrimary')
                        item4.children.forEach((item5: anyObj) => {
                            if ((!attrs['current-node-key']) && !nodeKey) {
                                nodeKey = item5.id
                                emit('init', item5)
                            }
                            item5.icon = 'fa-solid fa-location-dot'
                            item5.color = config.getColorVal('elementUiPrimary')
                            if (item5.comFlag === 0) {
                                item5.color = 'red !important'
                            } else if (item5.comFlag === 1) {
                                item5.color = '#00f93b !important'
                            } else if (item5.comFlag === 2) {
                                item5.color = '#8c8c8c !important'
                            }
                        })
                    })
                })
            })
        })
        tree.value = res.data
        if (nodeKey) {
            nextTick(() => {
                treRef.value.treeRef.setCurrentKey(nodeKey)
                treRef.value.treeRef.setExpandedKeys(nodeKey)
            })
        }
    })
}
loadData()
</script>