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

Solved: Big world test

Browse source
This commit is contained in:
Bastian Gruber 2023-05-22 12:50:04 +00:00
parent 52f92969b5
commit 3178c07f84
2 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
problem_06/src/server.rs
View file

@ -198,6 +198,7 @@ impl Handler {
ClientFrames::IAmCamera { road, mile, limit } => { ClientFrames::IAmCamera { road, mile, limit } => {
info!("Receive new camera: {road} at {mile} with limit {limit}"); info!("Receive new camera: {road} at {mile} with limit {limit}");
if self.connection_type.is_some() { if self.connection_type.is_some() {
let _ = send_message.send(ServerFrames::Error { msg: "Already connected as camera".to_string() }).await;
return Err("Already connected".into()); return Err("Already connected".into());
} }
self.set_connection_type(ConnectionType::Camera); self.set_connection_type(ConnectionType::Camera);
@ -213,6 +214,7 @@ impl Handler {
} }
ClientFrames::IAmDispatcher { roads } => { ClientFrames::IAmDispatcher { roads } => {
if self.connection_type.is_some() { if self.connection_type.is_some() {
let _ = send_message.send(ServerFrames::Error { msg: "Already connected as dispatcher".to_string() }).await;
return Err("Already connected".into()); return Err("Already connected".into());
} }

40
problem_06/src/ticketing.rs
View file

@ -51,25 +51,33 @@ pub(crate) async fn issue_possible_ticket(db: Arc<Mutex<Db>>, plate: Plate, came
let day_start = timestamp1 / 86400; let day_start = timestamp1 / 86400;
let day_end = timestamp2 / 86400; let day_end = timestamp2 / 86400;
let spans_multiple_days = day_start != day_end;
if spans_multiple_days && (db.is_plate_ticketed_for_day(day_start, plate_name.clone()) || db.is_plate_ticketed_for_day(day_end, plate_name.clone())) {
continue;
}
if db.is_plate_ticketed_for_day(day_start, plate_name.clone()) {
continue;
}
for day in day_start..=day_end { for day in day_start..=day_end {
info!("Day {day} for {ticket:?}"); info!("Ticket for day {day} for {ticket:?}");
if db.is_plate_ticketed_for_day(day, plate_name.clone()) {
info!("Ticket already issued: {ticket:?}");
continue;
}
db.ticket_plate(day, plate_name.clone()); db.ticket_plate(day, plate_name.clone());
let dispatcher = db.get_dispatcher_for_road(road.clone());
if dispatcher.is_none() {
info!("No dispatcher yet for this road: {ticket:?}");
db.add_open_ticket(ticket.clone());
continue;
}
info!("Sending ticket: {ticket:?}");
let _ = dispatcher.unwrap().send(ticket.clone().into()).await;
} }
let dispatcher = db.get_dispatcher_for_road(road.clone());
if dispatcher.is_none() {
info!("No dispatcher yet for this road: {ticket:?}");
db.add_open_ticket(ticket.clone());
continue;
}
info!("Sending ticket: {ticket:?}");
let _ = dispatcher.unwrap().send(ticket.clone().into()).await;
} }
} }