Merge pull request #58807 from etcwilde/ewilde/async-let-playground

Fix async_let playground transform
This commit is contained in:
Evan Wilde
2022-05-11 15:40:09 -07:00
committed by GitHub
2 changed files with 60 additions and 6 deletions

View File

@@ -576,6 +576,7 @@ public:
} else if (auto *D = Element.dyn_cast<Decl *>()) {
D->walk(CF);
if (auto *PBD = dyn_cast<PatternBindingDecl>(D)) {
if (!PBD->isAsyncLet()) {
if (VarDecl *VD = PBD->getSingleVar()) {
if (VD->getParentInitializer()) {
Added<Stmt *> Log = logVarDecl(VD);
@@ -585,6 +586,7 @@ public:
}
}
}
}
} else {
transformDecl(D);
}

View File

@@ -0,0 +1,52 @@
// RUN: %empty-directory(%t)
// RUN: cp %s %t/main.swift
// RUN: %target-build-swift -whole-module-optimization -module-name PlaygroundSupport -emit-module-path %t/PlaygroundSupport.swiftmodule -parse-as-library -c -o %t/PlaygroundSupport.o %S/Inputs/SilentPCMacroRuntime.swift %S/Inputs/PlaygroundsRuntime.swift
// RUN: %target-build-swift -Xfrontend -disable-availability-checking -Xfrontend -playground -o %t/main -I=%t %t/PlaygroundSupport.o %t/main.swift
// RUN: %target-codesign %t/main
// RUN: %target-run %t/main | %FileCheck %s
// RUN: %target-build-swift -Xfrontend -disable-availability-checking -Xfrontend -pc-macro -Xfrontend -playground -o %t/main2 -I=%t %t/PlaygroundSupport.o %t/main.swift
// RUN: %target-codesign %t/main2
// RUN: %target-run %t/main2 | %FileCheck %s
// REQUIRES: executable_test
// REQUIRES: concurrency
// rdar://76038845
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime
import PlaygroundSupport
func factorial(_ x : Int) async -> Int {
if x < 1 {
return 1
}
return await x * factorial(x - 1)
}
async let fac4 = factorial(4)
print(await fac4)
// return await x * factorial(x - 1)
// CHECK: [22:{{[[:digit:]]+}}-22:{{[[:digit:]]+}}] __builtin_log[='1']
// return await x * factorial(x - 1)
// CHECK: [24:{{[[:digit:]]+}}-24:{{[[:digit:]]+}}] __builtin_log[='1']
// return await x * factorial(x - 1)
// CHECK: [24:{{[[:digit:]]+}}-24:{{[[:digit:]]+}}] __builtin_log[='2']
// return await x * factorial(x - 1)
// CHECK: [24:{{[[:digit:]]+}}-24:{{[[:digit:]]+}}] __builtin_log[='6']
// return await x * factorial(x - 1)
// CHECK: [24:{{[[:digit:]]+}}-24:{{[[:digit:]]+}}] __builtin_log[='24']
// func factorial(_ x : Int) { ... }
// CHECK-NEXT: [20:{{[[:digit:]]+}}-25:{{[[:digit:]]+}}] __builtin_log_scope_exit
// (actually print the thing)
// CHECK-NEXT: 24
// print(await fac4)
// CHECK-NEXT: [28:{{[[:digit:]]+}}-28:{{[[:digit:]]+}}] __builtin_postPrint