parent
a277f3fd9d
commit
5b4fa57a9c
GPG Key ID:
284488987C31F630
@ -0,0 +1,29 @@
|
|||||||
|
use std::ptr;
|
||||||
|
|
||||||
|
/// Copy slice of bytes only
|
||||||
|
///
|
||||||
|
/// # Notes
|
||||||
|
/// `dst` and `src` must not overlap. See [move_slice].
|
||||||
|
pub fn copy_slice(dst: &mut [u8], src: &[u8]) -> usize
|
||||||
|
{
|
||||||
|
let sz = std::cmp::min(dst.len(),src.len());
|
||||||
|
unsafe {
|
||||||
|
//libc::memcpy(&mut dst[0] as *mut u8 as *mut c_void, &src[0] as *const u8 as *const c_void, sz);
|
||||||
|
ptr::copy_nonoverlapping(&src[0] as *const u8, &mut dst[0] as *mut u8, sz);
|
||||||
|
}
|
||||||
|
sz
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Move slice of bytes only
|
||||||
|
///
|
||||||
|
/// # Notes
|
||||||
|
/// `dst` and `src` can overlap.
|
||||||
|
pub fn move_slice(dst: &mut [u8], src: &[u8]) -> usize
|
||||||
|
{
|
||||||
|
let sz = std::cmp::min(dst.len(),src.len());
|
||||||
|
unsafe {
|
||||||
|
//libc::memmove(&mut dst[0] as *mut u8 as *mut c_void, &src[0] as *const u8 as *const c_void, sz);
|
||||||
|
ptr::copy(&src[0] as *const u8, &mut dst[0] as *mut u8, sz);
|
||||||
|
}
|
||||||
|
sz
|
||||||
|
}
|
||||||
@ -0,0 +1,153 @@
|
|||||||
|
//! Creating immutable images of state.
|
||||||
|
use super::*;
|
||||||
|
use std::{error,fmt};
|
||||||
|
|
||||||
|
|
||||||
|
/// An image of the entire post container
|
||||||
|
#[derive(Debug, Default, Serialize, Deserialize)]
|
||||||
|
pub struct Imouto
|
||||||
|
{
|
||||||
|
posts: Vec<post::Post>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<Imouto> for Oneesan
|
||||||
|
{
|
||||||
|
#[inline] fn from(from: Imouto) -> Self
|
||||||
|
{
|
||||||
|
Self::from_freeze(from)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TryFrom<Oneesan> for Imouto
|
||||||
|
{
|
||||||
|
type Error = FreezeError;
|
||||||
|
|
||||||
|
#[inline] fn try_from(from: Oneesan) -> Result<Self, Self::Error>
|
||||||
|
{
|
||||||
|
from.try_into_freeze()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// Error returned when a freeze operation fails
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct FreezeError{
|
||||||
|
held: Option<Arc<RwLock<post::Post>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FreezeError
|
||||||
|
{
|
||||||
|
/// The post associated with this error, if there is one.
|
||||||
|
pub fn post(&self) -> Option<&Arc<RwLock<post::Post>>>
|
||||||
|
{
|
||||||
|
self.held.as_ref()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl error::Error for FreezeError{}
|
||||||
|
impl fmt::Display for FreezeError
|
||||||
|
{
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||||
|
{
|
||||||
|
write!(f,"Failed to create freeze image")?;
|
||||||
|
|
||||||
|
if let Some(aref) = &self.held
|
||||||
|
{
|
||||||
|
let cnt = Arc::strong_count(&aref) - 1;
|
||||||
|
if cnt > 0 {
|
||||||
|
write!(f, "Post reference still held in {} other places.", cnt)
|
||||||
|
} else {
|
||||||
|
write!(f, "Post reference was still held in another place at the time, but no longer is.")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Oneesan
|
||||||
|
{
|
||||||
|
/// Create a serialisable image of this store by cloning each post into it.
|
||||||
|
pub async fn freeze(&self) -> Imouto
|
||||||
|
{
|
||||||
|
let read = self.posts.read().await;
|
||||||
|
let mut sis = Imouto{
|
||||||
|
posts: Vec::with_capacity(read.0.len()),
|
||||||
|
};
|
||||||
|
for (_, post) in read.0.iter()
|
||||||
|
{
|
||||||
|
sis.posts.push(post.read().await.clone());
|
||||||
|
}
|
||||||
|
|
||||||
|
sis
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Consume into a serialisable image of this store.
|
||||||
|
///
|
||||||
|
/// # Fails
|
||||||
|
/// If references to any posts are still held elsewhere.
|
||||||
|
pub fn try_into_freeze(self) -> Result<Imouto, FreezeError>
|
||||||
|
{
|
||||||
|
let read = self.posts.into_inner();
|
||||||
|
let mut sis = Imouto{
|
||||||
|
posts: Vec::with_capacity(read.0.len()),
|
||||||
|
};
|
||||||
|
for post in read.0.into_iter()
|
||||||
|
{
|
||||||
|
sis.posts.push(match Arc::try_unwrap(post) {
|
||||||
|
Ok(val) => val.into_inner(),
|
||||||
|
Err(arc) => return Err(FreezeError{held: Some(arc)}),
|
||||||
|
// Err(_arc) => panic!("Reference to post is still being used"),//arc.read().await.clone(), // something else holds the reference, we can't consume it.
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(sis)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Consume into a serialisable image of this store.
|
||||||
|
///
|
||||||
|
/// # Panics
|
||||||
|
/// If references to any posts are still held elsewhere.
|
||||||
|
pub fn into_freeze(self) -> Imouto
|
||||||
|
{
|
||||||
|
self.try_into_freeze().expect("Failed to consume into freeze")
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a new store from a serialisable image of one by cloning each post in it
|
||||||
|
pub fn unfreeze(freeze: &Imouto) -> Self
|
||||||
|
{
|
||||||
|
let mut posts = Arena::new();
|
||||||
|
let mut user_map = HashMap::new();
|
||||||
|
|
||||||
|
for post in freeze.posts.iter()
|
||||||
|
{
|
||||||
|
let idx = posts.insert(Arc::new(RwLock::new(post.clone())));
|
||||||
|
user_map.entry(post.owner().clone())
|
||||||
|
.or_insert_with(|| MaybeVec::new())
|
||||||
|
.push(idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
Self {
|
||||||
|
posts: RwLock::new((posts, user_map))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a new store by consuming serialisable image of one by cloning each post in it
|
||||||
|
pub fn from_freeze(freeze: Imouto) -> Self
|
||||||
|
{
|
||||||
|
let mut posts = Arena::new();
|
||||||
|
let mut user_map = HashMap::new();
|
||||||
|
|
||||||
|
for post in freeze.posts.into_iter()
|
||||||
|
{
|
||||||
|
let mapping = user_map.entry(post.owner().clone())
|
||||||
|
.or_insert_with(|| MaybeVec::new());
|
||||||
|
let idx = posts.insert(Arc::new(RwLock::new(post)));
|
||||||
|
mapping.push(idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
Self {
|
||||||
|
posts: RwLock::new((posts, user_map))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,5 +1,22 @@
|
|||||||
use super::*;
|
use super::*;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use generational_arena::{
|
||||||
|
Arena, Index as ArenaIndex,
|
||||||
|
};
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
pub mod session;
|
||||||
pub mod user;
|
pub mod user;
|
||||||
pub mod post;
|
pub mod post;
|
||||||
pub mod body;
|
pub mod body;
|
||||||
|
|
||||||
|
mod freeze; pub use freeze::*;
|
||||||
|
|
||||||
|
/// Entire post container
|
||||||
|
pub struct Oneesan
|
||||||
|
{
|
||||||
|
posts: RwLock<(Arena<Arc<RwLock<post::Post>>> // All posts
|
||||||
|
, HashMap<user::UserID, MaybeVec<ArenaIndex>> // Post lookup by user ID
|
||||||
|
)>,
|
||||||
|
}
|
||||||
|
|||||||
@ -0,0 +1,25 @@
|
|||||||
|
//! Session for each connected user
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
id_type!(SessionID; "A unique session ID, not bound to a user.");
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Session
|
||||||
|
{
|
||||||
|
id: SessionID,
|
||||||
|
user: user::User,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Session
|
||||||
|
{
|
||||||
|
/// The randomly generated ID of this session, irrespective of the user of this session.
|
||||||
|
#[inline] pub fn session_id(&self) -> &SessionID
|
||||||
|
{
|
||||||
|
&self.id
|
||||||
|
}
|
||||||
|
/// The unique user ID of this session
|
||||||
|
pub fn user_id(&self) -> user::UserID
|
||||||
|
{
|
||||||
|
self.user.id_for_session(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue