Fix copy-on-write in TimetableSnapshot

opentripplanner · Jul 1, 2024 · 42cc0fa · 42cc0fa
1 parent 15cf5f8
commit 42cc0fa
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/main/java/org/opentripplanner/model/TimetableSnapshot.java b/src/main/java/org/opentripplanner/model/TimetableSnapshot.java
@@ -330,10 +330,15 @@ public boolean revertTripToScheduledTripPattern(FeedScopedId tripId, LocalDate s
         }
 
         if (tripTimesToRemove != null) {
-          for (Timetable sortedTimetable : sortedTimetables) {
-            boolean isDirty = sortedTimetable.getTripTimes().remove(tripTimesToRemove);
+          for (Timetable originalTimetable : sortedTimetables) {
+            boolean isDirty = originalTimetable.getTripTimes().contains(tripTimesToRemove);
             if (isDirty) {
-              dirtyTimetables.add(sortedTimetable);
+              Timetable updatedTimetable = new Timetable(originalTimetable, serviceDate);
+              updatedTimetable.getTripTimes().remove(tripTimesToRemove);
+              sortedTimetables.remove(originalTimetable);
+              sortedTimetables.add(updatedTimetable);
+              dirtyTimetables.add(updatedTimetable);
+              dirty = true;
             }
           }
         }