Fix result expression for grouped statements

sqldelight-compiler/src/main/kotlin/app/cash/sqldelight/core/compiler/QueryGenerator.kt

@@ -272,6 +272,13 @@ abstract class QueryGenerator(
       }
     }
 
+    // Extract value from the result of a grouped statement in async,
+    // because the transaction is put in an QueryResult.AsyncValue block.
+    if (generateAsync && isNamedQuery && query.statement is SqlDelightStmtClojureStmtList) {
+      binder += "%L"
+      arguments.add(".await()")
+    }
+
     val statementId = if (needsFreshStatement) CodeBlock.of("null") else CodeBlock.of("%L", id)
 
     if (isNamedQuery) {

sqldelight-compiler/src/test/kotlin/app/cash/sqldelight/core/queries/async/AsyncSelectQueryTypeTest.kt

@@ -251,7 +251,7 @@ class AsyncSelectQueryTypeTest {
       |          |SELECT value
       |          |  FROM data
       |          |  WHERE id = last_insert_rowid()
-      |          ""${'"'}.trimMargin(), mapper, 0)
+      |          ""${'"'}.trimMargin(), mapper, 0).await()
       |    }
       |  }
       |

sqldelight-gradle-plugin/src/test/integration-mysql-async/src/main/sqldelight/app/cash/sqldelight/mysql/integration/async/Dog.sq

@@ -9,6 +9,13 @@ insertDog:
 INSERT INTO dog
 VALUES (?, ?, DEFAULT, ?);
 
+insertDogAndReturnAll {
+INSERT INTO dog
+VALUES (?, ?, DEFAULT, ?);
+
+SELECT * FROM dog;
+}
+
 selectDogs:
 SELECT *
 FROM dog;

sqldelight-gradle-plugin/src/test/integration-mysql-async/src/test/kotlin/app/cash/sqldelight/mysql/integration/async/MySqlTest.kt

@@ -51,6 +51,20 @@ class MySqlTest {
       )
   }
 
+  @Test
+  fun insertAndReturn() = runTest { database ->
+    val all = database.dogQueries.insertDogAndReturnAll("Tilda", "Pomeranian", null)
+    assertThat(all.awaitAsList())
+      .containsExactly(
+        Dog(
+          name = "Tilda",
+          breed = "Pomeranian",
+          is_good = true,
+          age = null,
+        ),
+      )
+  }
+
   @Test
   fun simpleSelectWithIn() = runTest { database ->
     with(database) {