Shorten ByteBuf trait bounds everywhere

This commit is contained in:
Igor Katson 2024-08-25 13:29:28 +01:00
parent c7b7dc300f
commit 9d1ef5c35c
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
6 changed files with 35 additions and 27 deletions

View file

@ -3,8 +3,10 @@
// //
// Not useful outside of librqbit. // Not useful outside of librqbit.
use std::borrow::Borrow;
use bytes::Bytes; use bytes::Bytes;
use serde::{Deserialize, Deserializer}; use serde::{Deserialize, Deserializer, Serialize};
use clone_to_owned::CloneToOwned; use clone_to_owned::CloneToOwned;
@ -15,7 +17,9 @@ pub struct ByteBufOwned(pub bytes::Bytes);
#[serde(transparent)] #[serde(transparent)]
pub struct ByteBuf<'a>(pub &'a [u8]); pub struct ByteBuf<'a>(pub &'a [u8]);
pub trait ByteBufT { pub trait ByteBufT:
AsRef<[u8]> + std::hash::Hash + Serialize + Eq + core::fmt::Debug + CloneToOwned + Borrow<[u8]>
{
fn as_slice(&self) -> &[u8]; fn as_slice(&self) -> &[u8];
} }

View file

@ -1682,7 +1682,10 @@ impl PeerHandler {
msg.added_peers().for_each(|peer| { msg.added_peers().for_each(|peer| {
self.state self.state
.add_peer_if_not_seen(peer.addr) .add_peer_if_not_seen(peer.addr)
.inspect_err(|error| warn!(?peer, ?error, "failed to add peer")) .map_err(|error| {
warn!(?peer, ?error, "failed to add peer");
error
})
.ok(); .ok();
}); });
} }

View file

@ -1,19 +1,18 @@
use std::{ use std::{collections::HashMap, net::IpAddr};
collections::HashMap,
net::IpAddr,
};
use buffers::ByteBuf; use buffers::{ByteBuf, ByteBufT};
use bytes::Bytes; use bytes::Bytes;
use clone_to_owned::CloneToOwned; use clone_to_owned::CloneToOwned;
use serde::{Deserialize, Deserializer, Serialize}; use serde::{Deserialize, Deserializer, Serialize};
use crate::{EXTENDED_UT_METADATA_KEY, EXTENDED_UT_PEX_KEY, MY_EXTENDED_UT_METADATA, MY_EXTENDED_UT_PEX}; use crate::{
EXTENDED_UT_METADATA_KEY, EXTENDED_UT_PEX_KEY, MY_EXTENDED_UT_METADATA, MY_EXTENDED_UT_PEX,
};
use super::PeerExtendedMessageIds; use super::PeerExtendedMessageIds;
#[derive(Deserialize, Serialize, Debug, Default)] #[derive(Deserialize, Serialize, Debug, Default)]
pub struct ExtendedHandshake<ByteBuf: Eq + std::hash::Hash> { pub struct ExtendedHandshake<ByteBuf: ByteBufT> {
#[serde(bound(deserialize = "ByteBuf: From<&'de [u8]>"))] #[serde(bound(deserialize = "ByteBuf: From<&'de [u8]>"))]
pub m: HashMap<ByteBuf, u8>, pub m: HashMap<ByteBuf, u8>,
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
@ -50,7 +49,7 @@ impl ExtendedHandshake<ByteBuf<'static>> {
impl<'a, ByteBuf> ExtendedHandshake<ByteBuf> impl<'a, ByteBuf> ExtendedHandshake<ByteBuf>
where where
ByteBuf: Eq + std::hash::Hash + std::borrow::Borrow<[u8]>, ByteBuf: ByteBufT,
{ {
fn get_msgid(&self, msg_type: &'a [u8]) -> Option<u8> { fn get_msgid(&self, msg_type: &'a [u8]) -> Option<u8> {
self.m.get(msg_type).copied() self.m.get(msg_type).copied()
@ -74,8 +73,8 @@ where
impl<ByteBuf> CloneToOwned for ExtendedHandshake<ByteBuf> impl<ByteBuf> CloneToOwned for ExtendedHandshake<ByteBuf>
where where
ByteBuf: CloneToOwned + Eq + std::hash::Hash, ByteBuf: ByteBufT,
<ByteBuf as CloneToOwned>::Target: Eq + std::hash::Hash, <ByteBuf as CloneToOwned>::Target: ByteBufT,
{ {
type Target = ExtendedHandshake<<ByteBuf as CloneToOwned>::Target>; type Target = ExtendedHandshake<<ByteBuf as CloneToOwned>::Target>;

View file

@ -1,9 +1,10 @@
use bencode::bencode_serialize_to_writer; use bencode::bencode_serialize_to_writer;
use bencode::from_bytes; use bencode::from_bytes;
use bencode::BencodeValue; use bencode::BencodeValue;
use buffers::ByteBufT;
use bytes::Bytes; use bytes::Bytes;
use clone_to_owned::CloneToOwned; use clone_to_owned::CloneToOwned;
use serde::{Deserialize, Serialize}; use serde::Deserialize;
use ut_pex::UtPex; use ut_pex::UtPex;
use crate::MY_EXTENDED_UT_PEX; use crate::MY_EXTENDED_UT_PEX;
@ -25,7 +26,7 @@ pub struct PeerExtendedMessageIds {
} }
#[derive(Debug)] #[derive(Debug)]
pub enum ExtendedMessage<ByteBuf: std::hash::Hash + Eq + AsRef<[u8]>> { pub enum ExtendedMessage<ByteBuf: ByteBufT> {
Handshake(ExtendedHandshake<ByteBuf>), Handshake(ExtendedHandshake<ByteBuf>),
UtMetadata(UtMetadata<ByteBuf>), UtMetadata(UtMetadata<ByteBuf>),
UtPex(UtPex<ByteBuf>), UtPex(UtPex<ByteBuf>),
@ -34,8 +35,8 @@ pub enum ExtendedMessage<ByteBuf: std::hash::Hash + Eq + AsRef<[u8]>> {
impl<ByteBuf> CloneToOwned for ExtendedMessage<ByteBuf> impl<ByteBuf> CloneToOwned for ExtendedMessage<ByteBuf>
where where
ByteBuf: CloneToOwned + std::hash::Hash + Eq + AsRef<[u8]>, ByteBuf: ByteBufT,
<ByteBuf as CloneToOwned>::Target: std::hash::Hash + Eq + AsRef<[u8]>, <ByteBuf as CloneToOwned>::Target: ByteBufT,
{ {
type Target = ExtendedMessage<<ByteBuf as CloneToOwned>::Target>; type Target = ExtendedMessage<<ByteBuf as CloneToOwned>::Target>;
@ -53,7 +54,7 @@ where
} }
} }
impl<'a, ByteBuf: 'a + std::hash::Hash + Eq + Serialize + AsRef<[u8]>> ExtendedMessage<ByteBuf> { impl<ByteBuf: ByteBufT> ExtendedMessage<ByteBuf> {
pub fn serialize( pub fn serialize(
&self, &self,
out: &mut Vec<u8>, out: &mut Vec<u8>,
@ -93,9 +94,9 @@ impl<'a, ByteBuf: 'a + std::hash::Hash + Eq + Serialize + AsRef<[u8]>> ExtendedM
Ok(()) Ok(())
} }
pub fn deserialize(mut buf: &'a [u8]) -> Result<Self, MessageDeserializeError> pub fn deserialize<'a>(mut buf: &'a [u8]) -> Result<Self, MessageDeserializeError>
where where
ByteBuf: Deserialize<'a> + From<&'a [u8]> + AsRef<[u8]>, ByteBuf: Deserialize<'a> + From<&'a [u8]>,
{ {
let emsg_id = buf.first().copied().ok_or_else(|| { let emsg_id = buf.first().copied().ok_or_else(|| {
MessageDeserializeError::Other(anyhow::anyhow!( MessageDeserializeError::Other(anyhow::anyhow!(

View file

@ -1,5 +1,6 @@
use bencode::bencode_serialize_to_writer; use bencode::bencode_serialize_to_writer;
use bencode::BencodeDeserializer; use bencode::BencodeDeserializer;
use buffers::ByteBufT;
use bytes::Bytes; use bytes::Bytes;
use clone_to_owned::CloneToOwned; use clone_to_owned::CloneToOwned;
use serde::Deserialize; use serde::Deserialize;
@ -39,7 +40,7 @@ impl<ByteBuf: CloneToOwned> CloneToOwned for UtMetadata<ByteBuf> {
} }
} }
impl<'a, ByteBuf: 'a> UtMetadata<ByteBuf> { impl<ByteBuf: ByteBufT> UtMetadata<ByteBuf> {
pub fn serialize(&self, buf: &mut Vec<u8>) pub fn serialize(&self, buf: &mut Vec<u8>)
where where
ByteBuf: AsRef<[u8]>, ByteBuf: AsRef<[u8]>,
@ -83,7 +84,7 @@ impl<'a, ByteBuf: 'a> UtMetadata<ByteBuf> {
} }
} }
} }
pub fn deserialize(buf: &'a [u8]) -> Result<Self, MessageDeserializeError> pub fn deserialize<'a>(buf: &'a [u8]) -> Result<Self, MessageDeserializeError>
where where
ByteBuf: From<&'a [u8]>, ByteBuf: From<&'a [u8]>,
{ {

View file

@ -5,7 +5,7 @@
pub mod extended; pub mod extended;
use bincode::Options; use bincode::Options;
use buffers::{ByteBuf, ByteBufOwned}; use buffers::{ByteBuf, ByteBufOwned, ByteBufT};
use byteorder::{ByteOrder, BE}; use byteorder::{ByteOrder, BE};
use bytes::Bytes; use bytes::Bytes;
use clone_to_owned::CloneToOwned; use clone_to_owned::CloneToOwned;
@ -186,7 +186,7 @@ impl From<anyhow::Error> for MessageDeserializeError {
} }
#[derive(Debug)] #[derive(Debug)]
pub enum Message<ByteBuf: std::hash::Hash + Eq + AsRef<[u8]>> { pub enum Message<ByteBuf: ByteBufT> {
Request(Request), Request(Request),
Cancel(Request), Cancel(Request),
Bitfield(ByteBuf), Bitfield(ByteBuf),
@ -212,8 +212,8 @@ pub struct Bitfield<'a> {
impl<ByteBuf> CloneToOwned for Message<ByteBuf> impl<ByteBuf> CloneToOwned for Message<ByteBuf>
where where
ByteBuf: CloneToOwned + std::hash::Hash + Eq + AsRef<[u8]>, ByteBuf: ByteBufT,
<ByteBuf as CloneToOwned>::Target: std::hash::Hash + Eq + AsRef<[u8]>, <ByteBuf as CloneToOwned>::Target: ByteBufT,
{ {
type Target = Message<<ByteBuf as CloneToOwned>::Target>; type Target = Message<<ByteBuf as CloneToOwned>::Target>;
@ -257,7 +257,7 @@ impl<'a> std::fmt::Debug for Bitfield<'a> {
impl<ByteBuf> Message<ByteBuf> impl<ByteBuf> Message<ByteBuf>
where where
ByteBuf: AsRef<[u8]> + std::hash::Hash + Eq + Serialize, ByteBuf: ByteBufT,
{ {
pub fn len_prefix_and_msg_id(&self) -> (u32, u8) { pub fn len_prefix_and_msg_id(&self) -> (u32, u8) {
match self { match self {