尝试以扩展 VS Code 中的代理模式!

语法高亮优化

2017 年 2 月 8 日 - Alexandru Dima

Visual Studio Code 1.9 版本包含我们一直致力于的酷炫性能改进,我想讲述它的故事。

TL;DR TextMate 主题在 VS Code 1.9 中将更符合作者的意图,同时渲染速度更快,内存消耗更少。


语法高亮

语法高亮通常包括两个阶段。将标记分配给源代码,然后主题会针对这些标记,分配颜色,然后,您的源代码就以颜色呈现了。这是将文本编辑器转变为代码编辑器的功能之一。

VS Code(以及 Monaco Editor)中的标记化逐行进行,从上到下,一次性完成。标记器可以在标记化行的末尾存储一些状态,该状态将在标记化下一行时传回。这是许多标记化引擎(包括 TextMate 语法)使用的一种技术,它允许编辑器在用户进行编辑时仅重新标记行的一个小子集。

大多数情况下,在某行上键入只会导致该行被重新标记,因为标记器返回相同的结束状态,并且编辑器可以假定后续行不会获得新标记

Tokenization Single Line

更罕见的是,在某行上键入会导致当前行和下面一些行被重新标记/重绘(直到遇到相同的结束状态)

Tokenization Multiple Lines

我们过去如何表示标记

VS Code 中的编辑器代码在 VS Code 出现很久之前就已编写。它以 Monaco Editor 的形式在各种 Microsoft 项目中发布,包括 Internet Explorer 的 F12 工具。我们的一项要求是减少内存使用。

过去,我们手动编写标记器(即使在今天,也没有可行的方法在浏览器中解释 TextMate 语法,但这又是另一个故事)。对于下面的行,我们将从我们手写的标记器中获得以下标记

Line offsets
tokens = [
  { startIndex: 0, type: 'keyword.js' },
  { startIndex: 8, type: '' },
  { startIndex: 9, type: 'identifier.js' },
  { startIndex: 11, type: 'delimiter.paren.js' },
  { startIndex: 12, type: 'delimiter.paren.js' },
  { startIndex: 13, type: '' },
  { startIndex: 14, type: 'delimiter.curly.js' }
];

在 Chrome 中保存该标记数组需要 648 字节,因此存储这样一个对象在内存方面相当昂贵(每个对象实例必须为指向其原型、属性列表等保留空间)。我们当前的机器确实有很多 RAM,但为 15 个字符的行存储 648 字节是不可接受的。

因此,当时我们提出了一种二进制格式来存储标记,该格式一直使用到 VS Code 1.8(包括 1.8)。鉴于会存在重复的标记类型,我们将它们收集在一个单独的映射中(每个文件),如下所示

//     0        1               2                  3                      4
map = ['', 'keyword.js', 'identifier.js', 'delimiter.paren.js', 'delimiter.curly.js'];
tokens = [
  { startIndex: 0, type: 1 },
  { startIndex: 8, type: 0 },
  { startIndex: 9, type: 2 },
  { startIndex: 11, type: 3 },
  { startIndex: 12, type: 3 },
  { startIndex: 13, type: 0 },
  { startIndex: 14, type: 4 }
];

然后,我们将 `startIndex`(32 位)和 `type`(16 位)编码为 JavaScript 数字的 53 位尾数中的 48 位。我们的标记数组最终将如下所示,并且映射数组将用于整个文件

tokens = [
  //       type                 startIndex
  4294967296, // 0000000000000001 00000000000000000000000000000000
  8, // 0000000000000000 00000000000000000000000000001000
  8589934601, // 0000000000000010 00000000000000000000000000001001
  12884901899, // 0000000000000011 00000000000000000000000000001011
  12884901900, // 0000000000000011 00000000000000000000000000001100
  13, // 0000000000000000 00000000000000000000000000001101
  17179869198 // 0000000000000100 00000000000000000000000000001110
];

在 Chrome 中保存此标记数组需要 104 字节。元素本身应该只占用 56 字节(7 x 64 位数字),其余部分可能由 v8 存储数组的其他元数据,或者可能以 2 的幂次分配后备存储来解释。然而,内存节省是显而易见的,并且每行标记越多,节省效果越好。我们对这种方法很满意,并且从那时起就一直使用这种表示形式。

注意:可能存在更紧凑的存储标记的方式,但将其存储为可二进制搜索的线性格式,在内存使用和访问性能方面提供了最佳的权衡。


标记 <-> 主题匹配

我们认为遵循浏览器最佳实践是个好主意,例如将样式留给 CSS,因此在渲染上述行时,我们将使用 `map` 解码二进制标记,然后使用标记类型进行渲染,如下所示

  <span class="token keyword js">function</span>
  <span class="token">&nbsp;</span>
  <span class="token identifier js">f1</span>
  <span class="token delimiter paren js">(</span>
  <span class="token delimiter paren js">)</span>
  <span class="token">&nbsp;</span>
  <span class="token delimiter curly js">{</span>

我们会在 CSS 中编写我们的主题(例如 Visual Studio 主题)

...
.monaco-editor.vs .token.delimiter          { color: #000000; }
.monaco-editor.vs .token.keyword            { color: #0000FF; }
.monaco-editor.vs .token.keyword.flow       { color: #AF00DB; }
...

结果相当不错,我们可以在某个地方切换类名,并立即将新主题应用于编辑器。


TextMate 语法

在 VS Code 发布时,我们有大约 10 个手写标记器,主要用于 Web 语言,这对于通用桌面代码编辑器来说肯定是不够的。引入了 TextMate 语法,这是一种描述性的指定标记化规则的形式,已在众多编辑器中采用。但有一个问题,TextMate 语法与我们手写的标记器的工作方式不太一样。

TextMate 语法通过使用 begin/end 状态或 while 状态,可以推送跨多个标记的范围。以下是在 JavaScript TextMate 语法下的相同示例(为简洁起见,忽略空白)

TextMate Scopes


VS Code 1.8 中的 TextMate 语法

如果我们要对作用域堆栈进行切片,每个标记基本上都会得到一个作用域名称数组,并且我们将从标记器中得到以下内容

Line offsets
tokens = [
  { startIndex: 0, scopes: ['source.js', 'meta.function.js', 'storage.type.function.js'] },
  { startIndex: 8, scopes: ['source.js', 'meta.function.js'] },
  {
    startIndex: 9,
    scopes: [
      'source.js',
      'meta.function.js',
      'meta.definition.function.js',
      'entity.name.function.js'
    ]
  },
  {
    startIndex: 11,
    scopes: [
      'source.js',
      'meta.function.js',
      'meta.parameters.js',
      'punctuation.definition.parameters.js'
    ]
  },
  { startIndex: 13, scopes: ['source.js', 'meta.function.js'] },
  {
    startIndex: 14,
    scopes: [
      'source.js',
      'meta.function.js',
      'meta.block.js',
      'punctuation.definition.block.js'
    ]
  }
];

所有的标记类型都是字符串,我们的代码还没有准备好处理字符串数组,更不用说对标记的二进制编码的影响了。因此,我们采用以下策略将作用域数组“近似”*为一个字符串

  • 忽略最不具体的范围(即 `source.js`);它很少增加任何价值。
  • 将每个剩余的范围在 `"."` 处分割。
  • 去除重复的唯一部分。
  • 使用稳定的排序函数对剩余部分进行排序(不一定是按字母顺序排序)。
  • 将各部分用 `"."` 连接起来。
tokens = [
  { startIndex: 0, type: 'meta.function.js.storage.type' },
  { startIndex: 9, type: 'meta.function.js' },
  { startIndex: 9, type: 'meta.function.js.definition.entity.name' },
  { startIndex: 11, type: 'meta.function.js.definition.parameters.punctuation' },
  { startIndex: 13, type: 'meta.function.js' },
  { startIndex: 14, type: 'meta.function.js.definition.punctuation.block' }
];

*: 我们当时的做法是完全错误的,“近似”是一个非常客气的说法:).

这些标记随后会“适应”并遵循与手动编写的标记器相同的代码路径(进行二进制编码),然后也会以相同的方式渲染

<span class="token meta function js storage type">function</span>
<span class="token meta function js">&nbsp;</span>
<span class="token meta function js definition entity name">f1</span>
<span class="token meta function js definition parameters punctuation">()</span>
<span class="token meta function js">&nbsp;</span>
<span class="token meta function js definition punctuation block">{</span>

TextMate 主题

TextMate 主题使用范围选择器来选择具有特定范围的标记,并向它们应用主题信息,例如颜色、粗细等。

给定具有以下范围的令牌

//            C                     B                             A
scopes = ['source.js', 'meta.definition.function.js', 'entity.name.function.js'];

以下是一些将匹配的简单选择器,按其排名(降序)排序

选择器 C B A
source source.js meta.definition.function.js entity.name.function.js
source.js source.js meta.definition.function.js entity.name.function.js
meta source.js meta.definition.function.js entity.name.function.js
meta.definition source.js meta.definition.function.js entity.name.function.js
meta.definition.function source.js meta.definition.function.js entity.name.function.js
entity source.js meta.definition.function.js entity.name.function.js
entity.name source.js meta.definition.function.js entity.name.function.js
entity.name.function source.js meta.definition.function.js entity.name.function.js
entity.name.function.js source.js meta.definition.function.js entity.name.function.js

观察:`entity` 优于 `meta.definition.function`,因为它匹配更具体的范围(分别是 `A` 而非 `B`)。

观察:`entity.name` 胜过 `entity`,因为它们都匹配相同的范围(`A`),但 `entity.name` 比 `entity` 更具体。

父级选择器

为了使事情更复杂一点,TextMate 主题还支持父级选择器。以下是使用简单选择器和父级选择器的一些示例(同样按其排名降序排列)

选择器 C B A
meta source.js meta.definition.function.js entity.name.function.js
source meta source.js meta.definition.function.js entity.name.function.js
source.js meta source.js meta.definition.function.js entity.name.function.js
meta.definition source.js meta.definition.function.js entity.name.function.js
source meta.definition source.js meta.definition.function.js entity.name.function.js
entity source.js meta.definition.function.js entity.name.function.js
source entity source.js meta.definition.function.js entity.name.function.js
meta.definition entity source.js meta.definition.function.js entity.name.function.js
entity.name source.js meta.definition.function.js entity.name.function.js
source entity.name source.js meta.definition.function.js entity.name.function.js

观察:`source entity` 胜过 `entity`,因为它们都匹配相同的范围(`A`),但 `source entity` 也匹配父级范围(`C`)。

观察:`entity.name` 胜过 `source entity`,因为它们都匹配相同的范围(`A`),但 `entity.name` 比 `entity` 更具体。

注意:还有第三种选择器,涉及排除范围,我们在此不讨论。我们没有增加对这种选择器的支持,并且我们注意到它在实际中很少使用。


VS Code 1.8 中的 TextMate 主题

以下是两个 Monokai 主题规则(为简洁起见,此处为 JSON 格式;原文件为 XML 格式)

...
// Function name
{ "scope": "entity.name.function", "fontStyle": "", "foreground":"#A6E22E" }
...
// Class name
{ "scope": "entity.name.class", "fontStyle": "underline", "foreground":"#A6E22E" }
...

在 VS Code 1.8 中,为了匹配我们的“近似”范围,我们会生成以下动态 CSS 规则

...
/* Function name */
.entity.name.function { color: #A6E22E; }
...
/* Class name */
.entity.name.class { color: #A6E22E; text-decoration: underline; }
...

然后,我们将其留给 CSS 来将“近似”范围与“近似”规则进行匹配。但是 CSS 匹配规则与 TextMate 选择器匹配规则不同,尤其是在排名方面。CSS 排名基于匹配的类名数量,而 TextMate 选择器排名对范围特异性有明确的规则。

这就是为什么 VS Code 中的 TextMate 主题看起来还行,但从未完全符合作者的意图。有时,差异很小,但有时这些差异会完全改变主题的感觉。


一些时机成熟的事件

随着时间的推移,我们已经逐步淘汰了手写标记器(最后一个,用于 HTML,仅在几个月前)。因此,在今天的 VS Code 中,所有文件都使用 TextMate 语法进行标记化。对于 Monaco Editor,我们已经迁移到使用 Monarch(一种描述性标记化引擎,其核心与 TextMate 语法相似,但表达力更强,可以在浏览器中运行)来处理大多数支持的语言,并且我们还为手动标记器添加了一个包装器。总而言之,这意味着支持一种新的标记化格式只需要更改 3 个标记提供程序(TextMate、Monarch 和手动包装器),而不是 10 多个。

几个月前,我们审查了 VS Code 核心中读取令牌类型的所有代码,我们注意到这些消费者只关心字符串、正则表达式或注释。例如,括号匹配逻辑会忽略包含范围为`"string"`、`"comment"` 或 `"regex"` 的令牌。

最近,我们得到了内部合作伙伴(Microsoft 内部使用 Monaco Editor 的其他团队)的同意,他们不再需要在 Monaco Editor 中支持 IE9 和 IE10。

也许最重要的是,编辑器的头号最受好评的功能是迷你图支持。为了在合理的时间内渲染迷你图,我们不能使用 DOM 节点和 CSS 匹配。我们可能会使用画布,并且我们需要在 JavaScript 中知道每个标记的颜色,这样我们就可以用正确的颜色绘制这些微小的字母。

也许我们最大的突破是,我们**不需要存储标记,也不需要存储它们的范围**,因为标记只在主题匹配它们或括号匹配跳过字符串方面产生效果。

最后,VS Code 1.9 有哪些新功能

表示 TextMate 主题

一个非常简单的主题可能如下所示

theme = [
  {                                  "foreground": "#F8F8F2"                           },
  { "scope": "var",                  "foreground": "#F8F8F2"                           },
  { "scope": "var.identifier",       "foreground": "#00FF00", "fontStyle": "bold"      },
  { "scope": "meta var.identifier",  "foreground": "#0000FF"                           },
  { "scope": "constant",             "foreground": "#100000", "fontStyle": "italic"    },
  { "scope": "constant.numeric",     "foreground": "#200000"                           },
  { "scope": "constant.numeric.hex",                          "fontStyle": "bold"      },
  { "scope": "constant.numeric.oct",                          "fontStyle": "underline" },
  { "scope": "constant.numeric.dec", "foreground": "#300000"                           },
];

加载时,我们将为主题中出现的每个唯一颜色生成一个 ID,并将其存储到颜色映射中(类似于我们之前对标记类型所做的)

//                          1          2          3          4          5           6
colorMap = ["reserved", "#F8F8F2", "#00FF00", "#0000FF", "#100000", "#200000", "#300000"]
theme = [
  {                                  "foreground": 1                           },
  { "scope": "var",                  "foreground": 1,                          },
  { "scope": "var.identifier",       "foreground": 2, "fontStyle": "bold"      },
  { "scope": "meta var.identifier",  "foreground": 3                           },
  { "scope": "constant",             "foreground": 4, "fontStyle": "italic"    },
  { "scope": "constant.numeric",     "foreground": 5                           },
  { "scope": "constant.numeric.hex",                  "fontStyle": "bold"      },
  { "scope": "constant.numeric.oct",                  "fontStyle": "underline" },
  { "scope": "constant.numeric.dec", "foreground": 6                           },
];

然后,我们将从主题规则中生成一个 Trie 数据结构,其中每个节点都包含已解析的主题选项

Theme Trie

观察:`constant.numeric.hex` 和 `constant.numeric.oct` 的节点包含将前景色更改为 `5` 的指令,因为它们*继承*了 `constant.numeric` 的此指令。

观察:`var.identifier` 节点包含额外的父规则 `meta var.identifier`,并将相应地回答查询。

当我们需要知道一个范围应该如何主题化时,我们可以查询这个 Trie。

例如

查询 结果
constant 将前景色设置为4,字体样式设置为斜体
constant.numeric 将前景色设置为5,字体样式设置为斜体
constant.numeric.hex 将前景色设置为5,字体样式设置为粗体
var 将前景色设置为1
var.baz 将前景色设置为1(匹配var
baz 什么都不做(不匹配)
var.identifier 如果存在父作用域meta,则将前景色设置为3,字体样式设置为粗体
否则,将前景色设置为2,字体样式设置为粗体

标记化的变化

VS Code 中使用的所有 TextMate 标记化代码都位于一个单独的项目中,vscode-textmate,它可以独立于 VS Code 使用。我们更改了 `vscode-textmate` 中表示范围堆栈的方式,使其成为一个不可变链表,该链表还存储了完全解析的 `metadata`。

当将新作用域推送到作用域堆栈时,我们将在主题 trie 中查找新作用域。然后,我们可以根据从作用域堆栈继承的内容以及主题 trie 返回的内容,立即计算出作用域列表的完全解析的所需前景色或字体样式。

一些示例

作用域堆栈 元数据
["source.js"] 前景色为1,字体样式为常规(无范围选择器的默认规则)
["source.js","constant"] 前景色为4,字体样式为斜体
["source.js","constant","baz"] 前景色为4,字体样式为斜体
["source.js","var.identifier"] 前景色为2,字体样式为粗体
["source.js","meta","var.identifier"] 前景色为3,字体样式为粗体

从作用域堆栈中弹出时,无需计算任何东西,因为我们可以直接使用存储在先前作用域列表元素中的元数据。

这是代表作用域列表中元素的 TypeScript 类

export class ScopeListElement {
    public readonly parent: ScopeListElement;
    public readonly scope: string;
    public readonly metadata: number;
    ...
}

我们存储 32 位元数据

/**
 * - -------------------------------------------
 *     3322 2222 2222 1111 1111 1100 0000 0000
 *     1098 7654 3210 9876 5432 1098 7654 3210
 * - -------------------------------------------
 *     xxxx xxxx xxxx xxxx xxxx xxxx xxxx xxxx
 *     bbbb bbbb bfff ffff ffFF FTTT LLLL LLLL
 * - -------------------------------------------
 *  - L = LanguageId (8 bits)
 *  - T = StandardTokenType (3 bits)
 *  - F = FontStyle (3 bits)
 *  - f = foreground color (9 bits)
 *  - b = background color (9 bits)
 */

最后,不再从标记化引擎中将标记作为对象发出

// These are generated using the Monokai theme.
tokens_before = [
  { startIndex: 0, scopes: ['source.js', 'meta.function.js', 'storage.type.function.js'] },
  { startIndex: 8, scopes: ['source.js', 'meta.function.js'] },
  {
    startIndex: 9,
    scopes: [
      'source.js',
      'meta.function.js',
      'meta.definition.function.js',
      'entity.name.function.js'
    ]
  },
  {
    startIndex: 11,
    scopes: [
      'source.js',
      'meta.function.js',
      'meta.parameters.js',
      'punctuation.definition.parameters.js'
    ]
  },
  { startIndex: 13, scopes: ['source.js', 'meta.function.js'] },
  {
    startIndex: 14,
    scopes: [
      'source.js',
      'meta.function.js',
      'meta.block.js',
      'punctuation.definition.block.js'
    ]
  }
];

// Every even index is the token start index, every odd index is the token metadata.
// We get fewer tokens because tokens with the same metadata get collapsed
tokens_now = [
  // bbbbbbbbb fffffffff FFF TTT LLLLLLLL
  0,
  16926743, // 000000010 000001001 001 000 00010111
  8,
  16793623, // 000000010 000000001 000 000 00010111
  9,
  16859159, // 000000010 000000101 000 000 00010111
  11,
  16793623 // 000000010 000000001 000 000 00010111
];

它们以以下方式渲染

<span class="mtk9 mtki">function</span>
<span class="mtk1">&nbsp;</span>
<span class="mtk5">f1</span>
<span class="mtk1">()&nbsp;{</span>

TextMate Scopes

标记直接从标记器作为 Uint32Array 返回。我们保留后备 ArrayBuffer,对于上面的示例,它在 Chrome 中占用 96 字节。元素本身应该只占用 32 字节(8 x 32 位数字),但我们可能再次观察到一些 v8 元数据开销。

一些数字

为了获得以下测量结果,我选择了三个具有不同特性和不同语法的文档

文件名 文件大小 行数 语言 观察
checker.ts 1.18 MB 22,253 TypeScript TypeScript 编译器中使用的实际源文件
bootstrap.min.css 118.36 KB 12 CSS 压缩的 CSS 文件
sqlite3.c 6.73 MB 200,904 C SQLite 的连接分发文件

我在一台性能不错的 Windows 桌面机器上运行了测试(使用 Electron 32 位)。

我不得不对源代码进行一些更改才能进行同类比较,例如确保在两个 VS Code 版本中使用完全相同的语法,在两个版本中关闭丰富的语言功能,或取消 VS Code 1.8 中已不再存在的 100 堆栈深度限制等。我还必须将 bootstrap.min.css 分割成多行,以使每行少于 20k 字符。

标记化时间

标记化在 UI 线程上以让步方式运行,因此我不得不添加一些代码强制它同步运行以测量以下时间(显示 10 次运行的中位数)

文件名 文件大小 VS Code 1.8 VS Code 1.9 加速
checker.ts 1.18 MB 4606.80 毫秒 3939.00 毫秒 14.50%
bootstrap.min.css 118.36 KB 776.76 毫秒 416.28 毫秒 46.41%
sqlite3.c 6.73 MB 16010.42 毫秒 10964.42 毫秒 31.52%
Tokenization times

尽管标记化现在也进行主题匹配,但时间节省可以通过对每一行进行单次遍历来解释。而之前,会有一个标记化遍历,一个将范围“近似”为字符串的二次遍历,以及一个二进制编码标记的三次遍历,现在标记直接以二进制编码方式从 TextMate 标记化引擎生成。需要垃圾回收的生成对象数量也大幅减少。

内存使用

折叠消耗了大量内存,特别是对于大型文件(这是另一个时间的优化),因此我在关闭折叠的情况下收集了以下堆快照数字。这显示了模型持有的内存,不包括原始文件字符串

文件名 文件大小 VS Code 1.8 VS Code 1.9 内存节省
checker.ts 1.18 MB 3.37 MB 2.61 MB 22.60%
bootstrap.min.css 118.36 KB 267.00 KB 201.33 KB 24.60%
sqlite3.c 6.73 MB 27.49 MB 21.22 MB 22.83%
Memory usage

内存使用减少可以通过不再持有令牌映射、相同元数据的连续令牌的折叠以及使用 `ArrayBuffer` 作为后端存储来解释。我们可以通过始终将仅包含空格的令牌折叠到前一个令牌中来进一步改进,因为空格的颜色无关紧要(空格是不可见的)。

新的 TextMate 作用域检查器小部件

我们添加了一个新的小部件,以帮助编写和调试主题或语法:您可以在**命令面板**中通过**开发人员:检查编辑器标记和作用域**运行它(⇧⌘P(Windows、Linux Ctrl+Shift+P)。

TextMate scope inspector

验证更改

对编辑器此组件的更改带来了一些严重的风险,因为我们方法中的任何错误(在新 trie 创建代码中、在新的二进制编码格式中等)都可能导致巨大的用户可见差异。

在 VS Code 中,我们有一个集成套件,可以断言我们发布的所有编程语言在五种主题(Light、Light+、Dark、Dark+、High Contrast)中的颜色。这些测试在更改我们的主题或更新特定语法时都非常有帮助。73 个集成测试中的每一个都包含一个固定文件(例如 test.c)以及五种主题的预期颜色(test_c.json),并且它们在我们 CI 构建的每次提交上运行。

为了验证标记化更改,我们使用旧的基于 CSS 的方法,从这些测试中收集了我们发布的全部 14 个主题(而不仅仅是我们编写的五个主题)的颜色化结果。然后,在每次更改后,我们使用新的基于 trie 的逻辑运行相同的测试,并使用自定义构建的可视化差异(和补丁)工具,检查每一个颜色差异,并找出颜色变化的根本原因。我们使用这种技术捕获了至少 2 个错误,并且我们能够更改我们的五个主题,以使 VS Code 版本之间的颜色变化最小化

Tokenization validation

之前和之后

以下是 VS Code 1.8 和现在 VS Code 1.9 中各种颜色主题的外观

Monokai 主题

Monokai before

Monokai after

安静之光主题

Quiet Light before

Quiet Light after

红色主题

Red before

Red after

结论

我希望您能欣赏升级到 VS Code 1.9 后获得的额外 CPU 时间和 RAM,并且我们能继续让您以高效愉悦的方式进行编码。

编程愉快!

Alexandru Dima,VS Code 团队成员 @alexdima123