rust/lib-hedgewars-engine/src/ipc.rs
author unC0Rr
Thu, 22 Nov 2018 13:39:51 +0100
changeset 14293 3152d9fdb499
parent 14282 1b8c889027a3
child 14315 21be7838a127
permissions -rw-r--r--
- Move EngineInstance into a separate module - Refactor ipc a bit and process messages
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
     1
use hedgewars_engine_messages::{messages::*, parser::extract_message};
14276
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     2
use netbuf::*;
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
     3
use std::io::*;
14276
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     4
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     5
pub struct IPC {
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     6
    in_buffer: Buf,
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     7
    out_buffer: Buf,
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     8
}
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
     9
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
    10
impl IPC {
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
    11
    pub fn new() -> Self {
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    12
        Self {
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    13
            in_buffer: Buf::new(),
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    14
            out_buffer: Buf::new(),
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    15
        }
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    16
    }
14276
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
    17
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    18
    pub fn send_message(&mut self, message: &EngineMessage) {
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    19
        self.out_buffer.write(&message.to_bytes()).unwrap();
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    20
    }
14293
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    21
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    22
    pub fn iter(& mut self) -> IPCMessagesIterator {
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    23
        IPCMessagesIterator::new(self)
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    24
    }
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    25
}
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    26
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    27
impl Write for IPC {
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    28
    fn write(&mut self, buf: &[u8]) -> Result<usize> {
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    29
        self.in_buffer.write(buf)
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    30
    }
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    31
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    32
    fn flush(&mut self) -> Result<()> {
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    33
        self.in_buffer.flush()
14276
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
    34
    }
a239e4243cf9 Start outlining ipc subsystem
unC0Rr
parents:
diff changeset
    35
}
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    36
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    37
impl Read for IPC {
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    38
    fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
14282
1b8c889027a3 Use question mark approach also in IPC::read
unC0Rr
parents: 14281
diff changeset
    39
        let read_bytes = self.out_buffer.as_ref().read(buf)?;
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    40
14282
1b8c889027a3 Use question mark approach also in IPC::read
unC0Rr
parents: 14281
diff changeset
    41
        self.out_buffer.consume(read_bytes);
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    42
14282
1b8c889027a3 Use question mark approach also in IPC::read
unC0Rr
parents: 14281
diff changeset
    43
        Ok(read_bytes)
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    44
    }
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    45
}
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    46
14293
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    47
pub struct IPCMessagesIterator<'a> {
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    48
    ipc: &'a mut IPC
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    49
}
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    50
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    51
impl<'a> IPCMessagesIterator<'a> {
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    52
    pub fn new(ipc: &'a mut IPC) -> Self {
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    53
        Self {
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    54
            ipc
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    55
        }
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    56
    }
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    57
}
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    58
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    59
impl<'a> Iterator for IPCMessagesIterator<'a> {
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    60
    type Item = EngineMessage;
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    61
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    62
    fn next(&mut self) -> Option<Self::Item> {
14293
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    63
        let (consumed, message) = extract_message(&self.ipc.in_buffer[..])?;
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    64
14293
3152d9fdb499 - Move EngineInstance into a separate module
unC0Rr
parents: 14282
diff changeset
    65
        self.ipc.in_buffer.consume(consumed);
14281
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    66
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    67
        Some(message)
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    68
    }
f0c0d2d217c3 IPC implementation
unC0Rr
parents: 14276
diff changeset
    69
}