puniyu_adapter_server 0.3.6

puniyu的server适配器
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

use actix_ws::Session;
use dashmap::DashMap;
use std::sync::Arc;
use tokio::sync::Mutex;

pub struct ConnectionStore {
	connections: DashMap<String, Arc<Mutex<Session>>>,
}

impl ConnectionStore {
	pub fn new() -> Self {
		Self { connections: DashMap::new() }
	}

	pub fn add(&self, bot_name: String, session: Arc<Mutex<Session>>) {
		self.connections.insert(bot_name, session);
	}

	pub fn remove(&self, bot_name: &str) {
		self.connections.remove(bot_name);
	}

	pub fn get(&self, bot_name: &str) -> Option<Arc<Mutex<Session>>> {
		self.connections.get(bot_name).map(|v| Arc::clone(&v))
	}

	pub fn get_all(&self) -> Vec<(String, Arc<Mutex<Session>>)> {
		self.connections
			.iter()
			.map(|entry| (entry.key().to_string(), Arc::clone(entry.value())))
			.collect()
	}
}