java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > java树形结构

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

作者:迪霸戈

基于springboot + vue3 elementPlus实现树形结构数据的添加、删除和页面展示,本文通过示例代码给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧

基于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>

到此这篇关于java+vue3+el-tree实现树形结构操作的文章就介绍到这了,更多相关java树形结构内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文