qushaowei
2023-08-19 3e79a609f2b7a46dae5085ca5050acb88fc6f0af
lxzn-module-system/lxzn-system-biz/src/main/java/org/jeecg/modules/base/controller/UnitCategoryController.java
@@ -34,6 +34,15 @@
    @GetMapping(value="/getTree")
    public Result<List<UnitCategory>> getTreeList(){
        Result<List<UnitCategory>> resule = new Result<>();
        List<UnitCategory> list = unitCategoryService.lambdaQuery().eq(UnitCategory::getPid, "-1").list();
        if(list.size() == 0){
            UnitCategory unitCategory = new UnitCategory();
            unitCategory.setPid("-1");
            unitCategory.setCode("0");
            unitCategory.setId("0");
            unitCategory.setName("计量单位分类");
            unitCategoryService.save(unitCategory);
        }
        List<UnitCategory> treeList = unitCategoryService.getTreeList();
        resule.setResult(treeList);
        return  resule;
@@ -63,6 +72,15 @@
        Result<List<UnitCategory>> resule = new Result<>();
        //id前端传空字符串时,查询pid为-1的数据返回
        if (id == "") {
            List<UnitCategory> list = unitCategoryService.lambdaQuery().eq(UnitCategory::getPid, "-1").list();
            if(list.size() == 0){
                UnitCategory unitCategory = new UnitCategory();
                unitCategory.setPid("-1");
                unitCategory.setCode("0");
                unitCategory.setId("0");
                unitCategory.setName("计量单位分类");
                unitCategoryService.save(unitCategory);
            }
            LambdaQueryWrapper<UnitCategory> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(UnitCategory::getPid,"-1");
            List<UnitCategory> treeList = unitCategoryService.getList(unitCategoryService.getOne(queryWrapper).getId());