p/pukram-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Made site infos saveable on their own

Browse Source
This commit is contained in:
p11 2023-07-19 21:18:28 +02:00
parent 4eb59a01b7
commit 8a1f94bb66
1 changed files with 55 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
src/main.rs
View File

@ -60,21 +60,10 @@ struct SiteInfo {
down: usize,
}
#[derive(Default)]
struct Context {
infos: HashMap<Box<str>, SiteInfo>,
}
impl<S: SizeSettings> ToStream<S> for Context {
impl<S: SizeSettings> ToStream<S> for SiteInfo {
fn to_stream<W: Write>(&self, stream: &mut W) -> Result<()> {
S::size_to_stream(self.infos.len(), stream)?;
for (key, info) in &self.infos {
let key_bytes = key.as_bytes();
S::size_to_stream(key_bytes.len(), stream)?;
stream.write_all(key_bytes)?;
S::size_to_stream(info.comments.len(), stream)?;
for Comment { name, text } in &info.comments {
S::size_to_stream(self.comments.len(), stream)?;
for Comment { name, text } in &self.comments {
let name_bytes = name.as_bytes();
S::size_to_stream(name_bytes.len(), stream)?;
stream.write_all(name_bytes)?;
@ -84,25 +73,16 @@ impl<S: SizeSettings> ToStream<S> for Context {
stream.write_all(text_bytes)?;
}
S::size_to_stream(info.visits, stream)?;
S::size_to_stream(info.up, stream)?;
S::size_to_stream(info.down, stream)?;
}
S::size_to_stream(self.visits, stream)?;
S::size_to_stream(self.up, stream)?;
S::size_to_stream(self.down, stream)?;
Ok(())
}
}
impl<S: SizeSettings> FromStream<S> for Context {
impl<S: SizeSettings> FromStream<S> for SiteInfo {
fn from_stream<R: Read>(stream: &mut R) -> Result<Self> {
let size = S::size_from_stream(stream)?;
let infos = (0..size)
.map(|_| {
let key_bytes = <Vec<_> as FromStream<S>>::from_stream(stream)?;
let key = std::str::from_utf8(&key_bytes)
.map_err(|e| Error::new(ErrorKind::InvalidData, e))?
.into();
let size = S::size_from_stream(stream)?;
let comments = (0..size)
.map(|_| {
@ -124,15 +104,46 @@ impl<S: SizeSettings> FromStream<S> for Context {
let up = S::size_from_stream(stream)?;
let down = S::size_from_stream(stream)?;
Ok((
key,
SiteInfo {
Ok(SiteInfo {
comments,
visits,
up,
down,
},
))
})
}
}
#[derive(Default)]
struct Context {
infos: HashMap<Box<str>, SiteInfo>,
}
impl<S: SizeSettings> ToStream<S> for Context {
fn to_stream<W: Write>(&self, stream: &mut W) -> Result<()> {
S::size_to_stream(self.infos.len(), stream)?;
for (key, info) in &self.infos {
let key_bytes = key.as_bytes();
S::size_to_stream(key_bytes.len(), stream)?;
stream.write_all(key_bytes)?;
to_stream::<S, _, _>(info, stream)?;
}
Ok(())
}
}
impl<S: SizeSettings> FromStream<S> for Context {
fn from_stream<R: Read>(stream: &mut R) -> Result<Self> {
let size = S::size_from_stream(stream)?;
let infos = (0..size)
.map(|_| {
let key_bytes = <Vec<_> as FromStream<S>>::from_stream(stream)?;
let key = std::str::from_utf8(&key_bytes)
.map_err(|e| Error::new(ErrorKind::InvalidData, e))?
.into();
Ok((key, from_stream::<S, SiteInfo, _>(stream)?))
})
.collect::<Result<HashMap<_, _>>>()?;