rain
2024-08-21 2db1aa88e8ab53096a936163d686b90d8e056a99
src/main/java/com/dji/sample/wayline/service/impl/WaylineJobBreakPointServiceImpl.java
@@ -1,5 +1,7 @@
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;
@@ -7,6 +9,9 @@
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
@@ -26,4 +31,15 @@
        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());
    }
}