| | |
|---|
| | | @Override |
|---|
| | | public IPage<MeasuringInstrumentLedger> listPage(Page page, MeasuringInstrumentLedger measuringInstrumentLedger) { |
|---|
| | | IPage<MeasuringInstrumentLedger> measuringInstrumentLedgerIPage = measuringInstrumentLedgerMapper.listPage(page, measuringInstrumentLedger); |
|---|
| | | List<Integer> types = new ArrayList<>(); |
|---|
| | | types.add(FileNameType.MEASURING.getValue()); |
|---|
| | | types.add(FileNameType.MEASURINGRecord.getValue()); |
|---|
| | | measuringInstrumentLedgerIPage.getRecords().forEach(item -> { |
|---|
| | | LambdaQueryWrapper<MeasuringInstrumentLedgerRecord> measuringInstrumentLedgerRecordLambdaQueryWrapper = new LambdaQueryWrapper<>(); |
|---|
| | | measuringInstrumentLedgerRecordLambdaQueryWrapper.eq(MeasuringInstrumentLedgerRecord::getMeasuringInstrumentLedgerId, item.getId()); |
|---|
| | | List<MeasuringInstrumentLedgerRecord> measuringInstrumentLedgerRecords = measuringInstrumentLedgerRecordMapper.selectList(measuringInstrumentLedgerRecordLambdaQueryWrapper); |
|---|
| | | List<Long> collect = new ArrayList<>(); |
|---|
| | | if(!CollectionUtils.isEmpty(measuringInstrumentLedgerRecords)){ |
|---|
| | | List<Long> collect = measuringInstrumentLedgerRecords.stream().map(MeasuringInstrumentLedgerRecord::getId).collect(Collectors.toList()); |
|---|
| | | collect.add(item.getId()); |
|---|
| | | LambdaQueryWrapper<CommonFile> salesLedgerFileWrapper = new LambdaQueryWrapper<>(); |
|---|
| | | salesLedgerFileWrapper.in(CommonFile::getCommonId, collect); |
|---|
| | | List<CommonFile> commonFiles = commonFileMapper.selectList(salesLedgerFileWrapper); |
|---|
| | | item.setCommonFiles(commonFiles); |
|---|
| | | collect = measuringInstrumentLedgerRecords.stream().map(MeasuringInstrumentLedgerRecord::getId).collect(Collectors.toList()); |
|---|
| | | } |
|---|
| | | collect.add(item.getId()); |
|---|
| | | LambdaQueryWrapper<CommonFile> salesLedgerFileWrapper = new LambdaQueryWrapper<>(); |
|---|
| | | salesLedgerFileWrapper.in(CommonFile::getCommonId, collect) |
|---|
| | | .in(CommonFile::getType,types); |
|---|
| | | List<CommonFile> commonFiles = commonFileMapper.selectList(salesLedgerFileWrapper); |
|---|
| | | item.setCommonFiles(commonFiles); |
|---|
| | | |
|---|
| | | }); |
|---|
| | | return measuringInstrumentLedgerIPage; |
|---|
