fix build

Author: sxlijin

engine/language_server/src/lib.rs

@@ -85,7 +85,7 @@ pub fn run_server() -> anyhow::Result<()> {
     Server::new(
         worker_threads,
         ServerArgs {
-            tokio_handle: tokio_runtime.handle().clone(),
+            tokio_runtime,
             broadcast_tx,
             playground_rx,
             playground_tx: playground_tx.clone(),

engine/language_server/src/server.rs

@@ -47,7 +47,7 @@ use crate::message::try_show_message;
 pub type Result<T> = std::result::Result<T, api::Error>;
 
 pub(crate) struct ServerArgs {
-    pub tokio_handle: tokio::runtime::Handle,
+    pub tokio_runtime: tokio::runtime::Runtime,
     pub broadcast_tx: broadcast::Sender<LangServerToWasmMessage>,
     pub playground_rx: broadcast::Receiver<PreLangServerToWasmMessage>,
     pub playground_tx: broadcast::Sender<PreLangServerToWasmMessage>,
@@ -195,7 +195,7 @@ impl Server {
         {
             let lsp_sender = server.connection.make_sender();
             let playground_tx = server.session.playground_tx.clone();
-            server.args.tokio_handle.spawn(async move {
+            server.args.tokio_runtime.spawn(async move {
                 lsp_sender
                     .send(Message::Notification(lsp_server::Notification::new(
                         "baml/port".to_string(),
@@ -211,7 +211,7 @@ impl Server {
             let mut playground_rx = server.args.playground_rx.resubscribe();
             let broadcast_tx = server.args.broadcast_tx.clone();
             let session = server.session.clone();
-            server.args.tokio_handle.spawn(async move {
+            server.args.tokio_runtime.spawn(async move {
                 tracing::info!("Starting playground rx loop");
                 while let Ok(msg) = playground_rx.recv().await {
                     tracing::info!("playground rx loop: {:?}", msg);

engine/language_server/src/tests.rs

@@ -4,8 +4,9 @@ use crossbeam_channel::{Receiver, Sender};
 use log::LevelFilter;
 use lsp_server::Message;
 use serde_json::json;
+use tokio::sync::broadcast;
 
-use crate::server::{connection::ConnectionInitializer, Server};
+use crate::server::{connection::ConnectionInitializer, Server, ServerArgs};
 
 pub struct TestServer {
     pub thread_join_handle: thread::JoinHandle<()>,
@@ -126,7 +127,23 @@ pub fn new_test_server(worker_threads: NonZeroUsize) -> anyhow::Result<TestServe
             )
             .unwrap();
 
-        let server = Server::new_with_connection(worker_threads, connection, init_params).unwrap();
+        let (playground_tx, playground_rx) = broadcast::channel(1);
+        let (broadcast_tx, _) = broadcast::channel(1);
+
+        let server = Server::new_with_connection(
+            worker_threads,
+            connection,
+            init_params,
+            ServerArgs {
+                tokio_runtime: tokio::runtime::Runtime::new().unwrap(),
+                broadcast_tx,
+                playground_rx,
+                playground_tx,
+                playground_port: 0,
+                proxy_port: 0,
+            },
+        )
+        .unwrap();
         server.run().unwrap();
     });