CassandraNode new method (#832)

Co-authored-by: Lucas Kent <rubickent@gmail.com>
shotover · Sep 29, 2022 · 3583a92 · 3583a92
1 parent ab63447
commit 3583a92
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 13 deletions.
diff --git a/shotover-proxy/src/transforms/cassandra/sink_cluster/node.rs b/shotover-proxy/src/transforms/cassandra/sink_cluster/node.rs
@@ -22,6 +22,22 @@ pub struct CassandraNode {
 }
 
 impl CassandraNode {
+    pub fn new(
+        address: SocketAddr,
+        rack: String,
+        tokens: Vec<Murmur3Token>,
+        host_id: Uuid,
+    ) -> Self {
+        Self {
+            address,
+            rack,
+            tokens,
+            host_id,
+            outbound: None,
+            is_up: true,
+        }
+    }
+
     pub async fn get_connection(
         &mut self,
         connection_factory: &ConnectionFactory,

diff --git a/shotover-proxy/src/transforms/cassandra/sink_cluster/topology.rs b/shotover-proxy/src/transforms/cassandra/sink_cluster/topology.rs
@@ -254,14 +254,7 @@ mod system_local {
                             return Err(anyhow!("system.local.rack not a varchar"));
                         };
 
-                        Ok(CassandraNode {
-                            address,
-                            rack,
-                            tokens,
-                            outbound: None,
-                            host_id,
-                            is_up: true,
-                        })
+                        Ok(CassandraNode::new(address, rack, tokens, host_id))
                     })
                     .collect(),
                 operation => Err(anyhow!(
@@ -400,14 +393,12 @@ mod system_peers {
                             9042
                         };
 
-                        Ok(CassandraNode {
-                            address: SocketAddr::new(ip, port.try_into()?),
+                        Ok(CassandraNode::new(
+                            SocketAddr::new(ip, port.try_into()?),
                             rack,
                             tokens,
-                            outbound: None,
                             host_id,
-                            is_up: true,
-                        })
+                        ))
                     })
                     .collect(),
                 operation => Err(anyhow!(