Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/wrapper/DeptWrapper.java

@@ -16,6 +16,9 @@
 */
package com.vci.ubcs.system.wrapper;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.vci.ubcs.common.utils.PageDO2PageVO;
import org.springblade.core.mp.support.BaseEntityWrapper;
import org.springblade.core.tool.constant.BladeConstant;
import org.springblade.core.tool.node.ForestNodeMerger;
@@ -75,4 +78,14 @@
      return ForestNodeMerger.merge(collect);
   }

   public IPage<DeptVO> pageNodeLazyVO(IPage<DeptVO> page) {
      List<DeptVO> deptVOS = page.getRecords().stream().peek(dept -> {
         String category = DictCache.getValue(DictEnum.ORG_CATEGORY, dept.getDeptCategory());
         Objects.requireNonNull(dept).setDeptCategoryName(category);
      }).collect(Collectors.toList());
      List<DeptVO> list = ForestNodeMerger.merge(deptVOS);
      page.setRecords(list);
      return page;
   }

}