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

<template>
    <div class="point-tree">
        <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="c" :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[2].id,
    scale: null
})
const loadData = (val: string) => {
    let obj = classificationData.find(function (i) {
        return i.id === formData.value.statisticalType
    })
    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)
        if (obj.code == 'Power_Network') {
            res.data = [
                {
                    name: '电网拓扑',
                    level: -1,
                    id: 0,
                    children: res.data
                }
            ]
        }
        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>
<style lang="scss">
.point-tree {
    height: 100%;
    width: 100%;
    display: flex;
    flex-direction: column;
    background: #fff;
    border: 1px solid var(--el-border-color);
}
</style>