|
@@ -5,18 +5,22 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
import com.fjhx.base.BaseEntity;
|
|
|
+import com.fjhx.base.BaseIdEntity;
|
|
|
+import com.fjhx.entity.FileInfo;
|
|
|
import com.fjhx.entity.working.WorkingProcedure;
|
|
|
+import com.fjhx.mapper.working.WorkingProcedureMapper;
|
|
|
import com.fjhx.params.working.WorkingProcedureVo;
|
|
|
+import com.fjhx.service.working.WorkingProcedureService;
|
|
|
import com.fjhx.utils.FileClientUtil;
|
|
|
+import com.fjhx.utils.PageUtil;
|
|
|
import com.fjhx.utils.WrapperUtil;
|
|
|
-import com.fjhx.mapper.working.WorkingProcedureMapper;
|
|
|
-import com.fjhx.service.working.WorkingProcedureService;
|
|
|
import org.springblade.core.secure.utils.AuthUtil;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
|
+import java.util.stream.Collectors;
|
|
|
|
|
|
/**
|
|
|
* <p>
|
|
@@ -30,7 +34,7 @@ import java.util.Map;
|
|
|
public class WorkingProcedureServiceImpl extends ServiceImpl<WorkingProcedureMapper, WorkingProcedure> implements WorkingProcedureService {
|
|
|
|
|
|
@Override
|
|
|
- public Page<WorkingProcedure> getPage(Map<String, String> condition) {
|
|
|
+ public Page<WorkingProcedureVo> getPage(Map<String, String> condition) {
|
|
|
|
|
|
QueryWrapper<WorkingProcedure> wrapper = Wrappers.query();
|
|
|
|
|
@@ -39,7 +43,22 @@ public class WorkingProcedureServiceImpl extends ServiceImpl<WorkingProcedureMap
|
|
|
.eq("whether_packing", "whetherPacking")
|
|
|
.createTimeDesc();
|
|
|
|
|
|
- return page(condition, wrapper);
|
|
|
+ Page<WorkingProcedure> page = page(condition, wrapper);
|
|
|
+
|
|
|
+ Page<WorkingProcedureVo> workingProcedureVoPage = PageUtil.copyPage(page, WorkingProcedureVo.class);
|
|
|
+ List<WorkingProcedureVo> records = workingProcedureVoPage.getRecords();
|
|
|
+ if (records.size() == 0) {
|
|
|
+ return workingProcedureVoPage;
|
|
|
+ }
|
|
|
+
|
|
|
+ List<Long> workingProcedureIdList = records.stream().map(BaseIdEntity::getId).collect(Collectors.toList());
|
|
|
+ Map<Long, FileInfo> fileInfoMap = FileClientUtil.getFileInfoMap(workingProcedureIdList);
|
|
|
+
|
|
|
+ for (WorkingProcedureVo workingProcedureVo : records) {
|
|
|
+ workingProcedureVo.setFileInfo(fileInfoMap.get(workingProcedureVo.getId()));
|
|
|
+ }
|
|
|
+
|
|
|
+ return workingProcedureVoPage;
|
|
|
}
|
|
|
|
|
|
@Transactional(rollbackFor = Exception.class)
|