Function drive_connection 

async fn drive_connection<C, R>(
    connection: &mut C,
    rounds: &R,
    shutdown: &CancellationToken,
    poll_timeout: Duration,
) -> bool
where
    C: SchedulerConnection,
    R: AssignmentRounds,