Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/wrapper/RoleWrapper.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;
@@ -52,10 +55,17 @@
      return roleVO;
   }


   public List<RoleVO> listNodeVO(List<Role> list) {
      List<RoleVO> collect = list.stream().map(this::entityVO).collect(Collectors.toList());
      return ForestNodeMerger.merge(collect);
   }

   public IPage<RoleVO> pageNodeVO(IPage<Role> page) {
      List<RoleVO> collect = page.getRecords().stream().map(this::entityVO).collect(Collectors.toList());
      List<RoleVO> mergeList = ForestNodeMerger.merge(collect);
      IPage<RoleVO> roleVOIPage = new Page<>();
      roleVOIPage.setRecords(mergeList);
      return PageDO2PageVO.pageDO2PageVO(page,roleVOIPage);
   }

}