buttplug 0.0.2-beta.1

Buttplug Intimate Hardware Control Library
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

// Buttplug Rust Source Code File - See https://buttplug.io for more info.
//
// Copyright 2016-2019 Nonpolynomial Labs LLC. All rights reserved.
//
// Licensed under the BSD 3-Clause license. See LICENSE file in the project root
// for full license information.

//! Handling of remote message pairing and future resolution.

use super::internal::ButtplugClientMessageStateShared;
use crate::core::messages::{ButtplugMessage, ButtplugMessageUnion};
use std::collections::HashMap;

pub struct ClientConnectorMessageSorter {
    future_map: HashMap<u32, ButtplugClientMessageStateShared>,
    current_id: u32,
}

impl ClientConnectorMessageSorter {
    pub fn new() -> Self {
        Self {
            future_map: HashMap::new(),
            current_id: 1,
        }
    }

    pub fn register_future(
        &mut self,
        msg: &mut ButtplugMessageUnion,
        state: &ButtplugClientMessageStateShared,
    ) {
        msg.set_id(self.current_id);
        self.future_map.insert(self.current_id, state.clone());
        self.current_id += 1;
    }

    pub fn maybe_resolve_message(&mut self, msg: &ButtplugMessageUnion) -> bool {
        match self.future_map.remove(&(msg.get_id())) {
            Some(_state) => {
                let mut waker_state = _state.lock().unwrap();
                waker_state.set_reply_msg(msg);
                true
            }
            None => {
                info!("Not found, may be event.");
                false
            }
        }
    }
}