Модуль coop

Утилиты для улучшенного кооперативного планирования.

Кооперативное планирование

Один вызов poll для задачи верхнего уровня потенциально может выполнять много работы, прежде чем вернуть Poll::Pending. Если задача выполняется длительное время, не уступая управление исполнителю, она может лишить другие задачи, ожидающие выполнения этим исполнителем, или базовые ресурсы возможности работать. Поскольку в Rust нет среды выполнения, принудительно прервать длительно выполняющуюся задачу сложно. Вместо этого этот модуль предоставляет механизм, позволяющий futures по желанию взаимодействовать с исполнителем, чтобы избежать голодания.

Рассмотрим такую future:

async fn drop_all<I: Stream + Unpin>(mut input: I) {
    while let Some(_) = input.next().await {}
}

Она может показаться безобидной, но подумайте, что произойдет при высокой нагрузке, если входной поток всегда готов. Если мы запустим drop_all, задача никогда не уступит управление и лишит другие задачи и ресурсы того же исполнителя возможности работать.

Чтобы учитывать это, Tokio добавляет явные точки уступки в ряд библиотечных функций, заставляя задачи периодически возвращать управление исполнителю.

unconstrained

При необходимости task::unconstrained позволяет отключить кооперативное планирование Tokio для future. Если future обернута в unconstrained, ее никогда не будут принудительно заставлять уступать управление Tokio. Например:

use tokio::{task, sync::mpsc};

let fut = async {
    let (tx, mut rx) = mpsc::unbounded_channel();

    for i in 0..1000 {
        let _ = tx.send(());
        // This will always be ready. If coop was in effect, this code would be forced to yield
        // periodically. However, if left unconstrained, then this code will never yield.
        rx.recv().await;
    }
};

task::coop::unconstrained(fut).await;