gitbook/深入拆解Java虚拟机/docs/41347.md

208 lines
14 KiB
Markdown
Raw Permalink Normal View History

2022-09-03 22:05:03 +08:00
# 35 | Truffle语言实现框架
今天我们来聊聊GraalVM中的语言实现框架Truffle。
我们知道,实现一门新编程语言的传统做法是实现一个编译器,也就是把用该语言编写的程序转换成可直接在硬件上运行的机器码。
通常来说,编译器分为前端和后端:前端负责词法分析、语法分析、类型检查和中间代码生成,后端负责编译优化和目标代码生成。
不过,许多编译器教程只涉及了前端中的词法分析和语法分析,并没有真正生成可以运行的目标代码,更谈不上编译优化,因此在生产环境中并不实用。
另一种比较取巧的做法则是将新语言编译成某种已知语言或者已知的中间形式例如将Scala、Kotlin编译成Java字节码。
这样做的好处是可以直接享用Java虚拟机自带的各项优化包括即时编译、自动内存管理等等。因此这种做法对所生成的Java字节码的优化程度要求不高。
不过,不管是附带编译优化的编译器,还是生成中间形式并依赖于其他运行时的即时编译优化的编译器,它们所针对的都是[编译型语言](https://en.wikipedia.org/wiki/Compiled_language),在运行之前都需要这一额外的编译步骤。
与编译型语言相对应的则是[解释型语言](https://en.wikipedia.org/wiki/Interpreted_language)例如JavaScript、Ruby、Python等。对于这些语言来说它们无须额外的编译步骤而是依赖于解释执行器进行解析并执行。
为了让该解释执行器能够高效地运行大型程序,语言实现开发人员通常会将其包装在虚拟机里,并实现诸如即时编译、垃圾回收等其他组件。这些组件对语言设计 本身并无太大贡献,仅仅是为了实用性而不得不进行的工程实现。
在理想情况下我们希望在不同的语言实现中复用这些组件。也就是说每当开发一门新语言时我们只需要实现它的解释执行器便能够直接复用即时编译、垃圾回收等组件从而达到高性能的效果。这也是Truffle项目的目标。接下来我们就来讲讲这个项目。
## Truffle项目简介
Truffle是一个用Java写就的语言实现框架。基于Truffle的语言实现仅需用Java实现词法分析、语法分析以及针对语法分析所生成的抽象语法树Abstract Syntax TreeAST的解释执行器便可以享用由Truffle提供的各项运行时优化。
就一个完整的Truffle语言实现而言由于实现本身以及其所依赖的Truffle框架部分都是用Java实现的因此它可以运行在任何Java虚拟机之上。
当然如果Truffle运行在附带了Graal编译器的Java虚拟机之上那么它将调用Graal编译器所提供的API主动触发对Truffle语言的即时编译将对AST的解释执行转换为执行即时编译后的机器码。
在这种情况下Graal编译器相当于一个提供了即时编译功能的库宿主虚拟机本身仍可使用C2作为其唯一的即时编译器或者分层编译模式下的4层编译器。
![](https://static001.geekbang.org/resource/image/20/30/20c7a514f226689536fafc6886a08e30.png)
我们团队实现并且开源了多个Truffle语言例如[JavaScript](https://github.com/graalvm/graaljs)[Ruby](https://github.com/oracle/truffleruby)[R](https://github.com/oracle/fastr)[Python](https://github.com/graalvm/graalpython)以及可用来解释执行LLVM bitcode的[Sulong](https://github.com/oracle/graal/tree/master/sulong)。关于Sulong项目任何能够编译为LLVM bitcode的编程语言例如C/C++,都能够在这上面运行。
下图展示了运行在GraalVM EE上的Java虚拟机语言以及除Python外Truffle语言的峰值性能指标2017年数据
![](https://static001.geekbang.org/resource/image/0a/44/0aa87b77b2d6eb0147d4a2b342b0d644.png)
这里我采用的基线是每个语言较有竞争力的语言实现。
* 对于Java虚拟机语言Java、Scala我比较的是使用C2的HotSpot虚拟机和使用Graal的HotSpot虚拟机。
* 对于Ruby我比较的是运行在HotSpot虚拟机之上的JRuby和Truffle Ruby。
* 对于R我比较的是GNU R和基于Truffle的FastR。
* 对于C/C++我比较的是利用LLVM编译器生成的二进制文件和基于Truffle的Sulong。
* 对于JavaScript我比较的是Google的V8和Graal.js。
针对每种语言,我们运行了上百个基准测试,求出各个基准测试**峰值性能**的加速比并且汇总成图中所示的几何平均值Geo. mean
简单地说明一下当GraalVM的加速比为1时代表使用其他语言实现和使用GraalVM的性能相当。当GraalVM加速比超过1时则代表GraalVM的性能较好反之则说明GraalVM的性能较差。
我们可以看到Java跑在Graal上和跑在C2上的执行效率类似而Scala跑在Graal上的执行效率则是跑在C2上的1.2倍。
对于Ruby或者R这类解释型语言经由Graal编译器加速的Truffle语言解释器的性能十分优越分别达到对应基线的4.1x和4.5x。这里便可以看出使用专业即时编译器的Truffle框架的优势所在。
不过对于同样拥有专业即时编译器的V8来说基于Truffle的Graal.js仍处于追赶者的位置。考虑到我们团队中负责Graal.js的工程师仅有个位数能够达到如此性能已属不易。现在Graal.js已经开源出来我相信借助社区的贡献它的性能能够得到进一步的提升。
Sulong与传统的C/C++相比,由于两者最终都将编译为机器码,因此原则上后者定义了前者的性能上限。
不过Sulong将C/C++代码放在托管环境中运行所有代码中的内存访问都会在托管环境的监控之下。无论是会触发Segfault的异常访问还是读取敏感数据的恶意访问都能够被Sulong拦截下来并作出相应处理。
## Partial Evaluation
如果要理解Truffle的原理我们需要先了解Partial Evaluation这一个概念。
假设有一段程序`P`,它将一系列输入`I`转换成输出`O`(即`P: I -> O`)。而这些输入又可以进一步划分为编译时已知的常量`IS`,和编译时未知的`ID`。
那么,我们可以将程序`P: I -> O`转换为等价的另一段程序`P': ID -> O`。这个新程序`P'`便是`P`的特化Specialization而从`P`转换到`P'`的这个过程便是所谓的Partial Evaluation。
回到Truffle这边我们可以将Truffle语言的解释执行器当成`P`将某段用Truffle语言写就的程序当作`IS`并通过Partial Evaluation特化为`P'`。由于Truffle语言的解释执行器是用Java写的因此我们可以利用Graal编译器将`P'`编译为二进制码。
下面我将用一个具体例子来讲解。
假设有一门语言X只支持读取整数参数和整数加法。这两种操作分别对应下面这段代码中的AST节点`Arg`和`Add`。
```
abstract class Node {
abstract int execute(int[] args);
}
class Arg extends Node {
final int index;
Arg(int i) { this.index = i; }
int execute(int[] args) {
return args[index];
}
}
class Add extends Node {
final Node left, right;
Add(Node left, Node right) {
this.left = left;
this.right = right;
}
int execute(int[] args) {
return left.execute(args) +
right.execute(args);
}
}
static int interpret(Node node, int[] args) {
return node.execute(args);
}
```
所谓AST节点的解释执行便是调用这些AST节点的`execute`方法而一段程序的解释执行则是调用这段程序的AST根节点的`execute`方法。
我们可以看到,`Arg`节点和`Add`节点均实现了`execute`方法接收一个用来指代程序输入的int数组参数并返回计算结果。其中`Arg`节点将返回int数组的第`i`个参数(`i`是硬编码在程序之中的常量);而`Add`节点将分别调用左右两个节点的`execute`方法, 并将所返回的值相加后再返回。
下面我们将利用语言X实现一段程序计算三个输入参数之和`arg0 + arg1 + arg2`。这段程序解析生成的AST如下述代码所示
```
// Sample program: arg0 + arg1 + arg2
sample = new Add(new Add(new Arg(0), new Arg(1)), new Arg(2));
```
这段程序对应的解释执行则是`interpret(sample, args)`,其中`args`为代表传入参数的int数组。由于`sample`是编译时常量因此我们可以将其通过Partial Evaluation特化为下面这段代码所示的`interpret0`方法:
```
static final Node sample = new Add(new Add(new Arg(0), new Arg(1)), new Arg(2));
static int interpret0(int[] args) {
return sample.execute(args);
}
```
Truffle的Partial Evaluator会不断进行方法内联直至遇到被\`\`@TruffleBoundary`注解的方法)。因此,上面这段代码的`interpret0`方法,在内联了对`Add.execute\`方法的调用之后,会转换成下述代码:
```
static final Node sample = new Add(new Add(new Arg(0), new Arg(1)), new Arg(2));
static int interpret0(int[] args) {
return sample.left.execute(args) + sample.right.execute(args);
}
```
同样,我们可以进一步内联对`Add.execute`方法的调用以及对`Arg.execute`方法的调用,最终将`interpret0`转换成下述代码:
```
static int interpret0(int[] args) {
return args[0] + args[1] + args[2];
}
```
至此我们已成功地将一段Truffle语言代码的解释执行转换为上述Java代码。接下来我们便可以让Graal编译器将这段Java代码编译为机器码从而实现Truffle语言的即时编译。
## 节点重写
Truffle的另一项关键优化是节点重写node rewriting
在动态语言中,许多变量的类型是在运行过程中方能确定的。以加法符号`+`为例,它既可以表示整数加法,还可以表示浮点数加法,甚至可以表示字符串加法。
如果是静态语言,我们可以通过推断加法的两个操作数的具体类型,来确定该加法的类型。但对于动态语言来说,我们需要在运行时动态确定操作数的具体类型,并据此选择对应的加法操作。这种在运行时选择语义的节点,会十分不利于即时编译,从而严重影响到程序的性能。
Truffle语言解释器会收集每个AST节点所代表的操作的类型并且在即时编译时作出针对所收集得到的类型profile的特化specialization
还是以加法操作为例如果所收集的类型profile显示这是一个整数加法操作那么在即时编译时我们会将对应的AST节点当成整数加法如果是一个字符串加法操作那么我们会将对应的AST节点当成字符串加法。
当然,如果该加法操作既有可能是整数加法也可能是字符串加法,那么我们只好在运行过程中判断具体的操作类型,并选择相应的加法操作。
![](https://static001.geekbang.org/resource/image/54/8f/543ee374164fd43f2773043c675b568f.png)
这种基于类型profile的优化与我们以前介绍过的Java虚拟机中解释执行器以及三层C1编译代码十分类似它们背后的核心都是基于假设的投机性优化以及在假设失败时的去优化。
在即时编译过后如果运行过程中发现AST节点的实际类型和所假设的类型不同Truffle会主动调用Graal编译器提供的去优化API返回至解释执行AST节点的状态并且重新收集AST节点的类型信息。之后Truffle会再次利用Graal编译器进行新一轮的即时编译。
当然如果能够在第一次编译时便已达到稳定状态不再触发去优化以及重新编译那么这会极大地减短程序到达峰值性能的时间。为此我们统计了各个Truffle语言的方法在进行过多少次方法调用后其AST节点的类型会固定下来。
据统计在JavaScript方法和Ruby方法中80%会在5次方法调用后稳定下来90%会在7次调用后稳定下来99%会在19次方法调用之后稳定下来。
R语言的方法则比较特殊即便是不进行任何调用有50%的方法已经稳定下来了。这背后的原因也不难推测这是因为R语言主要用于数值统计几乎所有的操作都是浮点数类型的。
## Polyglot
在开发过程中,我们通常会为工程项目选定一门语言,但问题也会接踵而至:一是这门语言没有实现我们可能需要用到的库,二是这门语言并不适用于某类问题。
Truffle语言实现框架则支持Polyglot允许在同一段代码中混用不同的编程语言从而使得开发人员能够自由地选择合适的语言来实现子组件。
与其他Polyglot框架不同的是Truffle语言之间能够共用对象。也就是说在不对某个语言中的对象进行复制或者序列化反序列化的情况下Truffle可以无缝地将该对象传递给另一门语言。因此Truffle的Polyglot在切换语言时性能开销非常小甚至经常能够达到零开销。
Truffle的Polyglot特性是通过Polyglot API来实现的。每个实现了Polyglot API的Truffle语言其对象都能够被其他Truffle语言通过Polyglot API解析。实际上当通过Polyglot API解析外来对象时我们并不需要了解对方语言便能够识别其数据结构访问其中的数据并进行进一步的计算。
## 总结与实践
今天我介绍了GraalVM中的Truffle项目。
Truffle是一个语言实现框架允许语言开发者在仅实现词法解析、语法解析以及AST解释器的情况下达到极佳的性能。目前Oracle Labs已经实现并维护了JavaScript、Ruby、R、Python以及可用于解析LLVM bitcode的Sulong。后者将支持在GraalVM上运行C/C++代码。
Truffle背后所依赖的技术是Partial Evaluation以及节点重写。Partial Evaluation指的是将所要编译的目标程序解析生成的抽象语法树当做编译时常量特化该Truffle语言的解释器从而得到指代这段程序解释执行过程的Java代码。然后我们可以借助Graal编译器将这段Java代码即时编译为机器码。
节点重写则是收集AST节点的类型根据所收集的类型profile进行的特化并在节点类型不匹配时进行去优化并重新收集、编译的一项技术。
Truffle的Polyglot特性支持在一段代码中混用多种不同的语言。与其他Polyglot框架相比它支持在不同的Truffle语言中复用内存中存储的同一个对象。
* * *
今天的实践环节请你试用GraalVM中附带的各项语言实现。你可以运行我们官网上的各个[示例程序](https://www.graalvm.org/docs/examples/)。