Merge pull request #526 from cvybhu/get_traced

tracing: Get list of nodes from  TracingInfo

Author: piodul

scylla/src/tracing.rs

@@ -1,4 +1,5 @@
 use crate::statement::Consistency;
+use itertools::Itertools;
 use std::collections::HashMap;
 use std::net::IpAddr;
 use std::num::NonZeroU32;
@@ -48,6 +49,17 @@ pub struct GetTracingConfig {
     pub consistency: Consistency,
 }
 
+impl TracingInfo {
+    /// Returns a list of unique nodes involved in the query
+    pub fn nodes(&self) -> Vec<IpAddr> {
+        self.events
+            .iter()
+            .filter_map(|e| e.source)
+            .unique()
+            .collect()
+    }
+}
+
 impl Default for GetTracingConfig {
     fn default() -> GetTracingConfig {
         GetTracingConfig {

scylla/src/transport/session_test.rs

@@ -919,6 +919,7 @@ async fn test_get_tracing_info(session: &Session, ks: String) {
     // Getting tracing info from session using this uuid works
     let tracing_info: TracingInfo = session.get_tracing_info(&tracing_id).await.unwrap();
     assert!(!tracing_info.events.is_empty());
+    assert!(!tracing_info.nodes().is_empty());
 }
 
 async fn test_tracing_query_iter(session: &Session, ks: String) {