Use cast to extract code

opentripplanner · May 31, 2024 · ab5ec25 · ab5ec25
1 parent 78cd79b
commit ab5ec25
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 11 deletions.
diff --git a/src/ext/java/org/opentripplanner/ext/geocoder/StopClusterMapper.java b/src/ext/java/org/opentripplanner/ext/geocoder/StopClusterMapper.java
@@ -9,13 +9,15 @@
 import java.util.Objects;
 import java.util.Optional;
 import java.util.stream.Collectors;
+import javax.annotation.Nullable;
 import org.opentripplanner.framework.collection.ListUtils;
 import org.opentripplanner.framework.geometry.WgsCoordinate;
 import org.opentripplanner.framework.i18n.I18NString;
 import org.opentripplanner.model.FeedInfo;
 import org.opentripplanner.transit.model.framework.FeedScopedId;
 import org.opentripplanner.transit.model.network.Route;
 import org.opentripplanner.transit.model.organization.Agency;
+import org.opentripplanner.transit.model.site.Station;
 import org.opentripplanner.transit.model.site.StopLocation;
 import org.opentripplanner.transit.model.site.StopLocationsGroup;
 import org.opentripplanner.transit.service.TransitService;
@@ -153,7 +155,7 @@ StopCluster.Location toLocation(FeedScopedId id) {
         .toList();
       return new StopCluster.Location(
         group.getId(),
-        group.getCode(),
+        extractCode(group),
         STATION,
         group.getName().toString(),
         new StopCluster.Coordinate(group.getLat(), group.getLon()),
@@ -164,6 +166,15 @@ StopCluster.Location toLocation(FeedScopedId id) {
     }
   }
 
+  @Nullable
+  private static String extractCode(StopLocationsGroup group) {
+    if (group instanceof Station station) {
+      return station.getCode();
+    } else {
+      return null;
+    }
+  }
+
   private static StopCluster.Coordinate toCoordinate(WgsCoordinate c) {
     return new StopCluster.Coordinate(c.latitude(), c.longitude());
   }

diff --git a/src/main/java/org/opentripplanner/transit/model/site/GroupOfStations.java b/src/main/java/org/opentripplanner/transit/model/site/GroupOfStations.java
@@ -51,12 +51,6 @@ public WgsCoordinate getCoordinate() {
     return coordinate;
   }
 
-  @Nullable
-  @Override
-  public String getCode() {
-    return null;
-  }
-
   @Nonnull
   public Collection<StopLocation> getChildStops() {
     return this.childStations.stream().flatMap(s -> s.getChildStops().stream()).toList();

diff --git a/src/main/java/org/opentripplanner/transit/model/site/StopLocationsGroup.java b/src/main/java/org/opentripplanner/transit/model/site/StopLocationsGroup.java
@@ -1,7 +1,6 @@
 package org.opentripplanner.transit.model.site;
 
 import java.util.Collection;
-import javax.annotation.Nullable;
 import org.opentripplanner.framework.geometry.WgsCoordinate;
 import org.opentripplanner.framework.i18n.I18NString;
 import org.opentripplanner.framework.lang.ObjectUtils;
@@ -40,7 +39,4 @@ default double getLon() {
   default String logName() {
     return ObjectUtils.ifNotNull(getName(), Object::toString, null);
   }
-
-  @Nullable
-  String getCode();
 }