/*
* meli - async module
*
* Copyright 2017 Manos Pitsidianakis
*
* This file is part of meli.
*
* meli is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* meli is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with meli. If not, see .
*/
/*!
* Primitive Async/Wait implementation.
*
* To create an Async promise, create an AsyncBuilder. Ask for its channel receiver/sender with
* `tx` and `rx` methods to pass them in your worker's closure. Build an `Async` with your
* `JoinHandle`. The thread must communicate with the `Async` object via `AsyncStatus`
* messages.
*
* When `Async` receives `AsyncStatus::Finished` it joins the thread and takes its value which
* can be extracted with `extract`.
*/
use chan;
use std::fmt;
use std::sync::Arc;
#[derive(Clone)]
pub struct Work(pub Arc ()>>);
impl Work {
pub fn compute(&self) {
(self.0)();
}
}
impl fmt::Debug for Work {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "Work object")
}
}
unsafe impl Send for Work {}
unsafe impl Sync for Work {}
/// Messages to pass between `Async` owner and its worker thread.
#[derive(Clone)]
pub enum AsyncStatus {
NoUpdate,
Payload(T),
Finished,
///The number may hold whatever meaning the user chooses.
ProgressReport(usize),
}
impl fmt::Debug for AsyncStatus {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
AsyncStatus::NoUpdate => write!(f, "AsyncStatus::NoUpdate"),
AsyncStatus::Payload(_) => write!(f, "AsyncStatus::Payload(_)"),
AsyncStatus::Finished => write!(f, "AsyncStatus::Finished"),
AsyncStatus::ProgressReport(u) => write!(f, "AsyncStatus::ProgressReport({})", u),
}
}
}
/// A builder object for `Async`
#[derive(Debug, Clone)]
pub struct AsyncBuilder {
tx: chan::Sender>,
rx: chan::Receiver>,
}
#[derive(Debug, Clone)]
pub struct Async {
value: Option,
work: Work,
active: bool,
tx: chan::Sender>,
rx: chan::Receiver>,
}
impl Default for AsyncBuilder {
fn default() -> Self {
AsyncBuilder::::new()
}
}
impl AsyncBuilder {
pub fn new() -> Self {
let (sender, receiver) = chan::sync(8 * ::std::mem::size_of::>());
AsyncBuilder {
tx: sender,
rx: receiver,
}
}
/// Returns the sender object of the promise's channel.
pub fn tx(&mut self) -> chan::Sender> {
self.tx.clone()
}
/// Returns the receiver object of the promise's channel.
pub fn rx(&mut self) -> chan::Receiver> {
self.rx.clone()
}
/// Returns an `Async` object that contains a `Thread` join handle that returns a `T`
pub fn build(self, work: Box ()>) -> Async {
Async {
work: Work(Arc::new(work)),
value: None,
tx: self.tx,
rx: self.rx,
active: false,
}
}
}
impl Async {
/// Consumes `self` and returns the computed value. Will panic if computation hasn't finished.
pub fn extract(self) -> T {
self.value.unwrap()
}
pub fn work(&mut self) -> Option {
if !self.active {
self.active = true;
Some(self.work.clone())
} else {
None
}
}
/// Returns the sender object of the promise's channel.
pub fn tx(&mut self) -> chan::Sender> {
self.tx.clone()
}
/// Polls worker thread and returns result.
pub fn poll(&mut self) -> Result, ()> {
if self.value.is_some() {
return Ok(AsyncStatus::Finished);
}
//self.tx.send(true);
let rx = &self.rx;
let result: T;
chan_select! {
default => {
return Ok(AsyncStatus::NoUpdate);
},
rx.recv() -> r => {
match r {
Some(AsyncStatus::Payload(payload)) => {
result = payload;
},
Some(a) => {
return Ok(a);
}
_ => {
return Err(());
},
}
},
};
self.value = Some(result);
Ok(AsyncStatus::Finished)
}
/// Blocks until thread joins.
pub fn join(&mut self) {
let result: T;
let rx = &self.rx;
loop {
chan_select! {
rx.recv() -> r => {
match r {
Some(AsyncStatus::Payload(payload)) => {
result = payload;
break;
},
_ => continue,
}
}
}
}
self.value = Some(result);
}
}