From e69b8e69c6f0bdde457a088b4acd1c8b9d7d9ff3 Mon Sep 17 00:00:00 2001
From: Lijun Wang <83639177+lijunwangs@users.noreply.github.com>
Date: Thu, 24 Oct 2024 08:51:12 -0700
Subject: [PATCH] Initial implementations of the TPU Vortexor

---
 Cargo.lock               | 3 +++
 vortexor/Cargo.toml      | 4 ++++
 vortexor/src/lib.rs      | 1 +
 vortexor/src/vortexor.rs | 3 +++
 4 files changed, 11 insertions(+)

diff --git a/Cargo.lock b/Cargo.lock
index 092b1c3fe988ea..5aabd4818af490 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -8742,6 +8742,7 @@ dependencies = [
  "assert_matches",
  "async-channel",
  "bytes",
+ "clap 2.33.3",
  "crossbeam-channel",
  "dashmap",
  "futures 0.3.31",
@@ -8761,9 +8762,11 @@ dependencies = [
  "rustls 0.23.15",
  "smallvec",
  "socket2 0.5.7",
+ "solana-clap-utils",
  "solana-logger",
  "solana-measure",
  "solana-metrics",
+ "solana-net-utils",
  "solana-perf",
  "solana-sdk",
  "solana-streamer",
diff --git a/vortexor/Cargo.toml b/vortexor/Cargo.toml
index e73f6dff8643f2..a17d668b81c025 100644
--- a/vortexor/Cargo.toml
+++ b/vortexor/Cargo.toml
@@ -2,6 +2,7 @@
 name = "solana-vortexor"
 description = "Solana TPU Vortexor"
 documentation = "https://docs.rs/solana-vortexor"
+default-run = "solana-vortexor"
 version = { workspace = true }
 authors = { workspace = true }
 repository = { workspace = true }
@@ -12,6 +13,7 @@ edition = { workspace = true }
 [dependencies]
 async-channel = { workspace = true }
 bytes = { workspace = true }
+clap = { workspace = true }
 crossbeam-channel = { workspace = true }
 dashmap = { workspace = true }
 futures  = { workspace = true }
@@ -31,8 +33,10 @@ rand = { workspace = true }
 rustls = { workspace = true }
 smallvec = { workspace = true }
 socket2 = { workspace = true }
+solana-clap-utils = { workspace = true }
 solana-measure = { workspace = true }
 solana-metrics = { workspace = true }
+solana-net-utils = { workspace = true }
 solana-perf = { workspace = true }
 solana-sdk = { workspace = true }
 solana-streamer = { workspace = true }
diff --git a/vortexor/src/lib.rs b/vortexor/src/lib.rs
index 1b2d151dadd43a..e92563196fb2d4 100644
--- a/vortexor/src/lib.rs
+++ b/vortexor/src/lib.rs
@@ -1 +1,2 @@
+pub mod cli;
 pub mod vortexor;
diff --git a/vortexor/src/vortexor.rs b/vortexor/src/vortexor.rs
index 2046aff73b1aa2..d6e891d8204edb 100644
--- a/vortexor/src/vortexor.rs
+++ b/vortexor/src/vortexor.rs
@@ -14,6 +14,9 @@ use {
     },
 };
 
+pub const MAX_QUIC_CONNECTIONS_PER_PEER: usize = 8;
+pub const NUM_QUIC_ENDPOINTS: usize = 8;
+
 pub struct TpuSockets {
     pub tpu_quic: Vec<UdpSocket>,
     pub tpu_quic_fwd: Vec<UdpSocket>,