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

use crate::error::Result;
use crate::events::event::Event;
use crate::events::event_handler::EventHandler;
use crate::ipc::context::Context;
use crate::ipc::stream_emitter::StreamEmitter;
use tokio::net::TcpStream;

/// The IPC Client to connect to an IPC Server.
/// Use the [IPCBuilder](crate::builder::IPCBuilder) to create the client.
/// Usually one does not need to use the IPCClient object directly.
pub struct IPCClient {
    pub(crate) handler: EventHandler,
}

impl IPCClient {
    /// Connects to a given address and returns an emitter for events to that address.
    /// Invoked by [IPCBuilder::build_client](crate::builder::IPCBuilder::build_client)
    pub async fn connect(self, address: &str) -> Result<StreamEmitter> {
        let stream = TcpStream::connect(address).await?;
        let (mut read_half, write_half) = stream.into_split();
        let emitter = StreamEmitter::new(write_half);
        let ctx = Context::new(StreamEmitter::clone(&emitter));
        let handler = self.handler;

        tokio::spawn(async move {
            while let Ok(event) = Event::from_async_read(&mut read_half).await {
                if let Err(e) = handler.handle_event(&ctx, event).await {
                    log::error!("Failed to handle event: {:?}", e);
                }
            }
        });

        Ok(emitter)
    }
}