atlasserver 0.3.0

Atlas is a rust library for the purpose of composing REST APIs out of re-usable and extensible modules, specifically with supporting the networking needs of online gaming services in mind.
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 super::{Session, SessionDB};
use crate::error::Result;
use async_trait::async_trait;
use std::{collections::HashMap, sync::Arc};
use tokio::sync::Mutex;

#[derive(Default)]
pub struct InMemorySessionDB {
	pub db: Arc<Mutex<HashMap<String, Session>>>,
}

#[async_trait]
impl SessionDB for InMemorySessionDB {
	async fn get(&self, key: &str) -> Option<Session> {
		self.db.lock().await.get(&key.to_string()).cloned()
	}

	async fn invalidate(&self, key: &str) -> Option<()> {
		self.db.lock().await.get_mut(&key.to_string()).map(
			|session| {
				session.valid = false;
			},
		)
	}

	async fn create(&self, session: Session) -> Result<String> {
		let key = Session::new_key();

		self.db.lock().await.insert(key.clone(), session);

		Ok(key)
	}
}