mul-agent-virtual-simulation-system/train-server/src/main/java/top/haijunit/train/service/TrainService.java

package top.haijunit.train.service;

import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import top.haijunit.train.config.properties.TrainProperties;
import top.haijunit.train.domain.entity.SignEntity;
import top.haijunit.train.mapper.SignMapper;

import java.util.ArrayList;
import java.util.List;

/**
 * @author zhanghaijun
 * @date 2023/11/11 13:40
 * @description 列车行驶业务处理
 */
@Service
@RequiredArgsConstructor
public class TrainService {

    private final TrainProperties trainProperties;
    private final SignMapper signMapper;


    public List<SignEntity> getSignAll() {
        return signMapper.selectList();
    }

    public List<SignEntity> getRouteList() {
        List<String> routeList = trainProperties.getRoutes();
        if (CollUtil.isEmpty(routeList)) {
            return new ArrayList<>();
        }
        LambdaQueryWrapper<SignEntity> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(SignEntity::getCode, routeList);
        return signMapper.selectList(wrapper);
    }

}