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

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.system.entity.EamBaseRepairDepart;
import org.jeecg.modules.system.entity.EamBaseRepairDepartUser;
import org.jeecg.modules.system.mapper.EamBaseRepairDepartUserMapper;
import org.jeecg.modules.system.model.RepairDepartIdModel;
import org.jeecg.modules.system.service.IEamBaseRepairDepartService;
import org.jeecg.modules.system.service.IEamBaseRepairDepartUserService;
import org.springframework.stereotype.Service;

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

/**
 * @Description: 用户维修部门/班组
 * @author: LiuS
 * @create: 2023-03-27 11:58
 */
@Service
public class EamBaseRepairDepartUserServiceImpl extends ServiceImpl<EamBaseRepairDepartUserMapper, EamBaseRepairDepartUser> implements IEamBaseRepairDepartUserService {

    @Resource
    private IEamBaseRepairDepartService eamBaseRepairDepartService;

    /**
     * 根据用户id查询维修部门/班组信息
     */
    @Override
    public List<RepairDepartIdModel> queryRepairDepartIdsOfUser(String userId) {
        LambdaQueryWrapper<EamBaseRepairDepartUser> queryUserRepDep = new LambdaQueryWrapper<>();
        LambdaQueryWrapper<EamBaseRepairDepart> queryRepDep = new LambdaQueryWrapper<>();
        try {
            queryUserRepDep.eq(EamBaseRepairDepartUser::getUserId, userId);
            List<String> repRepDepIdList = new ArrayList<>();
            List<RepairDepartIdModel> repDepIdModelList = new ArrayList<>();
            List<EamBaseRepairDepartUser> userRepDepList = this.list(queryUserRepDep);
            if (userRepDepList != null && !userRepDepList.isEmpty()) {
                for (EamBaseRepairDepartUser eamBaseRepairDepartUser : userRepDepList) {
                    repRepDepIdList.add(eamBaseRepairDepartUser.getDepartId());
                }
                queryRepDep.in(EamBaseRepairDepart::getId, repRepDepIdList);
                List<EamBaseRepairDepart> repDepList = eamBaseRepairDepartService.list(queryRepDep);
                if (repDepList != null && !repDepList.isEmpty()) {
                    for (EamBaseRepairDepart eamBaseRepairDepart : repDepList) {
                        repDepIdModelList.add(new RepairDepartIdModel().convertByUserRepairDepart(eamBaseRepairDepart));
                    }
                }
                return repDepIdModelList;
            }
        } catch (Exception e) {
            e.fillInStackTrace();
        }
        return null;
    }
}