gruberb/protohackers
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Remove open tickets when sent out

Browse source
This commit is contained in:
Bastian Gruber 2023-05-22 07:20:10 +02:00
parent f7c2292833
commit 81686a07ad
No known key found for this signature in database
GPG key ID: BE9F8C772B188CBF
2 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
problem_06/src/db.rs
View file

@ -163,6 +163,15 @@ impl Db {
state.open_tickets.values().flatten().cloned().collect() state.open_tickets.values().flatten().cloned().collect()
} }
pub(crate) fn remove_open_ticket(&self, road: Road, ticket: Ticket) -> bool {
let mut state = self.state.lock().unwrap();
if let Some(tickets) = state.open_tickets.get_mut(&road) {
tickets.retain(|t| t.plate != ticket.plate);
return true;
}
false
}
pub(crate) fn get_plates_by_road( pub(crate) fn get_plates_by_road(
&self, &self,
plate: Plate, plate: Plate,

3
problem_06/src/ticketing.rs
View file

@ -88,7 +88,8 @@ pub(crate) async fn send_out_waiting_tickets(db: Db) {
debug!(?tickets, "Sending out waiting tickets"); debug!(?tickets, "Sending out waiting tickets");
for ticket in tickets { for ticket in tickets {
if let Some(dispatcher) = db.get_dispatcher_for_road(Road(ticket.road)) { if let Some(dispatcher) = db.get_dispatcher_for_road(Road(ticket.road)) {
let _ = dispatcher.send(ticket.into()).await; let _ = dispatcher.send(ticket.clone().into()).await;
db.remove_open_ticket(Road(ticket.road), ticket);
} }
} }
} }