1.加入导航点位信息接口(常规、绕改道)

王通
1 parent c8fe5e60
Showing 3 changed files with 312 additions and 33 deletions
src/main/java/com/bsth/entity/LineRegion.java
src/main/java/com/bsth/server_rs/base_info/navigation/NavigationRestService.java
src/main/java/com/bsth/server_rs/base_info/station/entity/StationRotue.java
+package com.bsth.entity;
+
+import com.bsth.server_rs.base_info.station.entity.StationRotue;
+
+import javax.persistence.*;
+import java.util.ArrayList;
+import java.util.List;
+
+public class LineRegion {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Integer id;
+
+    private Integer line;
+
+    private Integer version;
+
+    private Integer direction;
+
+    private String regionAlias;
+
+    private Integer seq;
+
+    private List<Integer> stationRouteIds = new ArrayList<>();
+
+    public Integer getId() {
+        return id;
+    }
+
+    public void setId(Integer id) {
+        this.id = id;
+    }
+
+    public Integer getLine() {
+        return line;
+    }
+
+    public void setLine(Integer line) {
+        this.line = line;
+    }
+
+    public Integer getVersion() {
+        return version;
+    }
+
+    public void setVersion(Integer version) {
+        this.version = version;
+    }
+
+    public Integer getDirection() {
+        return direction;
+    }
+
+    public void setDirection(Integer direction) {
+        this.direction = direction;
+    }
+
+    public String getRegionAlias() {
+        return regionAlias;
+    }
+
+    public void setRegionAlias(String regionAlias) {
+        this.regionAlias = regionAlias;
+    }
+
+    public Integer getSeq() {
+        return seq;
+    }
+
+    public void setSeq(Integer seq) {
+        this.seq = seq;
+    }
+
+    public List<Integer> getStationRouteIds() {
+        return stationRouteIds;
+    }
+
+    public void setStationRouteIds(List<Integer> stationRouteIds) {
+        this.stationRouteIds = stationRouteIds;
+    }
+}
 package com.bsth.server_rs.base_info.navigation;
  
 import com.bsth.entity.Detour;
+import com.bsth.entity.LineRegion;
 import com.bsth.entity.Point;
 import com.bsth.server_rs.base_info.section.buffer.LD_SectionBufferData;
 import com.bsth.server_rs.base_info.section.entity.LD_Section;
@@ -37,7 +38,9 @@ public class NavigationRestService implements InitializingBean {
     private LD_SectionBufferData ldSectionBufferData;
  
     private static Map<String, Integer> line2version = new ConcurrentHashMap<>();
-
+    
+    private static Map<String, LineRegion> line2regionMap = new ConcurrentHashMap<>();
+    
     @GET
     @Path("/version/reload")
     @Scheduled(cron = "0 0 * * * *") // 每小时执行一次
@@ -50,6 +53,103 @@ public class NavigationRestService implements InitializingBean {
         if (!line2versionMap.isEmpty()) {
             line2version = line2versionMap;
         }
+        
+        loadLineRegions();
+    }
+    
+    /**
+     * 加载线路区间信息
+     */
+    private void loadLineRegions() {
+        Map<String, LineRegion> regionMap = new ConcurrentHashMap<>();
+        Map<Integer, List<Integer>> region2stationRouteIdsMap = new ConcurrentHashMap<>();
+        
+        List<LineRegion> regionList = jdbcTemplate.query("SELECT t1.* FROM `bsth_c_line_region` t1 join bsth_c_line_versions t2 on t1.line = t2.line and t1.version = t2.versions where NOW() BETWEEN start_date and end_date", BeanPropertyRowMapper.newInstance(LineRegion.class));
+        List<Map<String, Object>> mapList = jdbcTemplate.queryForList("SELECT region_id, station_route_id from bsth_c_line_region_stationroute");
+
+        Integer oldRegionId = null;
+        for (Map<String, Object> map : mapList) {
+            Integer regionId = (Integer) map.get("region_id"), stationRouteId = (Integer) map.get("station_route_id");
+            if (oldRegionId == null || !oldRegionId.equals(regionId)) {
+                oldRegionId = (Integer) map.get("region_id");
+                List<Integer> stationRouteIds = new ArrayList<>();
+                stationRouteIds.add(stationRouteId);
+                region2stationRouteIdsMap.put(oldRegionId, stationRouteIds);
+            } else {
+                List<Integer> stationRouteIds = region2stationRouteIdsMap.get(regionId);
+                stationRouteIds.add(stationRouteId);
+            }
+        }
+        for (LineRegion region : regionList) {
+            String regionKey = region.getLine() + "-" + region.getSeq();
+            List<Integer> stationRouteIds = region2stationRouteIdsMap.get(region.getId());
+            region.setStationRouteIds(stationRouteIds == null ? new ArrayList<Integer>() : stationRouteIds);
+            regionMap.put(regionKey, region);
+        }
+        
+        if (!regionMap.isEmpty()) {
+            line2regionMap = regionMap;
+        }
+    }
+    
+    /**
+     * 解析线路编码，区分普通线路和区间线路
+     * @param lineCode 线路编码
+     * @return 解析结果，包含isRegion(是否区间线路)、lineCode(线路代码)、seq(区间序号)
+     */
+    private Map<String, Object> parseLineCode(String lineCode) {
+        Map<String, Object> result = new HashMap<>();
+        
+        if (lineCode.contains("-")) {
+            String[] parts = lineCode.split("-");
+            result.put("isRegion", true);
+            result.put("lineCode", parts[0]);
+            result.put("seq", Integer.parseInt(parts[1]));
+        } else {
+            result.put("isRegion", false);
+            result.put("lineCode", lineCode);
+            result.put("seq", null);
+        }
+        
+        return result;
+    }
+    
+    /**
+     * 获取指定的 LineRegion
+     * @param lineCode 线路编码
+     * @param seq 区间序号
+     * @return LineRegion
+     */
+    private LineRegion getLineRegion(String lineCode, Integer seq) {
+        String regionKey = lineCode + "-" + seq;
+        return line2regionMap.get(regionKey);
+    }
+    
+    /**
+     * 根据区间过滤站点路由
+     * @param stationRotueList 站点路由列表
+     * @param lineRegion 线路区间
+     * @return 过滤后的站点路由列表
+     */
+    private List<StationRotue> filterStationRoutesByRegion(List<StationRotue> stationRotueList, LineRegion lineRegion) {
+        if (stationRotueList == null || stationRotueList.isEmpty() || lineRegion == null) {
+            return new ArrayList<>();
+        }
+
+        List<Integer> stationRouteIds = lineRegion.getStationRouteIds();
+        if (stationRouteIds.isEmpty()) {
+            return stationRotueList;
+        }
+        
+        // 过滤站点
+        List<StationRotue> filteredList = new ArrayList<>();
+        for (StationRotue stationRotue : stationRotueList) {
+            if (stationRotue.getId() != 0 && stationRouteIds.contains(stationRotue.getId())) {
+                filteredList.add(stationRotue);
+            }
+        }
+        
+        return filteredList;
     }
  
     /**
@@ -65,30 +165,63 @@ public class NavigationRestService implements InitializingBean {
         result.put("lineCode", lineCode);
         result.put("direction", direction);
  
+        // 解析线路编码，判断是否为区间线路
+        Map<String, Object> parseResult = parseLineCode(lineCode);
+        boolean isRegion = (boolean) parseResult.get("isRegion");
+        String actualLineCode = (String) parseResult.get("lineCode");
+        
+        // 如果是区间线路，检查版本是否一致
+        if (isRegion) {
+            LineRegion region = line2regionMap.get(lineCode);
+
+            if (region == null) {
+                result.put("list", new ArrayList<Point>());
+                return result;
+            }
+            direction = region.getDirection();
+            result.put("direction", direction);
+
+            // 获取线路当前版本
+            Integer currentVersion = line2version.get(actualLineCode);
+
+            // 判断版本是否一致
+            if (!currentVersion.equals(region.getVersion())) {
+                result.put("list", new ArrayList<Point>());
+                return result;
+            }
+        }
+        
         // 获取路段数据
-        Map<String, Collection<LD_SectionRoute>> sectionRoutesMap = ldSectionBufferData.findByLineCode(lineCode);
+        Map<String, Collection<LD_SectionRoute>> sectionRoutesMap = ldSectionBufferData.findByLineCode(actualLineCode);
         if (sectionRoutesMap.isEmpty()) {
             result.put("list", new ArrayList<Point>());
             return result;
         }
  
         // 获取对应方向的路段路由
-        Collection<LD_SectionRoute> sectionRoutes = sectionRoutesMap.get(lineCode + "_" + direction);
+        Collection<LD_SectionRoute> sectionRoutes = sectionRoutesMap.get(actualLineCode + "_" + direction);
         if (sectionRoutes == null || sectionRoutes.isEmpty()) {
             result.put("list", new ArrayList<Point>());
             return result;
         }
  
         // 获取站点数据
-        Map<String, Collection<StationRotue>> stationRoutesMap = StationBufferData.findRouteByLineCode(lineCode);
-        Collection<StationRotue> stationRoutes = stationRoutesMap.get(lineCode + "_" + direction);
+        Map<String, Collection<StationRotue>> stationRoutesMap = StationBufferData.findRouteByLineCode(actualLineCode);
+        Collection<StationRotue> stationRoutes = stationRoutesMap.get(actualLineCode + "_" + direction);
         List<StationRotue> stationRotueList = stationRoutes != null ? new ArrayList<>(stationRoutes) : new ArrayList<StationRotue>();
  
+        // 如果是区间线路，需要过滤站点
+        if (isRegion) {
+            Integer seq = (Integer) parseResult.get("seq");
+            LineRegion lineRegion = getLineRegion(actualLineCode, seq);
+            stationRotueList = filterStationRoutesByRegion(stationRotueList, lineRegion);
+        }
+        
         // 创建点位信息列表
-        List<Point> points = new ArrayList<>();
+        List<Point> points = new ArrayList<>(), filteredPoints = new ArrayList<>();
  
-        // 建立路段到站点的映射
-        Map<String, StationRotue> sectionToStationMap = new HashMap<>();
+        // 建立路段到站点的映射（一条路段可能有多个站点）
+        Map<String, List<StationRotue>> sectionToStationMap = new HashMap<>();
  
         // 遍历站点，为每个站点找最近的路段
         for (StationRotue stationRotue : stationRotueList) {
@@ -105,7 +238,11 @@ public class NavigationRestService implements InitializingBean {
             LD_Section nearestSection = findNearestSection(stationRotue, sectionRoutes);
             if (nearestSection != null) {
                 // 建立路段到站点的映射
-                sectionToStationMap.put(nearestSection.getSectionCode(), stationRotue);
+                String sectionCode = nearestSection.getSectionCode();
+                if (!sectionToStationMap.containsKey(sectionCode)) {
+                    sectionToStationMap.put(sectionCode, new ArrayList<StationRotue>());
+                }
+                sectionToStationMap.get(sectionCode).add(stationRotue);
             }
         }
  
@@ -117,16 +254,30 @@ public class NavigationRestService implements InitializingBean {
             }
  
             // 检查路段是否有对应的站点
-            StationRotue stationRotue = sectionToStationMap.get(section.getSectionCode());
+            List<StationRotue> stationRotueListInSection = sectionToStationMap.get(section.getSectionCode());
  
             // 创建点位信息
-            Point point = createPointFromSection(section, stationRotue);
-            if (point != null) {
-                points.add(point);
+            List<Point> sectionPoints = createPointsFromSection(section, stationRotueListInSection);
+            if (sectionPoints != null && !sectionPoints.isEmpty()) {
+                points.addAll(sectionPoints);
+            }
+        }
+        int i = 0;
+        for (;i < points.size();i++) {
+            if (points.get(i).isStation()) {
+                break;
+            }
+        }
+        boolean isStart = false;
+        for (int j = points.size() - 1;j >= i;j--) {
+            Point point = points.get(j);
+            if (isStart || point.isStation()) {
+                isStart = true;
+                filteredPoints.add(point);
             }
         }
  
-        result.put("list", points);
+        result.put("list", filteredPoints);
         return result;
     }
  
@@ -311,50 +462,86 @@ public class NavigationRestService implements InitializingBean {
     }
  
     /**
-     * 根据路段和站点创建点位信息
+     * 根据路段和站点列表创建点位信息
      * @param section 路段
-     * @param station 站点
-     * @return 点位信息
+     * @param stationList 站点列表
+     * @return 点位信息列表
      */
-    private Point createPointFromSection(LD_Section section, StationRotue station) {
-        Point point = new Point();
-        
-        if (station != null && station.getStation() != null) {
-            // 如果有站点，使用站点的坐标
-            point.setLon(station.getLon());
-            point.setLat(station.getLat());
-            point.setStation(true);
+    private List<Point> createPointsFromSection(LD_Section section, List<StationRotue> stationList) {
+        List<Point> points = new ArrayList<>();
+        
+        if (stationList != null && !stationList.isEmpty()) {
+            // 如果有站点，为每个站点创建一个点位
+            for (StationRotue station : stationList) {
+                if (station != null && station.getStation() != null) {
+                    Point point = new Point();
+                    point.setLon(station.getLon());
+                    point.setLat(station.getLat());
+                    point.setStation(true);
+                    points.add(point);
+                }
+            }
         } else {
-            // 否则使用路段的中间点坐标
+            // 如果没有站点，使用路段的中间点坐标
             Point midPoint = getSectionMidPoint(section);
             if (midPoint != null) {
+                Point point = new Point();
                 point.setLon(midPoint.getLon());
                 point.setLat(midPoint.getLat());
                 point.setStation(false);
-            } else {
-                return null;
+                points.add(point);
             }
         }
  
-        return point;
+        return points;
     }
  
     @GET
     @Path("/detour/{lineCode}/{direction}")
     public List<Detour> detour(@PathParam("lineCode") String lineCode, @PathParam("direction") int direction) {
-        Integer version = line2version.get(lineCode);
+        // 解析线路编码，判断是否为区间线路
+        Map<String, Object> parseResult = parseLineCode(lineCode);
+        boolean isRegion = (boolean) parseResult.get("isRegion");
+        String actualLineCode = (String) parseResult.get("lineCode");
+
+        Integer version = line2version.get(actualLineCode);
         if (version == null) {
             version = 1;
         }
+        // 如果是区间线路，检查版本是否一致
+        if (isRegion) {
+            LineRegion region = line2regionMap.get(lineCode);
+
+            if (region == null) {
+                return new ArrayList<>();
+            }
+            direction = region.getDirection();
+
+            // 获取线路当前版本
+            Integer currentVersion = line2version.get(actualLineCode);
+
+            // 判断版本是否一致
+            if (!currentVersion.equals(region.getVersion())) {
+                return new ArrayList<>();
+            }
+        }
         List<Detour> detourList = jdbcTemplate.query("select * from bsth_c_detour where line = ? and versions = ? and direction = ?", new Object[]{ lineCode, version, direction }, BeanPropertyRowMapper.newInstance(Detour.class));
-        Map<String, Collection<StationRotue>> map = StationBufferData.findRouteByLineCode(lineCode);
+        Map<String, Collection<StationRotue>> map = StationBufferData.findRouteByLineCode(actualLineCode);
         if (map.isEmpty()) {
             return new ArrayList<>();
         }
-        Collection<StationRotue> stationRotueCollection = map.get(String.format("%s_%d", lineCode, direction));
+        Collection<StationRotue> stationRotueCollection = map.get(String.format("%s_%d", actualLineCode, direction));
+        List<StationRotue> stationRotueList = new ArrayList<>(stationRotueCollection);
+
+        // 如果是区间线路，需要过滤站点
+        if (isRegion) {
+            Integer seq = (Integer) parseResult.get("seq");
+            LineRegion lineRegion = getLineRegion(actualLineCode, seq);
+            stationRotueList = filterStationRoutesByRegion(stationRotueList, lineRegion);
+        }
  
         // 处理 detourList，将 startStationCode 和 terminalStationCode 替换为对应的索引值+1
-        processDetourStations(detourList, stationRotueCollection);
+        processDetourStations(detourList, stationRotueList);
  
         // 过滤掉 terminalIdx 不大于 startIdx 的对象
         List<Detour> filteredDetourList = new ArrayList<>();
@@ -5,6 +5,8 @@ package com.bsth.server_rs.base_info.station.entity;
  */
 public class StationRotue {
  
+    private int id;
+
     /** 线路编码 */
     private String lineCode;
  
@@ -63,6 +65,14 @@ public class StationRotue {
  
     private Double lat;
  
+    public int getId() {
+        return id;
+    }
+
+    public void setId(int id) {
+        this.id = id;
+    }
+
     public String getLineCode() {
         return lineCode;
     }