From 1eef460423fc852ebfa1ff425dc1414d5e147100 Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Thu, 21 Dec 2023 09:02:01 +0800
Subject: [PATCH] 留言模块
---
src/main/java/org/springblade/modules/leave/wrapper/LeaveWordsWrapper.java | 26 +++++++++++++++++++++++++-
1 files changed, 25 insertions(+), 1 deletions(-)
diff --git a/src/main/java/org/springblade/modules/leave/wrapper/LeaveWordsWrapper.java b/src/main/java/org/springblade/modules/leave/wrapper/LeaveWordsWrapper.java
index 45b4ed3..3f8ee87 100644
--- a/src/main/java/org/springblade/modules/leave/wrapper/LeaveWordsWrapper.java
+++ b/src/main/java/org/springblade/modules/leave/wrapper/LeaveWordsWrapper.java
@@ -16,12 +16,14 @@
*/
package org.springblade.modules.leave.wrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springblade.core.mp.support.BaseEntityWrapper;
import org.springblade.core.tool.utils.BeanUtil;
-import org.springblade.modules.assessment.vo.AssessmentScoreVO;
+import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.modules.leave.entity.LeaveWordsEntity;
+import org.springblade.modules.leave.service.ILeaveWordsService;
import org.springblade.modules.leave.vo.LeaveWordsVO;
import java.util.Objects;
import org.springblade.core.tool.node.ForestNodeMerger;
@@ -35,6 +37,12 @@
* @since 2023-12-18
*/
public class LeaveWordsWrapper extends BaseEntityWrapper<LeaveWordsEntity, LeaveWordsVO> {
+
+ private static final ILeaveWordsService leaveWordsService;
+
+ static {
+ leaveWordsService = SpringUtil.getBean(ILeaveWordsService.class);
+ }
public static LeaveWordsWrapper build() {
return new LeaveWordsWrapper();
@@ -65,5 +73,21 @@
pageVo.setRecords(ForestNodeMerger.merge(collect));
return pageVo;
}
+ @Override
+ public IPage<LeaveWordsVO> pageVO(IPage<LeaveWordsEntity> pages) {
+ List<LeaveWordsVO> records = listVO(pages.getRecords());
+ for (LeaveWordsVO vo: records) {
+ QueryWrapper<LeaveWordsEntity> leaveWordsEntityQueryWrapper = new QueryWrapper<>();
+ leaveWordsEntityQueryWrapper.eq("parent_id", vo.getId());
+ List<LeaveWordsEntity> entities = leaveWordsService.list(leaveWordsEntityQueryWrapper);
+// vo.setHasChildren(entities.size() > 0);
+ if (entities.size() > 0) {
+ vo.setChildren(listVO(entities));
+ }
+ }
+ IPage<LeaveWordsVO> pageVo = new Page<>(pages.getCurrent(), pages.getSize(), pages.getTotal());
+ pageVo.setRecords(records);
+ return pageVo;
+ }
}
--
Gitblit v1.9.3