feat: add initial metrics for p2p comms

Cargo.toml

@@ -3,7 +3,7 @@ resolver = "2"
 
 members = [
     "atoma-confidential",
-    "atoma-daemon", 
+    "atoma-daemon",
     "atoma-p2p",
     "atoma-service",
     "atoma-state",
@@ -51,6 +51,9 @@ metrics = "0.23"
 metrics-exporter-prometheus = "0.14.0"
 nvml-wrapper = "0.10.0"
 once_cell = "1.20.3"
+opentelemetry = "0.27.0"
+opentelemetry_sdk = "0.27.0"
+opentelemetry-otlp = "0.27.0"
 prometheus = "0.13.4"
 rand = "0.8.5"
 reqwest = "0.12.1"
@@ -75,6 +78,7 @@ tower-http = "0.6.2"
 tracing = "0.1.40"
 tracing-appender = "0.2.3"
 tracing-subscriber = "0.3.18"
+tracing-opentelemetry = "0.28.0"
 url = "2.5.4"
 utoipa = "5.1.1"
 utoipa-swagger-ui = "8.0.1"

atoma-daemon/Cargo.toml

@@ -24,14 +24,14 @@ serde_yaml = { workspace = true }
 sui-sdk = { workspace = true }
 tokio = { workspace = true }
 once_cell = "1.20"
-opentelemetry = { version = "0.27.1", features = ["trace", "metrics", "logs"] }
-opentelemetry_sdk = { version = "0.27.1", features = [
+opentelemetry = { workspace = true, features = ["trace", "metrics", "logs"] }
+opentelemetry_sdk = { workspace = true, features = [
   "rt-tokio",
   "trace",
   "metrics",
   "logs",
 ] }
-opentelemetry-otlp = { version = "0.27.0", features = [
+opentelemetry-otlp = { workspace = true, features = [
   "metrics",
   "grpc-tonic",
   "trace",
@@ -45,7 +45,7 @@ tracing-subscriber = { workspace = true, features = [
   "json",
   "local-time",
 ] }
-tracing-opentelemetry = "0.28.0"
+tracing-opentelemetry = { workspace = true }
 utoipa = { workspace = true, features = ["axum_extras"] }
 utoipa-swagger-ui = { workspace = true, features = ["axum"] }
 

atoma-p2p/Cargo.toml

@@ -22,6 +22,7 @@ libp2p = { workspace = true, features = [
 	"tcp",
 	"yamux",
 	"noise",
+	"metrics",
 ] }
 fastcrypto = { workspace = true }
 flume = { workspace = true }
@@ -36,6 +37,8 @@ tokio = { workspace = true, features = ["full"] }
 tracing = { workspace = true }
 url = { workspace = true }
 validator = { workspace = true, features = ["derive"] }
+opentelemetry = { workspace = true }
+once_cell = { workspace = true }
 
 [dev-dependencies]
 rand = { workspace = true }

atoma-p2p/src/errors.rs

@@ -0,0 +1,63 @@
+use config::ConfigError;
+use libp2p::{
+    gossipsub::{ConfigBuilderError, PublishError, SubscriptionError},
+    swarm::DialError,
+    TransportError,
+};
+use thiserror::Error;
+
+use crate::service::StateManagerEvent;
+
+#[derive(Debug, Error)]
+pub enum AtomaP2pNodeError {
+    #[error("Failed to build gossipsub config: {0}")]
+    GossipsubConfigError(#[from] ConfigBuilderError),
+    #[error("Failed to build behaviour: {0}")]
+    BehaviourBuildError(String),
+    #[error("Failed to subscribe to topic: {0}")]
+    GossipsubSubscriptionError(#[from] SubscriptionError),
+    #[error("Failed to listen on address: {0}")]
+    SwarmListenOnError(#[from] TransportError<std::io::Error>),
+    #[error("Failed to dial bootstrap node: {0}")]
+    BootstrapNodeDialError(#[from] DialError),
+    #[error("Failed to parse signature: {0}")]
+    SignatureParseError(String),
+    #[error("Failed to verify signature: {0}")]
+    SignatureVerificationError(String),
+    #[error("Invalid public address: {0}")]
+    InvalidPublicAddressError(String),
+    #[error("Failed to parse listen address: {0}")]
+    ListenAddressParseError(#[from] libp2p::multiaddr::Error),
+    #[error("Failed to build TCP config: {0}")]
+    TcpConfigBuildError(#[from] ConfigError),
+    #[error("Failed to initialize noise encryption: {0}")]
+    NoiseError(#[from] libp2p::noise::Error),
+    #[error("Failed to send event to state manager: {0}")]
+    StateManagerError(#[from] flume::SendError<StateManagerEvent>),
+    #[error("Failed to sign hashed message, with error: {0}")]
+    SignatureError(String),
+    #[error("Failed to publish gossipsub message: {0}")]
+    GossipsubMessagePublishError(#[from] PublishError),
+    #[error("Failed to verify node small ID ownership: {0}")]
+    NodeSmallIdOwnershipVerificationError(String),
+    #[error("Failed to send usage metrics")]
+    UsageMetricsSendError,
+    #[error("Failed to serialize usage metrics: `{0}`")]
+    UsageMetricsSerializeError(#[from] ciborium::ser::Error<std::io::Error>),
+    #[error("Failed to deserialize usage metrics: `{0}`")]
+    UsageMetricsDeserializeError(#[from] ciborium::de::Error<std::io::Error>),
+    #[error("Failed to parse URL: `{0}`")]
+    UrlParseError(#[from] url::ParseError),
+    #[error("Country code is invalid: `{0}`")]
+    InvalidCountryCodeError(String),
+    #[error("Validation error: `{0}`")]
+    ValidationError(#[from] validator::ValidationError),
+    #[error("Failed to compute usage metrics: `{0}`")]
+    UsageMetricsComputeError(#[from] crate::metrics::NodeMetricsError),
+    #[error("Invalid config: `{0}`")]
+    InvalidConfig(String),
+    #[error("Invalid message length")]
+    InvalidMessageLengthError,
+    #[error("Failed to publish Message: `{0}`")]
+    PublishError(String),
+}

atoma-p2p/src/lib.rs

@@ -1,8 +1,13 @@
 pub mod config;
+pub mod errors;
 pub mod metrics;
 pub mod service;
 pub mod timer;
 pub mod types;
+pub mod utils;
+
+#[cfg(test)]
+mod tests;
 
 pub use config::AtomaP2pNodeConfig;
 pub use service::AtomaP2pNode;