kongdeqiang
2025-03-27 6f9623981926d78ce8ccd80bce55ad631bd2c76b
platformx-device-biz/src/main/java/com/by4cloud/platformx/device/service/impl/DeviceDemandSubServiceImpl.java
@@ -1,10 +1,18 @@
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;
/**
 * 设备需求计划设备子表
 *
@@ -12,5 +20,22 @@
 * @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;
   }
}