package org.jeecg.modules.mdc.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.base.entity.Factory;
import org.jeecg.modules.base.model.FactoryIdModel;
import org.jeecg.modules.base.service.IFactoryService;
import org.jeecg.modules.mdc.entity.MdcEquipment;
import org.jeecg.modules.mdc.entity.FactoryEquipment;
import org.jeecg.modules.mdc.mapper.FactoryEquipmentMapper;
import org.jeecg.modules.mdc.service.IMdcEquipmentService;
import org.jeecg.modules.mdc.service.IFactoryEquipmentService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author: LiuS
 * @create: 2023-03-28 10:31
 */
@Service
public class FactoryEquipmentServiceImpl extends ServiceImpl<FactoryEquipmentMapper, FactoryEquipment> implements IFactoryEquipmentService {

    @Resource
    private IFactoryService mdcFactoryService;
    @Resource
    private IMdcEquipmentService mdcEquipmentService;

    /**
     * 根据设备id查询产线信息
     */
    @Override
    public List<FactoryIdModel> queryFactoryIdsOfEquipment(String equipmentId) {
        LambdaQueryWrapper<FactoryEquipment> queryEquipmentFac = new LambdaQueryWrapper<>();
        LambdaQueryWrapper<Factory> queryFac = new LambdaQueryWrapper<>();
        try {
            queryEquipmentFac.eq(FactoryEquipment::getEquipmentId, equipmentId);
            List<String> facIdList = new ArrayList<>();
            List<FactoryIdModel> facIdModelList = new ArrayList<>();
            List<FactoryEquipment> equipmentProList = this.list(queryEquipmentFac);
            if (equipmentProList != null && !equipmentProList.isEmpty()) {
                for (FactoryEquipment factoryEquipment : equipmentProList) {
                    facIdList.add(factoryEquipment.getFactoryId());
                }
                queryFac.in(Factory::getId, facIdList);
                List<Factory> proList = mdcFactoryService.list(queryFac);
                if (proList != null && !proList.isEmpty()) {
                    for (Factory mdcFactory : proList) {
                        facIdModelList.add(new FactoryIdModel().convertByUserFactory(mdcFactory));
                    }
                }
                return facIdModelList;
            }
        } catch (Exception e) {
            e.fillInStackTrace();
        }
        return null;
    }
    /**
     * 根据指定产线id查询设备信息
     */
    @Override
    public List<MdcEquipment> queryEquipmentsOfFactory(String factoryId){
        List<String> equipmentIds = this.list(new LambdaQueryWrapper<FactoryEquipment>().eq(FactoryEquipment::getFactoryId, factoryId))
                .stream().map(FactoryEquipment::getEquipmentId).collect(Collectors.toList());
        if (!equipmentIds.isEmpty()) {
            return mdcEquipmentService.list(new LambdaQueryWrapper<MdcEquipment>().in(MdcEquipment::getId, equipmentIds));
        }
        return null;
    }

    /**
     * 根据指定产线ids查询设备信息
     */
    @Override
    public List<MdcEquipment> queryEquipmentsOfFactorys(List<String> factoryIds){
        List<String> equipmentIds = this.list(new LambdaQueryWrapper<FactoryEquipment>()
                .in(FactoryEquipment::getFactoryId, factoryIds)).stream().map(FactoryEquipment::getEquipmentId).collect(Collectors.toList());
        if (!equipmentIds.isEmpty()) {
            return mdcEquipmentService.list(new LambdaQueryWrapper<MdcEquipment>().in(MdcEquipment::getId, equipmentIds));
        }
        return null;
    }
}