Register local functions with async markers

pkg/compiler/lib/src/ssa/kernel_impact.dart

 // Copyright (c) 2016, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 import 'package:kernel/ast.dart' as ir;
 
 import '../common.dart';
 import '../common/names.dart';
 import '../compiler.dart';
 import '../constants/expressions.dart';
@@ skipping 119 matching lines @@
     return impactBuilder;
   }
 
   ResolutionImpact buildConstructor(ir.Constructor constructor) {
     handleSignature(constructor.function, checkReturnType: false);
     visitNodes(constructor.initializers);
     visitNode(constructor.function.body);
     return impactBuilder;
   }
 
-  ResolutionImpact buildProcedure(ir.Procedure procedure) {
-    handleSignature(procedure.function);
-    visitNode(procedure.function.body);
-    switch (procedure.function.asyncMarker) {
+  void handleAsyncMarker(ir.AsyncMarker asyncMarker) {
+    switch (asyncMarker) {
       case ir.AsyncMarker.Sync:
         break;
       case ir.AsyncMarker.SyncStar:
         impactBuilder.registerFeature(Feature.SYNC_STAR);
         break;
       case ir.AsyncMarker.Async:
         impactBuilder.registerFeature(Feature.ASYNC);
         break;
       case ir.AsyncMarker.AsyncStar:
         impactBuilder.registerFeature(Feature.ASYNC_STAR);
         break;
       case ir.AsyncMarker.SyncYielding:
         throw new SpannableAssertionFailure(CURRENT_ELEMENT_SPANNABLE,
-            "Unexpected async marker: ${procedure.function.asyncMarker}");
+            "Unexpected async marker: ${asyncMarker}");
     }
+  }
+
+  ResolutionImpact buildProcedure(ir.Procedure procedure) {
+    handleSignature(procedure.function);
+    visitNode(procedure.function.body);
+    handleAsyncMarker(procedure.function.asyncMarker);
     if (procedure.isExternal &&
         !elementAdapter.isForeignLibrary(procedure.enclosingLibrary)) {
       impactBuilder.registerNativeData(
           elementAdapter.getNativeBehaviorForMethod(procedure));
     }
     return impactBuilder;
   }
 
   void visitNode(ir.Node node) => node?.accept(this);
 
@@ skipping 296 matching lines @@
     impactBuilder.registerFeature(Feature.STRING_INTERPOLATION);
     impactBuilder.registerFeature(Feature.STRING_JUXTAPOSITION);
     visitNodes(node.expressions);
   }
 
   @override
   void visitFunctionDeclaration(ir.FunctionDeclaration node) {
     impactBuilder.registerStaticUse(
         new StaticUse.closure(elementAdapter.getLocalFunction(node)));
     handleSignature(node.function);
+    handleAsyncMarker(node.function.asyncMarker);
     visitNode(node.function.body);
   }
 
   @override
   void visitFunctionExpression(ir.FunctionExpression node) {
     impactBuilder.registerStaticUse(
         new StaticUse.closure(elementAdapter.getLocalFunction(node)));
     handleSignature(node.function);
+    handleAsyncMarker(node.function.asyncMarker);
     visitNode(node.function.body);
   }
 
   @override
   void visitVariableDeclaration(ir.VariableDeclaration node) {
     checkType(node.type);
     if (node.initializer != null) {
       visitNode(node.initializer);
     } else {
       impactBuilder.registerFeature(Feature.LOCAL_WITHOUT_INITIALIZER);
@@ skipping 80 matching lines @@
     ConstructorEntity target = elementAdapter.getConstructor(node.target);
     impactBuilder.registerStaticUse(new StaticUse.superConstructorInvoke(
         target, elementAdapter.getCallStructure(node.arguments)));
   }
 
   // TODO(johnniwinther): Make this throw and visit child nodes explicitly
   // instead to ensure that we don't visit unwanted parts of the ir.
   @override
   void defaultNode(ir.Node node) => node.visitChildren(this);
 }