MeterGroupTree.jsx

import React,{useState} from 'react';
import { connect } from 'umi';
import { TreeSelect } from 'antd'
// 部门树
const DepartTree = (props) => {
    const { treeData, onChange } = props
    let allUser = []
    const initTreeData = (tree) => {
        if (!tree || tree?.length === 0) return;
        // return tree && tree?.length > 0 && 
        if(tree && tree?.length > 0){
            const arr =  tree?.map(item => {
                if(item.meterAttrList && item.meterAttrList.length > 0){
                    const children = initTreeData(item.meterAttrList);
                    return {
                        title: item.name,
                        key: `par-${item.id}`,
                        id: item.id,
                        children,
                        selectable: true,
                        unitName: item.unitName || ''
                    }
                }
                return {
                    title: item.name,
                    key: `${item.id}`,
                    id: item.id,
                    selectable: true,
                    unitName: item.unitName || ''
                }
            })
            return arr
        }
    }
    const treeDataList = initTreeData(treeData)
    const onSelect = (value, node) => {
        // console.log(value, node)
        // const arr = node?.children?.length > 0 ? [...meterArr , ...node?.children] : [...meterArr , node]
        // setData(arr)
        // onChange(node.id)
        // props?.selectParams(arr)
    }
    const changeAction = (value) => {
        console.log(value)
        props?.selectParams(value)
    }

    return <TreeSelect
        // value={value}
        style={{ width: '100%' }}
        // allowClear
        placeholder="请选择"
        treeData={treeDataList}
        onSelect={onSelect}
        onChange={changeAction}
        multiple
        showSearch
        treeNodeFilterProp="title"
    />
}

export default connect(({ user,commonDevice, loading }) => ({
    user,commonDevice,
    loading:loading.models.commonDevice || loading.models.user
}))(DepartTree)