taq 1.2.0

A simple Tokio task abstraction.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use crate::{error::TaqResult, job, Handle, Job, Task};
use std::sync::mpsc;
use tokio::sync::oneshot;

/// Extentions for [`Handle`].
pub trait HandleExt<A, R> {
    /// Sends a [`Job`] to the task and return a receiver for its result.
    ///
    /// # Errors
    /// If the [`TaskManager`](crate::TaskManager) has been
    /// dropped, this method will return an error.
    ///
    /// # Example
    /// ```
    /// // handle: Handle<String>
    /// let res = handle.run(job!(|s: &mut String| s.clone()))?.await?;
    /// println!("handles string: {res}");
    /// ```
    fn recv(&self, func: Job<A, R>) -> TaqResult<oneshot::Receiver<R>>;
    fn blocking_recv(&self, func: Job<A, R>) -> TaqResult<mpsc::Receiver<R>>;
}

#[doc(hidden)]
impl<A: Task + Send + 'static, R: Send + 'static> HandleExt<A, R> for Handle<A> {
    fn recv(&self, func: Job<A, R>) -> TaqResult<oneshot::Receiver<R>> {
        let (tx, rx) = oneshot::channel();

        self.run(job!(|t| let _ = tx.send(func.with(t).await)))?;

        Ok(rx)
    }

    fn blocking_recv(&self, func: Job<A, R>) -> TaqResult<mpsc::Receiver<R>> {
        let (tx, rx) = mpsc::sync_channel(1);

        self.run(job!(|t| let _ = tx.send(func.with(t).await)))?;

        Ok(rx)
    }
}