package com.ruoyi.system.service.impl;
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.common.basic.PageInfo;
|
import com.ruoyi.system.mapper.TSamplingRecordMapper;
|
import com.ruoyi.system.mapper.TSamplingRecordOperationMapper;
|
import com.ruoyi.system.model.TSamplingRecord;
|
import com.ruoyi.system.model.TSamplingRecordOperation;
|
import com.ruoyi.system.query.TSamplingRecordQuery;
|
import com.ruoyi.system.service.TSamplingRecordService;
|
import com.ruoyi.system.vo.TSamplingRecordVO;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.CollectionUtils;
|
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 取样记录管理 服务实现类
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2025-04-08
|
*/
|
@Service
|
public class TSamplingRecordServiceImpl extends ServiceImpl<TSamplingRecordMapper, TSamplingRecord> implements TSamplingRecordService {
|
|
@Autowired
|
private TSamplingRecordOperationMapper samplingRecordOperationMapper;
|
|
@Override
|
public PageInfo<TSamplingRecordVO> pageList(TSamplingRecordQuery query) {
|
PageInfo<TSamplingRecordVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
|
List<TSamplingRecordVO> list = this.baseMapper.pageList(query,pageInfo);
|
if(CollectionUtils.isEmpty(list)){
|
return pageInfo;
|
}
|
|
List<String> recordIds = list.stream().map(TSamplingRecordVO::getId).collect(Collectors.toList());
|
List<TSamplingRecordOperation> recordOperationList = samplingRecordOperationMapper.selectList(Wrappers.lambdaQuery(TSamplingRecordOperation.class)
|
.in(TSamplingRecordOperation::getSamplingId, recordIds));
|
|
list.forEach(item->{
|
item.setSendCount(recordOperationList.stream().filter(recordOperation -> recordOperation.getSamplingId().equals(item.getId()) && recordOperation.getStatus() == 1).count());
|
item.setReceiveCount(recordOperationList.stream().filter(recordOperation -> recordOperation.getSamplingId().equals(item.getId()) && recordOperation.getStatus() == 2).count());
|
item.setReceivedCount(recordOperationList.stream().filter(recordOperation -> recordOperation.getSamplingId().equals(item.getId()) && recordOperation.getStatus() == 3).count());
|
});
|
|
pageInfo.setRecords(list);
|
return pageInfo;
|
}
|
}
|
