[AST/Sema] Allow @execution(...) attribute to be used on initializers and accessors

Author: xedin

include/swift/AST/DeclAttr.def

@@ -862,7 +862,7 @@ DECL_ATTR(abi, ABI,
 DECL_ATTR_FEATURE_REQUIREMENT(ABI, ABIAttribute)
 
 DECL_ATTR(execution, Execution,
-  OnFunc,
+  OnAbstractFunction,
   ABIBreakingToAdd | ABIBreakingToRemove | APIBreakingToAdd | APIBreakingToRemove,
   166)
 DECL_ATTR_FEATURE_REQUIREMENT(Execution, ExecutionAttribute)

lib/Sema/TypeCheckAttr.cpp

@@ -257,7 +257,7 @@ class AttributeChecker : public AttributeVisitor<AttributeChecker> {
 
 public:
   void visitExecutionAttr(ExecutionAttr *attr) {
-    auto *F = dyn_cast<FuncDecl>(D);
+    auto *F = dyn_cast<AbstractFunctionDecl>(D);
     if (!F)
       return;
 

test/IDE/complete_decl_attribute_feature_requirement.swift

@@ -63,7 +63,7 @@ struct _S {
   @#^ON_INIT^# init()
 // ON_INIT:              Begin completions
 // ON_INIT_ENABLED-DAG:  Keyword/None:              abi[#Constructor Attribute#]; name=abi
-// ON_INIT_ENABLED-NOT:  Keyword/None:              execution[#{{.*}} Attribute#]; name=execution
+// ON_INIT_ENABLED-DAG:  Keyword/None:              execution[#{{.*}} Attribute#]; name=execution
 // ON_INIT_DISABLED-NOT: Keyword/None:              abi[#{{.*}} Attribute#]; name=abi
 // ON_INIT_DISABLED-NOT: Keyword/None:              execution[#{{.*}} Attribute#]; name=execution
 // ON_INIT:              End completions

test/attr/attr_execution.swift

@@ -18,22 +18,22 @@
 
 struct Test {
   @execution(concurrent) init() {}
-  // expected-error@-1 {{@execution(concurrent) may only be used on 'func' declarations}}
+  // expected-error@-1 {{cannot use '@execution' on non-async initializer 'init()'}}
 
   @execution(concurrent) func member() {}
   // expected-error@-1 {{cannot use '@execution' on non-async instance method 'member()'}}
 
   @execution(concurrent) func member() async {} // Ok
 
-  // expected-error@+1 {{@execution(caller) may only be used on 'func' declarations}}
+  // expected-error@+1 {{'@execution(caller)' attribute cannot be applied to this declaration}}
   @execution(caller) subscript(a: Int) -> Bool {
     get { false }
     @execution(concurrent) set { }
-    // expected-error@-1 {{@execution(concurrent) may only be used on 'func' declarations}}
+    // expected-error@-1 {{cannot use '@execution' on non-async setter for subscript 'subscript(_:)'}}
   }
 
   @execution(caller) var x: Int
-  // expected-error@-1 {{@execution(caller) may only be used on 'func' declarations}}
+  // expected-error@-1 {{'@execution(caller)' attribute cannot be applied to this declaration}}
 }
 
 do {
@@ -93,3 +93,21 @@ _ = { @execution(concurrent) () -> Int in
 _ = { @execution(caller) (x: isolated (any Actor)?) in
   // expected-error@-1 {{cannot use '@execution' together with an isolated parameter}}
 }
+
+struct TestDifferentPositions {
+  @execution(caller)
+  init() async {
+  }
+
+  var x: Int {
+    @execution(concurrent)
+    get async {
+    }
+  }
+
+  subscript(x: Int) -> Bool {
+    @execution(concurrent)
+    get async {
+    }
+  }
+}