parent
31ecef8954
commit
0a6c1304dc
GPG Key ID:
284488987C31F630
@ -1,2 +1 @@
|
||||
use super::*;
|
||||
|
||||
//! Services
|
||||
|
||||
@ -1,14 +0,0 @@
|
||||
use super::*;
|
||||
use std::collections::{HashMap, HashSet};
|
||||
use super::*;
|
||||
|
||||
use user::{User, UserID};
|
||||
use post::Post;
|
||||
|
||||
/// Contains all posts
|
||||
#[derive(Debug)]
|
||||
pub struct State
|
||||
{
|
||||
users: HashSet<User>,
|
||||
posts: HashMap<UserID, MaybeVec<Post>>
|
||||
}
|
||||
@ -0,0 +1,91 @@
|
||||
//! Frozen, serialisable state
|
||||
use super::*;
|
||||
use futures::prelude::*;
|
||||
|
||||
/// An immutable image of `State`.
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
|
||||
pub struct Freeze
|
||||
{
|
||||
users: HashSet<User>,
|
||||
posts: Vec<(UserID, Post)>,
|
||||
}
|
||||
|
||||
impl Freeze
|
||||
{
|
||||
/// Create a `State` from this freeze
|
||||
///
|
||||
/// # Notes
|
||||
/// This clones all post and user data, to perform this operation without cloning, use `into_state`.
|
||||
pub fn unfreeze(&self) -> State
|
||||
{
|
||||
let users: HashMap<UserID, RwLock<User>> = self.users
|
||||
.iter()
|
||||
.map(|x| (*x.id(), RwLock::new(x.clone()))).collect();
|
||||
let mut posts: HashMap<UserID, MaybeVec<RwLock<Post>>> = HashMap::with_capacity(users.len());
|
||||
|
||||
for (id, post) in self.posts.iter()
|
||||
{
|
||||
posts.entry(*id).or_insert_with(move || MaybeVec::new()).push(RwLock::new(post.clone()));
|
||||
}
|
||||
|
||||
State(Arc::new(Inner{posts: RwLock::new( Posts {
|
||||
users,
|
||||
posts
|
||||
})}))
|
||||
}
|
||||
|
||||
/// Consume into a new `State`.
|
||||
pub fn into_state(self) -> State
|
||||
{
|
||||
State::from_freeze(self)
|
||||
}
|
||||
}
|
||||
|
||||
impl State
|
||||
{
|
||||
/// Create a serialisable image from this state
|
||||
pub async fn freeze(&self) -> Freeze
|
||||
{
|
||||
let posts = self.0.posts.read().await;
|
||||
let users: HashSet<User> = {
|
||||
stream::iter(posts.users.iter())
|
||||
.then(|(_, x)| async move { x.read().await })
|
||||
.map(|x| x.clone()).collect().await
|
||||
};
|
||||
let posts: Vec<_> = {
|
||||
stream::iter(posts.posts.iter()
|
||||
.map(|(x, y)| y.into_iter().map(move |z| (x, z))).flatten())
|
||||
.then(|x| async move { (x.0, x.1.read().await) })
|
||||
.map(|(&id, post)| (id, post.clone())).collect().await
|
||||
};
|
||||
|
||||
Freeze {users, posts}
|
||||
}
|
||||
|
||||
/// Create `State` from this image
|
||||
pub fn from_freeze(freeze: Freeze) -> Self
|
||||
{
|
||||
let users: HashMap<UserID, RwLock<User>> = freeze.users
|
||||
.into_iter()
|
||||
.map(|x| (*x.id(), RwLock::new(x))).collect();
|
||||
let mut posts: HashMap<UserID, MaybeVec<RwLock<Post>>> = HashMap::with_capacity(users.len());
|
||||
|
||||
for (id, post) in freeze.posts.into_iter()
|
||||
{
|
||||
posts.entry(id).or_insert_with(move || MaybeVec::new()).push(RwLock::new(post));
|
||||
}
|
||||
|
||||
Self(Arc::new(Inner{posts: RwLock::new( Posts {
|
||||
users,
|
||||
posts
|
||||
})}))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<Freeze> for State
|
||||
{
|
||||
#[inline] fn from(from: Freeze) -> Self
|
||||
{
|
||||
Self::from_freeze(from)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,27 @@
|
||||
use super::*;
|
||||
use std::collections::{HashMap, HashSet};
|
||||
use std::sync::Arc;
|
||||
use tokio::sync::RwLock;
|
||||
|
||||
use user::{User, UserID};
|
||||
use post::Post;
|
||||
|
||||
mod freeze;
|
||||
pub use freeze::*;
|
||||
|
||||
#[derive(Debug)]
|
||||
struct Posts
|
||||
{
|
||||
users: HashMap<UserID, RwLock<User>>,
|
||||
posts: HashMap<UserID, MaybeVec<RwLock<Post>>>,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
struct Inner
|
||||
{
|
||||
posts: RwLock<Posts>,
|
||||
}
|
||||
|
||||
/// Contains all posts and users
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct State(Arc<Inner>);
|
||||
Loading…
Reference in new issue