From dd2cbe9c6f7360d0f50da4e5f0fc506ad3915d8b Mon Sep 17 00:00:00 2001
From: lihang <lihang@vci-tech.com>
Date: 星期一, 10 七月 2023 18:04:43 +0800
Subject: [PATCH] 业务类型修改功能修正

---
 Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/wrapper/RoleWrapper.java |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

diff --git a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/wrapper/RoleWrapper.java b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/wrapper/RoleWrapper.java
index 39da219..ce183f9 100644
--- a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/wrapper/RoleWrapper.java
+++ b/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);
+	}
+
 }

--
Gitblit v1.9.3