From 0ff160f93f14b4c04fd533d6d74bf14a86298b0b Mon Sep 17 00:00:00 2001
From: Sheetal Nandi <shkamat@microsoft.com>
Date: Tue, 5 Sep 2017 19:36:32 -0700
Subject: [PATCH] Add files to change set instead of delay reloading project on
 "change" command

---
 src/server/editorServices.ts | 17 +++++++++++------
 src/server/session.ts        |  2 +-
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/server/editorServices.ts b/src/server/editorServices.ts
index 101f9fbb720..4b761288290 100644
--- a/src/server/editorServices.ts
+++ b/src/server/editorServices.ts
@@ -1948,12 +1948,7 @@ namespace ts.server {
                         const change = file.changes[i];
                         scriptInfo.editContent(change.span.start, change.span.start + change.span.length, change.newText);
                     }
-                    if (!this.changedFiles) {
-                        this.changedFiles = [scriptInfo];
-                    }
-                    else if (!contains(this.changedFiles, scriptInfo)) {
-                        this.changedFiles.push(scriptInfo);
-                    }
+                    this.addChangedFile(scriptInfo);
                 }
             }
 
@@ -1969,6 +1964,16 @@ namespace ts.server {
             }
         }
 
+        /* @internal */
+        addChangedFile(scriptInfo: ScriptInfo) {
+            if (!this.changedFiles) {
+                this.changedFiles = [scriptInfo];
+            }
+            else if (!contains(this.changedFiles, scriptInfo)) {
+                this.changedFiles.push(scriptInfo);
+            }
+        }
+
         private closeConfiguredProject(configFile: NormalizedPath): void {
             const configuredProject = this.findConfiguredProjectByProjectName(configFile);
             if (configuredProject && configuredProject.deleteOpenRef() === 0) {
diff --git a/src/server/session.ts b/src/server/session.ts
index b4f2b46f99c..4b307d27970 100644
--- a/src/server/session.ts
+++ b/src/server/session.ts
@@ -1287,9 +1287,9 @@ namespace ts.server {
                 const end = scriptInfo.lineOffsetToPosition(args.endLine, args.endOffset);
                 if (start >= 0) {
                     scriptInfo.editContent(start, end, args.insertString);
+                    this.projectService.addChangedFile(scriptInfo);
                     this.changeSeq++;
                 }
-                this.projectService.delayUpdateProjectGraphAndInferredProjectsRefresh(project);
             }
         }