connection_pool/node: expose node sharder

Make connection_pool sharder accessible in node

Author: wyfo

scylla/src/transport/connection_pool.rs

@@ -200,6 +200,14 @@ impl NodeConnectionPool {
         }
     }
 
+    pub fn sharder(&self) -> Option<Sharder> {
+        self.with_connections(|pool_conns| match pool_conns {
+            PoolConnections::NotSharded(_) => None,
+            PoolConnections::Sharded { sharder, .. } => Some(sharder.clone()),
+        })
+        .unwrap_or(None)
+    }
+
     pub fn connection_for_token(&self, token: Token) -> Result<Arc<Connection>, QueryError> {
         trace!(token = token.value, "Selecting connection for token");
         self.with_connections(|pool_conns| match pool_conns {

scylla/src/transport/node.rs

@@ -1,5 +1,5 @@
 /// Node represents a cluster node along with it's data and connections
-use crate::routing::Token;
+use crate::routing::{Sharder, Token};
 use crate::transport::connection::Connection;
 use crate::transport::connection::VerifiedKeyspaceName;
 use crate::transport::connection_pool::{NodeConnectionPool, PoolConfig};
@@ -53,6 +53,10 @@ impl Node {
         }
     }
 
+    pub fn sharder(&self) -> Option<Sharder> {
+        self.pool.sharder()
+    }
+
     /// Get connection which should be used to connect using given token
     /// If this connection is broken get any random connection to this Node
     pub(crate) async fn connection_for_token(