如何在单个线程上按顺序运行异步任务？

Question

我正在开发一个使用 rust-tokio 进行异步执行的程序。 主 function 会定期调用 function 到 append 到 CSV 文件来记录一段时间内的操作。

我想让 CSV 创建 function 异步并将其作为单独的任务运行，这样我可以继续主要的 function 如果 CSV 创建需要一些时间（比如等待另一个应用程序，如 Excel 来释放它）。

有没有一种优雅的方法来完成这个？

LocalSet几乎看起来可以完成这项工作，但任务需要按顺序执行，因此 CSV 是按时间顺序排列的。 对我来说，文档似乎并不能保证这一点。

下面是一些伪代码来说明这个想法。 本质上，我在考虑需要完成的任务队列。

#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
    let local = task::LocalSetOrdered::new(); //This is a fictitious struct

    let mut data: usize = 10; //For simplicity, just store a single number
    loop {
        // Some operations here
        data = data + 1;

        let data_clone = data.clone();
        //Add a new task to complete after all prior tasks
        local.push(async move {
            match append_to_csv(data_clone).await {
                Ok(_) => Ok(()),
                Err(_) => Err(()),
            }
        });

        sleep(Duration::from_secs(60)).await;
    }

    Ok(())
}

async fn append_to_csv(data_in: usize) -> Result<(), Box<dyn Error>> {
    loop {
        let file = match OpenOptions::new().write(true).append(true).open(filename) {
            Ok(f) => f,
            Err(_) => {
                //Error opening the file, try again
                sleep(Duration::from_secs(60)).await;
                continue;
            }
        };
        let wtr = csv::Writer::from_writer(file);

        let date_time = Utc::now();
        wtr.write_field(format!("{}", date_time.format("%Y/%m/%d %H:%M")))?;
        wtr.write_field(format!("{}", data_in))?;
        wtr.write_record(None::<&[u8]>)?; //Finish the line
    }
}

Answer 1

您可以使用工作任务写入 csv 文件，并使用通道传递要写入的数据

use tokio::sync::mpsc::{channel, Receiver};

#[derive(Debug)]
pub struct CsvData(i32, &'static str);

async fn append_to_csv(mut rx: Receiver<CsvData>) {
    let mut wtr = csv::Writer::from_writer(std::io::stdout());
    while let Some(data) = rx.recv().await {
        wtr.write_record([&data.0.to_string(), data.1]).unwrap();
        wtr.flush().unwrap();
    }
}

#[tokio::main]
async fn main() {
    let (tx, rx) = channel(10);
    tokio::spawn(async {
        append_to_csv(rx).await;
    });

    for i in 0.. {
        tx.send(CsvData(i, "Hello world")).await.unwrap();
    }
}

如果您需要写入来自多个任务的数据，则可以克隆通道发送方。