update...

潘钊
1 parent e75e4ca5
Showing 2 changed files with 19 additions and 0 deletions
src/main/java/com/bsth/server_rs/base_info/section/LD_SectionRestService.java
src/main/java/com/bsth/server_rs/base_info/section/buffer/LD_SectionBufferData.java
@@ -30,4 +30,11 @@ public class LD_SectionRestService {
     public Map<String, Collection<LD_SectionRoute>> findByCompany(@PathParam("company") String company){
         return sectionBufferData.findRouteByCompany(company);
     }
+
+
+    @GET
+    @Path("/line/{lineCode}")
+    public Map<String, Collection<LD_SectionRoute>> findByLineCode(@PathParam("lineCode") String lineCode){
+        return sectionBufferData.findByLineCode(lineCode);
+    }
 }
@@ -123,4 +123,16 @@ public class LD_SectionBufferData implements CommandLineRunner {
     public List<RoadSpeed> roadSpeedList(String company) {
         return roadSpeedList;
     }
+
+    public Map<String, Collection<LD_SectionRoute>> findByLineCode(String lineCode) {
+
+        ArrayListMultimap<String, LD_SectionRoute> listMap = ArrayListMultimap.create();
+
+        String k1 = lineCode + "_0";
+        String k2 = lineCode + "_1";
+
+        listMap.putAll(k1, routeListMap.get(k1));
+        listMap.putAll(k2, routeListMap.get(k2));
+        return listMap.asMap();
+    }
 }