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.system.entity.MdcEquipmentDepart;
import org.jeecg.modules.system.mapper.MdcEquipmentDepartMapper;
import org.jeecg.modules.mdc.service.IMdcEquipmentDepartService;
import org.jeecg.modules.system.entity.SysDepart;
import org.jeecg.modules.system.model.DepartIdModel;
import org.jeecg.modules.system.service.ISysDepartService;
import org.springframework.stereotype.Service;

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

/**
 * @author: LiuS
 * @create: 2023-03-28 10:14
 */
@Service
public class MdcEquipmentDepartServiceImpl extends ServiceImpl<MdcEquipmentDepartMapper, MdcEquipmentDepart> implements IMdcEquipmentDepartService {

    @Resource
    private ISysDepartService sysDepartService;

    /**
     * 根据设备id查询部门信息
     */
    @Override
    public List<DepartIdModel> queryDepartIdsOfEquipment(String equipmentId) {
        LambdaQueryWrapper<MdcEquipmentDepart> queryEquipmentDep = new LambdaQueryWrapper<>();
        LambdaQueryWrapper<SysDepart> queryDep = new LambdaQueryWrapper<>();
        try {
            queryEquipmentDep.eq(MdcEquipmentDepart::getEquipmentId, equipmentId);
            List<String> depIdList = new ArrayList<>();
            List<DepartIdModel> depIdModelList = new ArrayList<>();
            List<MdcEquipmentDepart> equipmentDepartList = this.list(queryEquipmentDep);
            if (equipmentDepartList != null && !equipmentDepartList.isEmpty()) {
                for (MdcEquipmentDepart equipmentDepart : equipmentDepartList) {
                    depIdList.add(equipmentDepart.getDepId());
                }
                queryDep.in(SysDepart::getId, depIdList);
                List<SysDepart> depList = sysDepartService.list(queryDep);
                if (depList != null && !depList.isEmpty()) {
                    for (SysDepart depart : depList) {
                        depIdModelList.add(new DepartIdModel().convertByUserDepart(depart));
                    }
                }
                return depIdModelList;
            }
        } catch (Exception e) {
            e.fillInStackTrace();
        }
        return null;
    }
}