Add option to enable telemetry.

paritytech · Jul 25, 2020 · 3206576 · 3206576
1 parent 7104c42
commit 3206576
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/client/src/lib.rs b/client/src/lib.rs
@@ -53,6 +53,7 @@ use sc_service::{
     config::{
         NetworkConfiguration,
         TaskType,
+        TelemetryEndpoints,
     },
     ChainSpec,
     Configuration,
@@ -217,6 +218,8 @@ pub struct SubxtClientConfig<C: ChainSpec + 'static> {
     pub chain_spec: C,
     /// Role of the node.
     pub role: Role,
+    /// Enable telemetry.
+    pub enable_telemetry: bool,
 }
 
 impl<C: ChainSpec + 'static> SubxtClientConfig<C> {
@@ -235,6 +238,14 @@ impl<C: ChainSpec + 'static> SubxtClientConfig<C> {
             wasm_external_transport: None,
             use_yamux_flow_control: true,
         };
+        let telemetry_endpoints = if self.enable_telemetry {
+            let endpoints =
+                TelemetryEndpoints::new(vec![("/ip4/127.0.0.1/tcp/99000/ws".into(), 0)])
+                    .expect("valid config; qed");
+            Some(endpoints)
+        } else {
+            None
+        };
         let service_config = Configuration {
             network,
             impl_name: self.impl_name.to_string(),
@@ -253,8 +264,8 @@ impl<C: ChainSpec + 'static> SubxtClientConfig<C> {
             max_runtime_instances: 8,
             announce_block: true,
             dev_key_seed: self.role.into(),
+            telemetry_endpoints,
 
-            telemetry_endpoints: Default::default(),
             telemetry_external_transport: Default::default(),
             default_heap_pages: Default::default(),
             disable_grandpa: Default::default(),
@@ -345,6 +356,7 @@ mod tests {
             keystore: KeystoreConfig::InMemory,
             chain_spec,
             role: Role::Light,
+            enable_telemetry: false,
         };
         let client = ClientBuilder::<NodeTemplateRuntime>::new()
             .set_client(
@@ -377,6 +389,7 @@ mod tests {
             keystore: KeystoreConfig::InMemory,
             chain_spec: test_node::chain_spec::development_config().unwrap(),
             role: Role::Authority(AccountKeyring::Alice),
+            enable_telemetry: false,
         };
         let client = ClientBuilder::<NodeTemplateRuntime>::new()
             .set_client(

diff --git a/src/lib.rs b/src/lib.rs
@@ -492,6 +492,7 @@ mod tests {
             },
             chain_spec: test_node::chain_spec::development_config().unwrap(),
             role: Role::Authority(key),
+            enable_telemetry: false,
         };
         let client = ClientBuilder::new()
             .set_client(