rabble 0.4.1

A library for creating location transparent actor based systems
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use std::thread::{self, JoinHandle};
use amy::Sender;

use rabble::{
    Pid,
    Node,
    Envelope,
    CorrelationId,
    Msg,
    TcpServerHandler,
    Service,
    ConnectionMsg,
    ConnectionHandler
};

use rabble::serialize::MsgpackSerializer;

use super::messages::{RabbleUserMsg, ApiClientMsg};

#[allow(dead_code)] // Not used in all tests
const API_SERVER_IP: &'static str  = "127.0.0.1:22001";

#[allow(dead_code)] // Not used in all tests
pub fn start(node: Node<RabbleUserMsg>)
    -> (Pid, Sender<Envelope<RabbleUserMsg>>, JoinHandle<()>)
{
    let server_pid = Pid {
        name: "api-server".to_string(),
        group: None,
        node: node.id.clone()
    };

    // Start the API tcp server
    let handler: TcpServerHandler<ApiServerConnectionHandler, MsgpackSerializer<ApiClientMsg>> =
        TcpServerHandler::new(server_pid.clone(), API_SERVER_IP, 5000, None);
    let mut service = Service::new(server_pid, node, handler).unwrap();
    let service_tx = service.tx.try_clone().unwrap();
    let service_pid = service.pid.clone();
    let h = thread::spawn(move || {
        service.wait();
    });
    (service_pid, service_tx, h)
}


pub struct ApiServerConnectionHandler {
    pid: Pid,
    id: u64,
    total_requests: u64
}

impl ConnectionHandler for ApiServerConnectionHandler {
    type Msg = RabbleUserMsg;
    type ClientMsg = ApiClientMsg;

    fn new(pid: Pid, id: u64) -> ApiServerConnectionHandler {
        ApiServerConnectionHandler {
            pid: pid,
            id: id,
            total_requests: 0
        }
    }

    fn handle_envelope(&mut self,
                       envelope: Envelope<RabbleUserMsg>,
                       output: &mut Vec<ConnectionMsg<ApiServerConnectionHandler>>)
    {
        let Envelope {msg, correlation_id, ..} = envelope;
        let correlation_id = correlation_id.unwrap();
        match msg {
            Msg::User(RabbleUserMsg::History(h)) => {
                output.push(ConnectionMsg::Client(ApiClientMsg::History(h), correlation_id));
            },
            Msg::User(RabbleUserMsg::OpComplete) => {
                output.push(ConnectionMsg::Client(ApiClientMsg::OpComplete, correlation_id));
            },
            Msg::Timeout => {
                output.push(ConnectionMsg::Client(ApiClientMsg::Timeout, correlation_id));
            },
            _ => unreachable!()
        }
    }

    fn handle_network_msg(&mut self,
                          msg: ApiClientMsg,
                          output: &mut Vec<ConnectionMsg<ApiServerConnectionHandler>>)
    {
        match msg {
            ApiClientMsg::Op(pid, val) => {
                output.push(self.new_envelope(pid, RabbleUserMsg::Op(val)));
            },
            ApiClientMsg::GetHistory(pid) => {
                output.push(self.new_envelope(pid, RabbleUserMsg::GetHistory));
            }

            // We only handle client requests. Client replies come in as Envelopes and are handled
            // in handle_envelope().
            _ => unreachable!()
        }
    }
}

impl ApiServerConnectionHandler {
    pub fn new_envelope(&mut self, to: Pid, user_msg: RabbleUserMsg)
        -> ConnectionMsg<ApiServerConnectionHandler>
    {
        let msg = Msg::User(user_msg);
        let correlation_id = CorrelationId::request(self.pid.clone(), self.id, self.total_requests);
        self.total_requests += 1;
        ConnectionMsg::Envelope(Envelope::new(to, self.pid.clone(), msg, Some(correlation_id)))
    }
}