[SourceKit] Report number of instructions executed since SourceKit was started in statistics request

This can be used to measure how many instructions a request executes by retrieving the number of instructions executed since the process’s start before and after executing the request.

Author: ahoppen

test/SourceKit/Misc/stats.swift

@@ -2,6 +2,7 @@ func foo() {}
 
 // RUN: %sourcekitd-test -req=syntax-map %s == -req=stats | %FileCheck %s -check-prefix=SYNTAX_1
 
+// SYNTAX_1:   {{.*}} source.statistic.instruction-count
 // SYNTAX_1: 2 {{.*}} source.statistic.num-requests
 // SYNTAX_1: 0 {{.*}} source.statistic.num-semantic-requests
 // SYNTAX_1: 0 {{.*}} source.statistic.num-ast-builds

tools/SourceKit/tools/sourcekitd/lib/API/Requests.cpp

@@ -29,6 +29,7 @@
 
 #include "swift/Basic/ExponentialGrowthAppendingBinaryByteStream.h"
 #include "swift/Basic/Mangler.h"
+#include "swift/Basic/Statistic.h"
 #include "swift/Basic/Version.h"
 #include "swift/Demangling/Demangler.h"
 #include "swift/Demangling/ManglingMacros.h"
@@ -918,6 +919,11 @@ void handleRequestImpl(sourcekitd_object_t ReqObj, ResponseReceiver Rec) {
         dict.set(KeyValue, stat->value);
       };
 
+      Statistic instructionCount(
+          UIdentFromSKDUID(KindStatInstructionCount),
+          "# of instructions executed since the SourceKit process was started");
+      instructionCount.value.store(swift::getInstructionsExecuted());
+      addStat(&instructionCount);
       addStat(&numRequests);
       addStat(&numSemaRequests);
       std::for_each(stats.begin(), stats.end(), addStat);

utils/gyb_sourcekit_support/UIDs.py

@@ -176,7 +176,7 @@ def __init__(self, internal_name, external_name):
     KEY('ExpressionLength', 'key.expression_length'),
     KEY('ExpressionType', 'key.expression_type'),
     KEY('CanonicalizeType', 'key.canonicalize_type'),
-    KEY('InternalDiagnostic', "key.internal_diagnostic"),
+    KEY('InternalDiagnostic', 'key.internal_diagnostic'),
     KEY('VFSName', 'key.vfs.name'),
     KEY('VFSOptions', 'key.vfs.options'),
     KEY('Files', 'key.files'),
@@ -453,6 +453,7 @@ def __init__(self, internal_name, external_name):
     KIND('Unknown', 'source.syntacticrename.unknown'),
     KIND('StatNumRequests', 'source.statistic.num-requests'),
     KIND('StatNumSemaRequests', 'source.statistic.num-semantic-requests'),
+    KIND('StatInstructionCount', 'source.statistic.instruction-count'),
     KIND('SyntaxTreeOff', 'source.syntaxtree.transfer.off'),
     KIND('SyntaxTreeFull', 'source.syntaxtree.transfer.full'),
     KIND('Swift', 'source.lang.swift'),