fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-04-25 03:02:12 +00:00
Code Issues Projects Releases Wiki Activity

add fn size

Browse Source
This commit is contained in:
folex 2020-08-04 20:26:21 +03:00
parent ecbd044c30
commit a7d8e21550
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
protocols/kad/src/record.rs
View File

@ -37,6 +37,8 @@ use std::fmt::Debug;
pub trait RecordT: Eq + Send + Clone + Debug + Into<proto::Record> + TryFrom<proto::Record, Error = io::Error> + 'static {
type Key: Clone + Send + Debug + AsRef<[u8]> + Borrow<[u8]> + From<Vec<u8>> + Hash + Eq;
fn size(&self) -> usize;
fn key(&self) -> &Self::Key;
fn into_key(self) -> Self::Key;
@ -51,6 +53,10 @@ pub trait RecordT: Eq + Send + Clone + Debug + Into<proto::Record> + TryFrom<pro
impl RecordT for Record {
type Key = Key;
fn size(&self) -> usize {
self.value.len()
}
fn key(&self) -> &Self::Key {
&self.key
}

6
protocols/kad/src/record/store/memory.rs
View File

@ -111,9 +111,9 @@ impl<'a, TRecord: RecordT> RecordStore<'a, TRecord> for MemoryStore<TRecord> {
}
fn put(&'a mut self, r: TRecord) -> Result<()> {
// if r.value.len() >= self.config.max_value_bytes {
// return Err(Error::ValueTooLarge)
// }
if r.size() >= self.config.max_value_bytes {
return Err(Error::ValueTooLarge)
}
let num_records = self.records.len();