package com.dji.sample.wayline.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.dji.sample.wayline.dao.IWaylineJobBreakPointMapper;
import com.dji.sample.wayline.model.entity.WaylineJobBreakPointEntity;
import com.dji.sample.wayline.service.IWaylineJobBreakPointService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

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

/**
 * @PROJECT_NAME: drone
 * @DESCRIPTION:
 * @USER: aix
 * @DATE: 2024/3/23 11:21
 */
@Service
@Transactional
public class WaylineJobBreakPointServiceImpl extends ServiceImpl<IWaylineJobBreakPointMapper,WaylineJobBreakPointEntity> implements IWaylineJobBreakPointService {

    @Autowired
    private IWaylineJobBreakPointMapper mapper;

    @Override
    public boolean addWaylineJobBreakPoint(WaylineJobBreakPointEntity entity) {
        int insertId = mapper.insert(entity);
        return insertId>0;
    }

    @Override
    public List<String> selectHaveBreak(List<String> ids) {
        LambdaQueryWrapper<WaylineJobBreakPointEntity> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.select(WaylineJobBreakPointEntity::getJobId).in(WaylineJobBreakPointEntity::getJobId, ids);
        List<WaylineJobBreakPointEntity> entities = mapper.selectList(queryWrapper);
        // 将查询结果转换为List<Long>
        return entities.stream()
                .map(WaylineJobBreakPointEntity::getJobId)
                .collect(Collectors.toList());
    }
}