set number of threads to cpu count

dojoengine · Sep 17, 2024 · 5ffe50e · 5ffe50e
1 parent f60ef4f
commit 5ffe50e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bin/torii/src/main.rs b/bin/torii/src/main.rs
@@ -166,6 +166,10 @@ async fn main() -> anyhow::Result<()> {
     sqlx::query("PRAGMA journal_mode = WAL;").execute(&pool).await?;
     sqlx::query("PRAGMA synchronous = NORMAL;").execute(&pool).await?;
 
+    // Set the number of threads based on CPU count
+    let cpu_count = std::thread::available_parallelism().unwrap().get();
+    sqlx::query(&format!("PRAGMA threads = {};", cpu_count)).execute(&pool).await?;
+
     sqlx::migrate!("../../crates/torii/migrations").run(&pool).await?;
 
     let provider: Arc<_> = JsonRpcClient::new(HttpTransport::new(args.rpc)).into();