package com.whyc.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.dto.Response;
import com.whyc.mapper.StationMapper;
import com.whyc.pojo.Station;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.sql.Wrapper;
import java.util.List;

@Service
public class StationService {

    @Resource
    private StationMapper mapper;

    public Response<List<Station>> listAll() {
        QueryWrapper<Station> wrapper = Wrappers.query();
        wrapper.select("stationId","stationName")
                .groupBy("stationId")
                .orderByAsc("stationId");
        List<Station> stationList = mapper.selectList(wrapper);
        //对机房站点名称进行处理
        stationList.stream().forEach(station -> {
            String[] nameSplit = station.getStationName().split("-");
            String newStationName = new StringBuilder()
                    .append(nameSplit[0]).append("-")
                    .append(nameSplit[1]).append("-")
                    .append(nameSplit[2]).append("-")
                    .append(nameSplit[3]).toString();

            station.setStationName(newStationName);
        });
        return new Response<List<Station>>().set(1,stationList);
    }
}