7 个版本
0.2.4 | 2024年5月21日 |
---|---|
0.2.3 | 2024年5月6日 |
0.1.1 | 2024年5月4日 |
#719 in 异步
用于 asyncs
68KB
1.5K SLoC
Spawns
Rust 的线程上下文任务生成器,简化异步运行时无关的编码。
动机
目前,Rust 没有一个标准的异步运行时。这使我们面临一个难题:选择哪一个,并且使得创建运行时无关的库变得非常困难。我们面临的最具挑战性的事情是如何生成任务?
spawns
为 Rust std
和异步运行时提出了一种线程上下文任务生成器。一旦交付,我们就能以运行时无关的方式生成任务。结合其他运行时无关的 io、定时器、通道等 crate,我们能够轻松编写运行时无关的代码。
异步运行时的 API
/// Thin wrapper around task to accommodate possible new members.
#[non_exhaustive]
pub struct Task {
pub id: Id,
pub name: Name,
pub future: Box<dyn Future<Output = ()> + Send + 'static>,
}
/// Trait to spawn task.
pub trait Spawn {
fn spawn(&self, task: Task);
}
/// Scope where tasks are [spawn]ed through given [Spawn].
pub struct SpawnScope<'a> {}
/// Enters a scope where new tasks will be [spawn]ed through given [Spawn].
pub fn enter(spawner: &dyn Spawn) -> SpawnScope<'_>;
异步运行时必须执行两个操作来适应其他运行时无关的 API。
- 实现
Spawn
以生成异步任务。 - 在所有执行线程中调用
enter
。
客户端的 API
impl<T> JoinHandle<T> {
/// Gets id of the associated task.
pub fn id(&self) -> Id {}
/// Cancels associated task with this handle.
///
/// Cancellation is inherently concurrent with task execution. Currently, there is no guarantee
/// about promptness, the task could even run to complete normally after cancellation.
pub fn cancel(&self) { }
/// Attaches to associated task to gain cancel on [Drop] permission.
pub fn attach(self) -> TaskHandle<T> { }
}
impl<T> Future for JoinHandle<T> {
type Output = Result<T, JoinError>;
}
/// Spawns a new task.
///
/// # Panics
/// 1. Panic if no spawner.
/// 2. Panic if [Spawn::spawn] panic.
pub fn spawn<T, F>(f: F) -> JoinHandle<T>
where
F: Future<Output = T> + Send + 'static,
T: Send + 'static;
该 API 能够生成、连接和取消任务,就像 tokio
、smol
和 async-std
所做的那样。
关注点
- 装箱?是的,它需要
GlobalAlloc
。 - 即使是入口 future 也需要装箱吗?不,但是
try_id()
将返回None
。我想我们可以提供一些包装功能。 no_std
?不,它目前需要thread_local!
。一旦稳定,我们可以将其移动到#[thread_local]
。spawn_local
用于!Send
的 future?不,至少现在还不是。我只看到async-global-executor
能够自由地使用spawn_local
。我认为这是 Rust 的责任,不应该将拥有!Send
的 future 视为!Send
。这样我们就不太可能创建!Send
的 future。请参见 Async Rust needs Await and 'thread forSend
Future
了解我对这个问题的看法。对于首先捕获!Send
并存储线程局部!Send
的 future,它们需要当前线程的 executor。
包
- spawns-core 为异步运行时提供
Spawn
和enter()
,用于设置线程上下文任务 spawner。 - spawns-compat 通过功能门提供对
tokio
、smol
和async-global-executor
(由async-std
使用)的兼容性。 - spawns-executor 提供了带有当前线程 executor 和多线程 executor 的完整功能的
block_on
。 - spawns 导出所有上述包,包括功能门
tokio
、smol
和async-global-executor
。此外,它还提供功能门executor
以包含spawns-executor
。
示例
请参阅 示例。这里列出了一个最小运行时无关的 echo 服务器作为演示。
use async_net::*;
use futures_lite::io;
pub async fn echo_server(port: u16) {
let listener = TcpListener::bind(("127.0.0.1", port)).await.unwrap();
println!("Listen on port: {}", listener.local_addr().unwrap().port());
let mut echos = vec![];
let mut id_counter = 0;
loop {
let (stream, remote_addr) = listener.accept().await.unwrap();
id_counter += 1;
let id = id_counter;
let handle = spawns::spawn(async move {
eprintln!("{:010}[{}]: serving", id, remote_addr);
let (reader, writer) = io::split(stream);
match io::copy(reader, writer).await {
Ok(_) => eprintln!("{:010}[{}]: closed", id, remote_addr),
Err(err) => eprintln!("{:010}[{}]: {:?}", id, remote_addr, err),
}
})
.attach();
echos.push(handle);
}
}
要使它能够正常工作,您只需设置线程上下文任务 spawner。
许可证
依赖关系
~0–10MB
~97K SLoC