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

use super::{User, UserDB};
use crate::error::Result;
use async_trait::async_trait;
use std::{collections::HashMap, sync::Arc};
use tokio::sync::Mutex;

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

#[async_trait]
impl UserDB for InMemoryUserDB {
	async fn get_user(&self, key: &str) -> Option<User> {
		let db = self.db.lock().await;
		db.get(&key.to_string()).cloned()
	}

	async fn save_user(&self, u: &User) -> Result<()> {
		let db = &mut self.db.lock().await;
		db.insert(u.id.clone(), u.clone());

		Ok(())
	}
}