package com.by4cloud.platformx.device.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.by4cloud.platformx.device.entity.DeviceDemandSub;
|
import com.by4cloud.platformx.device.entity.DeviceDemandTotal;
|
import com.by4cloud.platformx.device.mapper.DeviceDemandSubMapper;
|
import com.by4cloud.platformx.device.mapper.DeviceDemandTotalMapper;
|
import com.by4cloud.platformx.device.service.DeviceDemandSubService;
|
import lombok.AllArgsConstructor;
|
import org.springframework.stereotype.Service;
|
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 设备需求计划设备子表
|
*
|
* @author pig
|
* @date 2025-03-05 15:40:29
|
*/
|
@Service
|
@AllArgsConstructor
|
public class DeviceDemandSubServiceImpl extends ServiceImpl<DeviceDemandSubMapper, DeviceDemandSub> implements DeviceDemandSubService {
|
private DeviceDemandTotalMapper totalMapper;
|
|
@Override
|
public List<DeviceDemandSub> getByPlanId(Long planId) {
|
QueryWrapper<DeviceDemandTotal> wrapper = new QueryWrapper<>();
|
wrapper.lambda()
|
.eq(DeviceDemandTotal::getPlanId,planId);
|
List<DeviceDemandTotal> deviceDemandTotals = totalMapper.selectList(wrapper);
|
List<Long> collect = deviceDemandTotals.stream().map(DeviceDemandTotal::getId).collect(Collectors.toList());
|
|
QueryWrapper<DeviceDemandSub> subwrapper = new QueryWrapper<>();
|
subwrapper.lambda()
|
.in(DeviceDemandSub::getTotalId,collect);
|
List<DeviceDemandSub> subs = list(subwrapper);
|
return subs;
|
}
|
}
|
