fix(ev): unqueuing a vehicle

matsim-org · Dec 7, 2024 · 4c02429 · 4c02429
1 parent bee9a70
commit 4c02429
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/contribs/ev/src/main/java/org/matsim/contrib/ev/charging/ChargingWithQueueingLogic.java b/contribs/ev/src/main/java/org/matsim/contrib/ev/charging/ChargingWithQueueingLogic.java
@@ -118,10 +118,19 @@ public void removeVehicle(ElectricVehicle ev, double now) {
 				}
 			}
 		} else {
-			// make sure ev was in the queue
-			Preconditions.checkState(queuedVehicles.remove(ev), "Vehicle (%s) is neither queued nor plugged at charger (%s)", ev.getId(),
-				charger.getId());
-			eventsManager.processEvent(new QuitQueueAtChargerEvent(now, charger.getId(), ev.getId()));
+			var queuedVehiclesIter = queuedVehicles.iterator();
+			while (queuedVehiclesIter.hasNext()) {
+				var queuedVehicle = queuedVehiclesIter.next();
+
+				if (queuedVehicle.ev() == ev) {
+					queuedVehiclesIter.remove();
+					eventsManager.processEvent(new QuitQueueAtChargerEvent(now, charger.getId(), ev.getId()));
+					return; // found the vehicle
+				}
+			}
+
+			throw new IllegalStateException(String.format("Vehicle (%s) is neither queued nor plugged at charger (%s)", ev.getId(),
+				charger.getId()));
 		}
 	}