Added ensure-runtime-vthreads check on flag=target AOT emit

Author: fogus

src/main/clojure/clojure/core/async.clj

@@ -514,13 +514,10 @@ use vthreads in io-thread when available
   Returns a channel which will receive the result of the body when
   completed"
   [& body]
-  (if (or (dispatch/aot-vthreads?) (dispatch/runtime-vthreads?))
-    `(if (and (= dispatch/compiled-vthreads-flag "target") (not (dispatch/runtime-vthreads?)))
-       (throw (ex-info "Code compiled to target virtual threads, but is running on a JVM without vthread support."
-                       {:compiled-vthreads-flag dispatch/compiled-vthreads-flag
-                        :runtime-jvm-version (System/getProperty "java.version")}))
-       (thread-call (^:once fn* [] ~@body) :io))
-    ((find-var 'clojure.core.async.impl.go/go-impl) &env body)))
+  (cond (dispatch/aot-vthreads?) `(do (dispatch/ensure-runtime-vthreads!)
+                                      (thread-call (^:once fn* [] ~@body) :io))
+        (dispatch/runtime-vthreads?) `(thread-call (^:once fn* [] ~@body) :io)
+        :default ((find-var 'clojure.core.async.impl.go/go-impl) &env body)))
 
 (defonce ^:private thread-macro-executor nil)
 

src/main/clojure/clojure/core/async/impl/dispatch.clj

@@ -85,8 +85,6 @@
   []
   (System/getProperty "clojure.core.async.vthreads"))
 
-(def compiled-vthreads-flag (vthreads-directive))
-
 (defn aot-vthreads? []
   (and clojure.core/*compile-files*
        (= (vthreads-directive) "target")))
@@ -98,6 +96,11 @@
           (not= (vthreads-directive) "avoid")
           @virtual-threads-available?))))
 
+(defn ensure-runtime-vthreads! []
+  (when (not (runtime-vthreads?))
+    (throw (ex-info "Code compiled to target virtual threads, but is running on a JVM without vthread support."
+                    {:runtime-jvm-version (System/getProperty "java.version")}))))
+
 (defn- make-io-executor
   []
   (if (runtime-vthreads?)