[dnm] envd startup simulation

Author: teskje

Cargo.lock

@@ -116,6 +116,7 @@ members = [
     "src/walkabout",
     "src/workspace-hack",
     "test/metabase/smoketest",
+    "test/simulation",
     "test/test-util",
 ]
 
@@ -361,13 +362,15 @@ duckdb = { git = "https://github.com/MaterializeInc/duckdb-rs.git", rev = "752c7
 
 turmoil = { git = "https://github.com/teskje/turmoil.git", branch = "mz" }
 
+[patch.'https://github.com/tokio-rs/turmoil']
+turmoil = { git = "https://github.com/teskje/turmoil.git", branch = "mz" }
 
 # BEGIN LINT CONFIG
 # DO NOT EDIT. Automatically generated by bin/gen-lints.
 [workspace.lints.rust]
 unknown_lints = "allow"
 non_local_definitions = "allow"
-unexpected_cfgs = { level = "warn", check-cfg = ['cfg(stamped, coverage, nightly_doc_features, release, tokio_unstable)'] }
+unexpected_cfgs = { level = "warn", check-cfg = ['cfg(coverage, nightly_doc_features, release, simulation, tokio_unstable)'] }
 
 [workspace.lints.rustdoc]
 

Cargo.toml

@@ -185,7 +185,7 @@
 
 EXCLUDE_CRATES = ["workspace-hack"]
 
-CHECK_CFGS = "stamped, coverage, nightly_doc_features, release, tokio_unstable"
+CHECK_CFGS = "coverage, nightly_doc_features, release, simulation, tokio_unstable"
 
 
 def main() -> None:

misc/python/materialize/cli/gen-lints.py

@@ -14,7 +14,7 @@ rust-version = "1.76.0"
 [workspace.lints.rust]
 unknown_lints = "allow"
 non_local_definitions = "allow"
-unexpected_cfgs = { level = "warn", check-cfg = ['cfg(stamped, coverage, nightly_doc_features, release, tokio_unstable)'] }
+unexpected_cfgs = { level = "warn", check-cfg = ['cfg(coverage, nightly_doc_features, release, simulation, tokio_unstable)'] }
 
 [workspace.lints.rustdoc]
 

misc/wasm/Cargo.toml

@@ -31,7 +31,6 @@ use mz_ore::instrument;
 use mz_ore::now::{EpochMillis, NowFn, to_datetime};
 use mz_ore::result::ResultExt;
 use mz_ore::task::AbortOnDropHandle;
-use mz_ore::thread::JoinOnDropHandle;
 use mz_ore::tracing::OpenTelemetryContext;
 use mz_repr::{CatalogItemId, ColumnIndex, Row, SqlScalarType};
 use mz_sql::ast::{Raw, Statement};
@@ -46,6 +45,7 @@ use mz_sql_parser::parser::{ParserStatementError, StatementParseResult};
 use prometheus::Histogram;
 use serde_json::json;
 use tokio::sync::{mpsc, oneshot};
+use tokio::task::LocalSet;
 use tracing::{debug, error};
 use uuid::Uuid;
 
@@ -75,7 +75,7 @@ use crate::{AdapterNotice, AppendWebhookError, PeekClient, PeekResponseUnary, St
 pub struct Handle {
     pub(crate) session_id: Uuid,
     pub(crate) start_instant: Instant,
-    pub(crate) _thread: JoinOnDropHandle<()>,
+    pub(crate) local: LocalSet,
 }
 
 impl Handle {
@@ -92,6 +92,10 @@ impl Handle {
     pub fn start_instant(&self) -> Instant {
         self.start_instant
     }
+
+    pub async fn join(self) {
+        self.local.await;
+    }
 }
 
 /// A coordinator client.