soulseek-rs-lib 13.0.0

Library for Soulseek protocol implementation in Rust
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

use std::collections::HashMap;

use crate::message::Message;
use std::sync::mpsc::Sender;

pub trait MessageHandler<Op>: Send {
    fn get_code(&self) -> u8;
    fn handle(&self, message: &mut Message, sender: Sender<Op>);
}
pub struct Handlers<Op> {
    handlers: HashMap<u8, Box<dyn MessageHandler<Op> + Send>>,
}

impl<Op> Default for Handlers<Op> {
    fn default() -> Self {
        Self::new()
    }
}

impl<Op> Handlers<Op> {
    #[must_use]
    pub fn new() -> Self {
        Self {
            handlers: HashMap::new(),
        }
    }

    pub fn register_handler<H>(&mut self, handler: H) -> &mut Self
    where
        H: 'static + MessageHandler<Op> + Send + Sync,
    {
        self.handlers.insert(handler.get_code(), Box::new(handler));
        self
    }
    #[must_use]
    pub fn get_handler(
        &self,
        code: u8,
    ) -> Option<&(dyn MessageHandler<Op> + Send)> {
        self.handlers.get(&code).map(|v| &**v)
    }
}