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());

		if(collect !=null && collect.size()>0){
			QueryWrapper<DeviceDemandSub> subwrapper = new QueryWrapper<>();
			subwrapper.lambda()
					.in(DeviceDemandSub::getTotalId,collect);
			List<DeviceDemandSub> subs = list(subwrapper);
			return subs;
		}else {
			return null;
		}
	}
}