.

build.mill

@@ -132,7 +132,7 @@ object benchmark extends Cross[BenchmarkModule](build.scalaVersions) with RunMod
     println(s"Waiting for server to start..., vt:$vt")
     if (!waitForServer(url)) {
       serverApp.destroy()
-      println("Failed to start regular server")
+      println("Failed to start server")
       sys.exit(1)
     }
 
@@ -144,7 +144,7 @@ object benchmark extends Cross[BenchmarkModule](build.scalaVersions) with RunMod
     ).call(stderr = os.Pipe)
     serverApp.destroy()
 
-    println(s"\nResults with $vt:")
+    println(s"""\nResults with ${if (vt) "(virtual threads)" else "(platform threads)"}:""")
     println(results.out.text())
   }
 

example/minimalApplicationWithLoom/package.mill

@@ -10,14 +10,12 @@ trait AppModule extends CrossScalaModule{
     argsStr.split(" ").filter(_.nonEmpty).toSeq
   }
 
-  def forkArgs = T {
+  def forkArgs = Task.Input {
     //TODO not sure why the env passing is not working
-    val envVirtualThread: String = T.env.getOrElse("CASK_VIRTUAL_THREAD", "null")
+    val envVirtualThread: String = T.env.getOrElse("CASK_VIRTUAL_THREAD", "false")
     println("envVirtualThread: " + envVirtualThread)
 
-    val systemProps = if (envVirtualThread == "true") {
-      Seq("-Dcask.virtual-thread.enabled=true")
-    } else Nil
+    val systemProps = Seq(s"-Dcask.virtual-thread.enabled=$envVirtualThread")
 
     val baseArgs = Seq(
       "--add-opens", "java.base/java.lang=ALL-UNNAMED"

example/staticFilesWithLoom/package.mill

@@ -15,14 +15,12 @@ trait AppModule extends CrossScalaModule{ app =>
     argsStr.split(" ").filter(_.nonEmpty).toSeq
   }
 
-  def forkArgs = T {
+  def forkArgs = Task.Input {
     //TODO not sure why the env passing is not working
-    val envVirtualThread: String = T.env.getOrElse("CASK_VIRTUAL_THREAD", "null")
+    val envVirtualThread: String = T.env.getOrElse("CASK_VIRTUAL_THREAD", "false")
     println("envVirtualThread: " + envVirtualThread)
 
-    val systemProps = if (envVirtualThread == "true") {
-      Seq("-Dcask.virtual-thread.enabled=true")
-    } else Nil
+    val systemProps = Seq(s"-Dcask.virtual-thread.enabled=$envVirtualThread")
 
     val baseArgs = Seq(
       "--add-opens", "java.base/java.lang=ALL-UNNAMED"

example/todoDbWithLoom/package.mill

@@ -9,14 +9,12 @@ trait AppModule extends CrossScalaModule{
     argsStr.split(" ").filter(_.nonEmpty).toSeq
   }
 
-  def forkArgs = T {
+  def forkArgs = Task.Input {
     //TODO not sure why the env passing is not working
-    val envVirtualThread: String = T.env.getOrElse("CASK_VIRTUAL_THREAD", "null")
+    val envVirtualThread: String = T.env.getOrElse("CASK_VIRTUAL_THREAD", "false")
     println("envVirtualThread: " + envVirtualThread)
 
-    val systemProps = if (envVirtualThread == "true") {
-      Seq("-Dcask.virtual-thread.enabled=true")
-    } else Nil
+    val systemProps = Seq(s"-Dcask.virtual-thread.enabled=$envVirtualThread")
 
     val baseArgs = Seq(
       "--add-opens", "java.base/java.lang=ALL-UNNAMED"