feat(socket): req-socket tracing

Author: thedevbirb

Cargo.lock

@@ -15,3 +15,5 @@ rand.workspace = true
 futures.workspace = true
 tokio.workspace = true
 tokio-util.workspace = true
+derive_more.workspace = true
+tracing.workspace = true

msg-common/Cargo.toml

@@ -9,6 +9,7 @@ use std::{
     time::SystemTime,
 };
 
+use derive_more::{Deref, DerefMut};
 use futures::future::BoxFuture;
 
 mod channel;
@@ -52,14 +53,17 @@ pub trait SocketAddrExt: Sized {
 }
 
 impl SocketAddrExt for SocketAddr {
+    #[inline]
     fn unspecified_v4() -> Self {
         Self::V4(SocketAddrV4::new(Ipv4Addr::UNSPECIFIED, 0))
     }
 
+    #[inline]
     fn unspecified_v6() -> Self {
         Self::V6(SocketAddrV6::new(Ipv6Addr::UNSPECIFIED, 0, 0, 0))
     }
 
+    #[inline]
     fn as_unspecified(&self) -> Self {
         match self {
             Self::V4(_) => Self::unspecified_v4(),
@@ -75,6 +79,7 @@ pub trait IpAddrExt: Sized {
 }
 
 impl IpAddrExt for IpAddr {
+    #[inline]
     fn as_localhost(&self) -> Self {
         match self {
             Self::V4(_) => Self::V4(Ipv4Addr::LOCALHOST),
@@ -91,18 +96,21 @@ impl IdBase58 {
     const ALPHABET: &[u8] = b"123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz";
 
     /// Generates a new random Base58 short ID.
+    #[inline]
     pub fn new() -> Self {
         let raw: u64 = rand::random();
         Self::to_base58(raw)
     }
 
     /// Generates a new random Base58 short ID using the given RNG.
+    #[inline]
     pub fn new_with_rng<T: Rng>(rng: &mut T) -> Self {
         let raw: u64 = rng.r#gen();
         Self::to_base58(raw)
     }
 
     /// Converts the given u64 to a Base58 short ID.
+    #[inline]
     pub fn to_base58(mut x: u64) -> Self {
         let mut out = [b'1'; 6]; // '1' is the first in base58
 
@@ -127,3 +135,34 @@ impl std::fmt::Display for IdBase58 {
         f.write_str(unsafe { core::str::from_utf8_unchecked(&self.0) })
     }
 }
+
+/// A container with a [`tracing::Span`] attached that can be entered.
+#[derive(Debug, Clone, Deref, DerefMut)]
+pub struct Spanned<T> {
+    #[deref]
+    #[deref_mut]
+    pub inner: T,
+    pub span: tracing::Span,
+}
+
+impl<T> Spanned<T> {
+    #[inline]
+    pub fn new(inner: T) -> Self {
+        Self { inner, span: tracing::Span::current() }
+    }
+
+    #[inline]
+    pub fn new_with_span(inner: T, span: tracing::Span) -> Self {
+        Self { inner, span }
+    }
+
+    #[inline]
+    pub fn into_parts(self) -> (T, tracing::Span) {
+        (self.inner, self.span)
+    }
+
+    #[inline]
+    pub fn into_inner(self) -> T {
+        self.inner
+    }
+}