package com.zy.acs.manager.manager.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.acs.framework.common.Cools;
import com.zy.acs.manager.manager.entity.AreaAgv;
import com.zy.acs.manager.manager.mapper.AreaAgvMapper;
import com.zy.acs.manager.manager.service.AreaAgvService;
import org.springframework.stereotype.Service;

import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

@Service("areaAgvService")
public class AreaAgvServiceImpl extends ServiceImpl<AreaAgvMapper, AreaAgv> implements AreaAgvService {

    @Override
    public List<Long> queryAgvIdsByAreaIds(List<Long> areaIds) {
        if (Cools.isEmpty(areaIds)) {
            return Collections.emptyList();
        }
        return this.list(new LambdaQueryWrapper<AreaAgv>().in(AreaAgv::getAreaId, areaIds).select(AreaAgv::getAgvId))
                .stream().map(AreaAgv::getAgvId).collect(Collectors.toList());
    }

    @Override
    public List<Long> queryAreaIdsByAgvId(Long agvId) {
        if (agvId == null) {
            return Collections.emptyList();
        }
        return this.list(new LambdaQueryWrapper<AreaAgv>().eq(AreaAgv::getAgvId, agvId).select(AreaAgv::getAreaId))
                .stream().map(AreaAgv::getAreaId).collect(Collectors.toList());
    }

    @Override
    public void removeByAgvId(Long agvId) {
        this.remove(new LambdaQueryWrapper<AreaAgv>().eq(AreaAgv::getAgvId, agvId));
    }

    @Override
    public void removeByAreaId(Long areaId) {
        this.remove(new LambdaQueryWrapper<AreaAgv>().eq(AreaAgv::getAreaId, areaId));
    }

    @Override
    public List<Long> findAgvIdsWithoutAreaAgv() {
        return this.baseMapper.findAgvIdsWithoutAreaAgv();
    }

}