Correcting method AsyncMode determination to use AsyncTask if that be…

…gan on the GeneratedMethod. Bumps to 3.1.1
JasperFx · Sep 25, 2023 · f0752fd · f0752fd
1 parent 8f9b866
commit f0752fd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Directory.Build.props b/Directory.Build.props
@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="utf-8"?>
 <Project>
   <PropertyGroup>
-    <VersionPrefix>3.1.0</VersionPrefix>
+    <VersionPrefix>3.1.1</VersionPrefix>
     <LangVersion>10.0</LangVersion>
     <Authors>Jeremy D. Miller;Babu Annamalai;Oskar Dudycz;Joona-Pekka Kokko</Authors>
     <TargetFrameworks>net6.0;net7.0</TargetFrameworks>

diff --git a/src/JasperFx.CodeGeneration/GeneratedMethod.cs b/src/JasperFx.CodeGeneration/GeneratedMethod.cs
@@ -206,6 +206,8 @@ public void ArrangeFrames(GeneratedType type, IServiceVariableSource? services)
 
         services?.StartNewMethod();
 
+        var startingAsyncMode = _asyncMode;
+
         var compiler = new MethodFrameArranger(this, type, services);
         try
         {
@@ -216,6 +218,12 @@ public void ArrangeFrames(GeneratedType type, IServiceVariableSource? services)
             e.Type = type;
             throw;
         }
+
+        // Correct the async mode even someone tried to override this
+        if (startingAsyncMode == AsyncMode.AsyncTask)
+        {
+            _asyncMode = startingAsyncMode;
+        }
     }
 
     public string ToExitStatement()