fix(pb-envoy): reduce round trips for ws connection

Author: MasterPtato

Cargo.lock

@@ -56,9 +56,7 @@ members = [
   "engine/sdks/rust/envoy-protocol",
   "engine/sdks/rust/epoxy-protocol",
   "engine/sdks/rust/test-envoy",
-  "engine/sdks/rust/ups-protocol",
-  "rivetkit-typescript/packages/rivetkit-native",
-  "rivetkit-typescript/packages/sqlite-native"
+  "engine/sdks/rust/ups-protocol"
 ]
 
   [workspace.package]
@@ -137,7 +135,6 @@ members = [
   tokio-cron-scheduler = "0.13.0"
   tokio-native-tls = "0.3.1"
   tokio-stream = "0.1.15"
-  tokio-tungstenite = { version = "0.26.0", features = ["rustls-tls-native-roots"] }
   tokio-util = "0.7"
   tower = "0.5"
   tracing = "0.1.40"
@@ -147,6 +144,10 @@ members = [
   vergen-gitcl = "1.0.0"
   reqwest-eventsource = "0.6.0"
 
+    [workspace.dependencies.tokio-tungstenite]
+    version = "0.26.0"
+    features = [ "rustls-tls-native-roots" ]
+
     [workspace.dependencies.rdkafka]
     version = "0.38.0"
     features = [ "ssl" ]
@@ -531,9 +532,6 @@ members = [
     [workspace.dependencies.rivet-ups-protocol]
     path = "engine/sdks/rust/ups-protocol"
 
-    [workspace.dependencies.rivetkit-sqlite-native]
-    path = "rivetkit-typescript/packages/sqlite-native"
-
 [profile.dev]
 overflow-checks = false
 debug = true

Cargo.toml

@@ -23,6 +23,8 @@ pub enum RunnerConfigKind {
 		/// Seconds.
 		request_lifespan: u32,
 		max_concurrent_actors: Option<u64>,
+		/// Seconds.
+		drain_grace_period: Option<u32>,
 		slots_per_runner: u32,
 		min_runners: Option<u32>,
 		max_runners: u32,
@@ -50,6 +52,7 @@ impl Into<rivet_types::runner_configs::RunnerConfig> for RunnerConfig {
 				headers,
 				request_lifespan,
 				max_concurrent_actors,
+				drain_grace_period,
 				slots_per_runner,
 				min_runners,
 				max_runners,
@@ -60,6 +63,8 @@ impl Into<rivet_types::runner_configs::RunnerConfig> for RunnerConfig {
 				headers: headers.unwrap_or_default(),
 				request_lifespan,
 				max_concurrent_actors: max_concurrent_actors.unwrap_or(max_runners as u64),
+				// Default to deprecated config value (config.pegboard.serverless_drain_grace_period)
+				drain_grace_period: drain_grace_period.unwrap_or(10_000),
 				slots_per_runner,
 				min_runners: min_runners.unwrap_or_default(),
 				max_runners,

engine/artifacts/config-schema.json

@@ -138,6 +138,8 @@ pub struct Pegboard {
 	pub envoy_eligible_threshold: Option<i64>,
 
 	// === Serverless Settings ===
+	/// **Deprecated** Configure the drain period in the runner config.
+	///
 	/// Drain grace period for serverless runners.
 	///
 	/// This time is subtracted from the configured request duration. Once `duration - grace` is reached, the