From 372ec8892f5ed3d11736e75abdd7f33ea34e0d2d Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Tue, 03 Aug 2021 10:40:09 +0800
Subject: [PATCH] 个人考试信息查询接口修改

---
 src/main/java/org/springblade/modules/training/service/impl/TrainingRegistrationServiceImpl.java |   20 +++++++++++++++++++-
 1 files changed, 19 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/springblade/modules/training/service/impl/TrainingRegistrationServiceImpl.java b/src/main/java/org/springblade/modules/training/service/impl/TrainingRegistrationServiceImpl.java
index a5086db..2c3cb68 100644
--- a/src/main/java/org/springblade/modules/training/service/impl/TrainingRegistrationServiceImpl.java
+++ b/src/main/java/org/springblade/modules/training/service/impl/TrainingRegistrationServiceImpl.java
@@ -3,11 +3,15 @@
 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springblade.modules.apply.excel.ApplyInfoExcel;
 import org.springblade.modules.training.entity.TrainingRegistration;
 import org.springblade.modules.training.mapper.TrainingRegistrationMapper;
 import org.springblade.modules.training.service.TrainingRegistrationService;
+import org.springblade.modules.training.util.LongTimeUtil;
 import org.springblade.modules.training.vo.TrainingRegistrationVo;
 import org.springframework.stereotype.Service;
+
+import java.util.List;
 
 /**
  * 报名培训服务实现类
@@ -18,7 +22,11 @@
 
 	@Override
 	public IPage<TrainingRegistrationVo> selectTrainingRegistrationPage(IPage<TrainingRegistrationVo> page, TrainingRegistrationVo trainingRegistration) {
-		return page.setRecords(baseMapper.selectTrainingRegistrationPage(page, trainingRegistration));
+		List<TrainingRegistrationVo> trainingRegistrationVos = baseMapper.selectTrainingRegistrationPage(page, trainingRegistration);
+		trainingRegistrationVos.forEach(trainingRegistrationVo -> {
+			trainingRegistrationVo.setRemainingTime(LongTimeUtil.getDistanceTime(trainingRegistrationVo.getTrainExamTime().getTime(), System.currentTimeMillis()));
+		});
+		return page.setRecords(trainingRegistrationVos);
 	}
 
 	/**
@@ -29,4 +37,14 @@
 	public TrainingRegistrationVo selectTrainingRegistrationInfo(TrainingRegistration trainingRegistration) {
 		return baseMapper.selectTrainingRegistrationInfo(trainingRegistration);
 	}
+
+	/**
+	 * 导出报名数数据
+	 * @param trainingRegistration 培训报名对象
+	 * @return
+	 */
+	@Override
+	public List<ApplyInfoExcel> getTrainingRegistrationExcelList(TrainingRegistrationVo trainingRegistration) {
+		return baseMapper.getTrainingRegistrationExcelList(trainingRegistration);
+	}
 }

--
Gitblit v1.9.3