修复

xieb

2025-01-21 bd8d88e5e984e59a5c895ec1a254e3dbae1f9f21

 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;
   }

}