chore: Move network services to services flags. (#1074)

googleforgames · Jan 22, 2025 · 0e4b949 · 0e4b949
1 parent 588a4ca
commit 0e4b949
Show file tree

Hide file tree

Showing 7 changed files with 467 additions and 19 deletions.
diff --git a/crates/test/tests/proxy.rs b/crates/test/tests/proxy.rs
@@ -109,7 +109,6 @@ trace_test!(uring_receiver, {
         ),
     }
     .spawn(pending_sends)
-    .await
     .expect("failed to spawn task");
 
     // Drop the socket, otherwise it can
@@ -161,7 +160,6 @@ trace_test!(
             &sessions,
             BUFFER_POOL.clone(),
         )
-        .await
         .unwrap();
 
         let socket = std::sync::Arc::new(sb.client());

diff --git a/src/cli.rs b/src/cli.rs
@@ -28,7 +28,7 @@ use strum_macros::{Display, EnumString};
 
 pub use self::{
     agent::Agent, generate_config_schema::GenerateConfigSchema, manage::Manage, proxy::Proxy,
-    qcmp::Qcmp, relay::Relay,
+    qcmp::Qcmp, relay::Relay, service::Service,
 };
 
 macro_rules! define_port {
@@ -47,6 +47,7 @@ pub mod manage;
 pub mod proxy;
 pub mod qcmp;
 pub mod relay;
+mod service;
 
 const ETC_CONFIG_PATH: &str = "/etc/quilkin/quilkin.yaml";
 const PORT_ENV_VAR: &str = "QUILKIN_PORT";
@@ -121,6 +122,8 @@ pub struct Cli {
     pub admin: AdminCli,
     #[command(flatten)]
     pub locality: LocalityCli,
+    #[command(flatten)]
+    pub service: Service,
 }
 
 /// The various log format options