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
41
42
43
44
45
46
47
48
49
50
51
52
53

use error::Result;
use std::sync::mpsc::{self, Sender};
use std::thread;
use super::{RowData, Worker};
use super::immediate::ImmediateWorker;

enum WorkerMsg {
    Start(RowData),
    AppendRow((usize, Vec<i16>)),
    GetResult((usize, Sender<Vec<u8>>)),
}

pub struct ThreadedWorker {
    sender: Sender<WorkerMsg>,
}

impl Worker for ThreadedWorker {
    fn new() -> Result<Self> {
        let thread_builder = thread::Builder::new().name("worker thread".to_owned());
        let (tx, rx) = mpsc::channel();

        thread_builder.spawn(move || {
            let mut worker = ImmediateWorker::new_immediate();

            while let Ok(message) = rx.recv() {
                match message {
                    WorkerMsg::Start(data) => {
                        worker.start_immediate(data);
                    },
                    WorkerMsg::AppendRow(row) => {
                        worker.append_row_immediate(row);
                    },
                    WorkerMsg::GetResult((index, chan)) => {
                        let _ = chan.send(worker.get_result_immediate(index));
                    },
                }
            }
        })?;

        Ok(ThreadedWorker { sender: tx })
    }
    fn start(&mut self, row_data: RowData) -> Result<()> {
        Ok(self.sender.send(WorkerMsg::Start(row_data))?)
    }
    fn append_row(&mut self, row: (usize, Vec<i16>)) -> Result<()> {
        Ok(self.sender.send(WorkerMsg::AppendRow(row))?)
    }
    fn get_result(&mut self, index: usize) -> Result<Vec<u8>> {
        let (tx, rx) = mpsc::channel();
        self.sender.send(WorkerMsg::GetResult((index, tx)))?;
        Ok(rx.recv()?)
    }
}