java+vue3+el-tree实现树形结构操作

基于springboot + vue3 elementPlus实现树形结构数据的添加、删除和页面展示

效果如下

代码如下，业务部分可以自行修改

• java后台代码

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.daztk.mes.common.annotation.LogOperation;
import com.daztk.mes.common.utils.Result;
import com.daztk.mes.module.cad.dto.CadPartDto;
import com.daztk.mes.module.cad.entity.CadPart;
import com.daztk.mes.module.cad.entity.CadRelation;
import com.daztk.mes.module.cad.service.ICadPartService;
import com.daztk.mes.module.cad.service.ICadRelationService;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.*;
import java.util.stream.Collectors;

/**
 * <p>
 * 零部件 
 * </p>
 *
 * @since 2024-05-30
 */
@RestController
@RequestMapping("cad/part")
public class CadPartController {

    @Autowired
    ICadPartService service;

    @Autowired
    ICadRelationService relationService;


    /**
     * 获取所有零部件
     * @param dto
     * @return
     */
    @GetMapping("listAll")
    public Result listAll(CadPartDto dto){
        List<CadPart> list = new ArrayList<>();
        try {
            list = service.getList(dto);

            if(CollectionUtils.isNotEmpty(list)){
                list = getTree(list);
            }
        } catch (Exception e) {
            e.printStackTrace();
            return new Result().err();
        }
        return new Result().ok().setData(list);
    }

    @PostMapping("add")
    @LogOperation(tags = "CAD管理-零部件",serviceClass = ICadPartService.class,entityClass = CadPart.class)
    public Result add(@RequestBody CadPart cadPart){
        try {
            service.saveData(cadPart);
        } catch (Exception e) {
            e.printStackTrace();
            return new Result().err();
        }
        return new Result().ok().setData(cadPart);
    }


    @DeleteMapping("delete")
    @LogOperation(tags = "CAD管理-零部件",serviceClass = ICadPartService.class,entityClass = CadPart.class)
    public Result delete(@RequestBody Long[] ids){
        try {
            List<Long> idList = Arrays.asList(ids);
            List<Long> list = recurveIdList(idList);
            service.removeByIds(list);
        } catch (Exception e) {
            e.printStackTrace();
            return new Result().err();
        }
        return new Result().ok();
    }

    /**
     * 递归获取id树
     *
     * @param ids
     * @return
     */
    public List<Long> recurveIdList(List<Long> ids){
        List<Long> list = new ArrayList<>();
        for(Long id: ids){
            list.add(id);
            //关系表查子集
            List<CadRelation> cadRelations = relationService.list(new QueryWrapper<CadRelation>().eq("parent_id", id));
            if(CollectionUtils.isNotEmpty(cadRelations)){
                List<Long> partIds = cadRelations.stream().map(CadRelation::getPartId).collect(Collectors.toList());
                List<Long> list1 = recurveIdList(partIds);
                list.addAll(list1);
            }
        }
        return list;
    }


    public List<CadPart> getTree(List<CadPart> treeList){
        List<CadPart> collect = treeList.stream()
                .filter(item -> item.getParentId() == 0)//构造最外层节点，即id=0的节点
                .map(item -> {
                    item.setChildren(getChildren(item, treeList));//id=0的节点就为他设置子节点
                    return item;
                }).
                collect(Collectors.toList());
        return collect;

    }

    private static List<CadPart> getChildren(CadPart treeEntity, List<CadPart> treeEntityList) {
        List<CadPart> collect = treeEntityList.stream()
                .filter(item -> item.getParentId().equals(treeEntity.getId()))//判断当前节点的父id是不是要设置节点的id
                .map(item -> {
                    item.setChildren(getChildren(item, treeEntityList));//如果是 为其设置子节点 通过递归 为每个除了最外层节点的节点设置子节点
                    return item;
                })
                .collect(Collectors.toList());
        return collect;
    }

• 前端代码

<template>
    <div class="custom-tree-container">
        <!-- <p>Using render-content</p>
      <el-tree style="max-width: 600px" :data="dataSource" show-checkbox node-key="id" default-expand-all
        :expand-on-click-node="false" :render-content="renderContent" /> -->
        <el-button type="primary" class="add-btn" icon="el-icon-circle-plus-outline" @click="addEdit">新增零部件</el-button>
        <div style="padding-top: 20px;">
            <el-tree style="max-width: 600px; " :data="dataSource" show-checkbox node-key="id" default-expand-all
            :expand-on-click-node="false" ref="treeRef" :props="mapProps">
            <template #default="{ node, data }">
                <span class="custom-tree-node">
                    <span>{{ node.label }}</span>
                    <span>
                        <a @click="append(data)"> 增加子节点 </a>
                        <a style="margin-left: 8px" @click="removeNode(node, data)"> 删除 </a>
                    </span>
                </span>
            </template>
        </el-tree>
        </div>
    

        <el-dialog :title="title" v-model="drawer" width="600px">
            <el-form :model="partForm" :rules="rules" ref="ruleFormsss" label-width="120px" class="demo-partForm">
                <el-form-item label="数模号：" prop="partNo">
                    <el-input v-model="partForm.partNo" clearable></el-input>
                </el-form-item>
                <el-form-item label="版本号：" prop="partRev">
                    <el-input v-model="partForm.partRev" clearable></el-input>
                </el-form-item>
                <el-form-item label="名称：" prop="name">
                    <el-input v-model="partForm.name" clearable></el-input>
                </el-form-item>
                <el-form-item label="中文名称：" prop="cName">
                    <el-input v-model="partForm.cName" clearable></el-input>
                </el-form-item>
                <el-form-item label="材料：" prop="material">
                    <el-input v-model="partForm.material" clearable></el-input>
                </el-form-item>
                <el-form-item label="规格：" prop="gauge">
                    <el-input v-model="partForm.gauge" clearable></el-input>
                </el-form-item>
                <el-form-item label="重量：" prop="massqty">
                    <el-input v-model="partForm.massqty" clearable></el-input>
                </el-form-item>
                <el-form-item label="数据类型：" prop="partType">
                    <el-input v-model="partForm.partType" clearable></el-input>
                </el-form-item>
            </el-form>
            <template #footer>
                <span class="dialog-footer">
                    <el-button @click="cancelEdit">取 消</el-button>
                    <el-button type="primary" @click="saveEdit">确 定</el-button>
                </span>
            </template>
        </el-dialog>
    </div>
</template>

<script>
import { reactive, ref, toRefs, unref, onMounted } from 'vue'
import { add, remove, listAll } from '@api/cad/part'
import { ElMessage } from 'element-plus'


export default {
    name: 'part',
    setup() {
        const state = reactive({
            drawer: false,
            title: '',
            query: {
                partName: ''
            },
            dataSource: [
                // {
                //   id: 1,
                //   label: 'Level one 1',
                //   children: []
                //   // children: [
                //   //   {
                //   //     id: 4,
                //   //     label: 'Level two 1-1',
                //   //     children: [
                //   //       {
                //   //         id: 9,
                //   //         label: 'Level three 1-1-1'
                //   //       }
                //   //     ]
                //   //   }
                //   // ]
                // }
            ],
            mapProps: {
                id: 'id',
                label: 'name',
                children: 'children'
            }
        })

        const treeRef = ref(null)

        const append = (data) => {
            state.drawer = true
            state.title = '添加子节点'
            partForm.parentId = data.id
            partForm.partNo = ''
            partForm.partRev = ''
            partForm.name = ''
            partForm.cName = ''
            partForm.material = ''
            partForm.gauge = ''
            partForm.massqty = ''
            partForm.partType = ''

            // debugger
            // 获取当前节点
            // const currentNode = treeRef.value.getNode(data.id)
            // // 获取当前节点在树中的索引
            // // const currentIndex = getNod(data.id)
            // // // 如果索引大于 0，获取前一个节点
            // // if (currentIndex > 0) {
            // //   const prevNode = getNodeAt(currentIndex - 1);
            // //   const prevId = prevNode.id; 
            // //   // 处理前一个节点的 id
            // // }

            // let treeNodeId = data.$treeNodeId //节点id
            // console.info(data)
            // // alert('当前id'+data.id)
            // // alert(data.$treeNodeId)
            // let id = null
            // if (!data.children || data.children.length == 0) {
            //   //没有子节点
            //   id = data.id * 100 + 1
            // } else {
            //   id = data.id * 100 + 1 + data.children.length
            // }
            // const newChild = { id: id, label: data.label + '-' + id, children: [] }
            // const child = [{ id: id, label: data.label + '-' + id, children: [] }]
            // data.children  = child
            // data.push()
            // state.dataSource = [data];
        }

        const removeNode = (node, data) => {
            const ids = []
            ids.push(data.id)
            remove(ids).then(res => {
                if (res.code == 200) {
                    ElMessage({
                        type: 'success',
                        message: '删除成功'
                    })
                    getPartTableData(state.query)
                } else {
                    ElMessage({
                        type: 'error',
                        message: res.msg
                    })
                }
            })
            // dataSource.value = [...dataSource.value];
        }

        const partForm = reactive({
            partNo: '',
            partRev: '',
            name: '',
            cName: '',
            material: '',
            gauge: '',
            massqty: '',
            partType: '',
            parentId: ''
        })

        const ruleFormsss = ref(null)


        const addEdit = () => {
            state.drawer = true
            state.title = '添加零部件'
            partForm.parentId = ''
            partForm.partNo = ''
            partForm.partRev = ''
            partForm.name = ''
            partForm.cName = ''
            partForm.material = ''
            partForm.gauge = ''
            partForm.massqty = ''
            partForm.partType = ''
        }

        // 取消
        const cancelEdit = () => {
            state.drawer = false
            setTimeout(() => {
                const form = unref(partFormsss)
                form.resetFields()
            }, 100)
        }
        // 新建/编辑零部件
        const saveEdit = async () => {
            const form = unref(ruleFormsss)
            if (!form) return
            try {
                await form.validate()
                if (state.title == '添加零部件' || state.title == '添加子节点') {
                    addSavePart()
                }
                if (state.title == '编辑零部件') {
                    updateSavePart()
                }
                state.drawer = false
            } catch (error) {
                console.error(error)
                ElMessage.error('抱歉，您有必填项未填！')
            }
        }


        const addSavePart = () => {
            // let quantity = partForm.quantity + partForm.iQuantity
            const data = {
                partNo: partForm.partNo,
                partRev: partForm.partRev,
                name: partForm.name,
                cName: partForm.cName,
                material: partForm.material,
                gauge: partForm.gauge,
                massqty: partForm.massqty,
                partType: partForm.partType,
                parentId: partForm.parentId
            }
            add(data).then(res => {
                state.drawer = false
                // const treeData = {
                //   id: res.data.id,
                //   label: res.data.name,
                //   children: []
                // }
                // state.dataSource = res.data
                getPartTableData(state.query)
                ElMessage.success('恭喜您，添加成功！')
            })
        }

        const updateSavePart = () => {
            const data = {
                id: state.handleId,
                name: partForm.name,
                manager: partForm.manager,
                startDate: partForm.startDate,
                endDate: partForm.endDate
            }
            update(data).then(res => {
                state.drawer = false
                getPartTableData(state.query)
                ElMessage.success('恭喜您，保存成功！')
            })
        }

        const rules = {
            name: [{ required: true, message: '此处为必填项', trigger: 'blur' }]
        }

        function getPartTableData() {
            listAll(state.query).then(res => {
                state.dataSource = res.data
            })
        }


        onMounted(() => {
            getPartTableData()
        })

        return {
            ...toRefs(state),
            append,
            removeNode,
            addEdit,
            cancelEdit,
            treeRef,
            partForm,
            ruleFormsss,
            saveEdit,
            rules
        }
    }
}




</script>

<style lang="scss" >
.el-tree-node {
  margin-top: 8px;
}
.custom-tree-node {
  flex: 1;
  display: flex;
  align-items: center;
  justify-content: space-between;
  font-size: 18px;
  padding-right: 8px;
}

</style>