32 lines
1.1 KiB
Rust
32 lines
1.1 KiB
Rust
use chrono::offset::TimeZone;
|
|
use chrono::{Duration, Utc};
|
|
use rangemap::RangeMap;
|
|
|
|
fn main() {
|
|
let people = ["Alice", "Bob", "Carol"];
|
|
let mut roster = RangeMap::new();
|
|
|
|
// Set up initial roster.
|
|
let start_of_roster = Utc.with_ymd_and_hms(2019, 1, 7, 0, 0, 0).unwrap();
|
|
let mut week_start = start_of_roster;
|
|
for _ in 0..3 {
|
|
for person in &people {
|
|
let next_week = week_start + Duration::weeks(1);
|
|
roster.insert(week_start..next_week, person);
|
|
week_start = next_week;
|
|
}
|
|
}
|
|
|
|
// Bob is covering Alice's second shift (the fourth shift overall).
|
|
let fourth_shift_start = start_of_roster + Duration::weeks(3);
|
|
let fourth_shift_end = fourth_shift_start + Duration::weeks(1);
|
|
roster.insert(fourth_shift_start..fourth_shift_end, &"Bob");
|
|
|
|
// Print out the roster, and observe that
|
|
// the fourth and fifth shifts have been coalesced
|
|
// into one range.
|
|
for (range, person) in roster.iter() {
|
|
println!("{} ({}): {}", range.start, range.end - range.start, person);
|
|
}
|
|
}
|