From 91a15dc6095227c4a01ce984f4d33e7cdb86efde Mon Sep 17 00:00:00 2001
From: Zen <843968788@qq.com>
Date: Wed, 9 May 2018 22:42:44 +0800
Subject: [PATCH] improve performance

---
 src/services/documentHighlights.ts                 | 2 +-
 tests/cases/fourslash/getOccurrencesAsyncAwait.ts  | 5 +++++
 tests/cases/fourslash/getOccurrencesAsyncAwait3.ts | 2 --
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/services/documentHighlights.ts b/src/services/documentHighlights.ts
index 3c898551ac9..65653483d07 100644
--- a/src/services/documentHighlights.ts
+++ b/src/services/documentHighlights.ts
@@ -395,7 +395,7 @@ namespace ts.DocumentHighlights {
                 pushKeywordIf(keywords, node.getFirstToken(), SyntaxKind.AwaitKeyword);
             }
             // Do not cross function boundaries.
-            if (!isFunctionLike(node)) {
+            if (!isFunctionLike(node) && !isClassLike(node) && !isInterfaceDeclaration(node) && !isModuleDeclaration(node) && !isTypeAliasDeclaration(node) && !isTypeNode(node)) {
                 forEachChild(node, aggregate);
             }
         }
diff --git a/tests/cases/fourslash/getOccurrencesAsyncAwait.ts b/tests/cases/fourslash/getOccurrencesAsyncAwait.ts
index f2224255d14..f337e755096 100644
--- a/tests/cases/fourslash/getOccurrencesAsyncAwait.ts
+++ b/tests/cases/fourslash/getOccurrencesAsyncAwait.ts
@@ -3,6 +3,11 @@
 ////[|async|] function f() {
 //// [|await|] 100;
 //// [|a/**/wait|] [|await|] 200;
+////class Foo {
+////    async memberFunction() {
+////        await 1;
+////    }
+////}
 //// return [|await|] async function () {
 ////   await 300;
 //// }
diff --git a/tests/cases/fourslash/getOccurrencesAsyncAwait3.ts b/tests/cases/fourslash/getOccurrencesAsyncAwait3.ts
index 67d56167b82..972b6b8b07d 100644
--- a/tests/cases/fourslash/getOccurrencesAsyncAwait3.ts
+++ b/tests/cases/fourslash/getOccurrencesAsyncAwait3.ts
@@ -7,7 +7,5 @@
 ////    await 300;
 ////}
 
-verify.rangesAreOccurrences(false);
-
 goTo.marker();
 verify.occurrencesAtPositionCount(0);
\ No newline at end of file