深入探究Lua的GC算法

对于内存的管理,是程序在应用的时候的必需知识点,《Lua设计与实现》中对Lua语言的GC原理做了一个详细的讲解,云风的blog也对其进行了详尽的讲解Lua GC 的源码剖析 系列

给出作者 

@codedump
 在github上的lua源码链接: github.com/lichuang/Lua

这儿就继续做《Lua设计与实现》的阅读笔记,对Lua GC原理及其过程做一个详尽的讲解,由于篇幅较大,就一分为二,写上下篇来讲解整个过程。


一、GC的原理及其算法设计

不同的语言,对GC算法的设计不同,常见的GC算法是引用计数和Mark-Sweep算法, c#采用的是Mark-sweep && compact算法, Lua采用的是Mark-sweep算法,分开说一下:

引用计数算法:在一个对象被引用的情况下,将其引用计数加1,反之则减1,如果计数值为0,则在GC的时候回收,这个算法有个问题就是循环引用。

Mark-sweep算法:每次GC的时候,对所有对象进行一次扫描,如果该对象不存在引用,则被回收,反之则保存。

在Lua5.0及其更早的版本中,Lua的GC是一次性不可被打断的过程,使用的Mark算法是双色标记算法(Two color mark),这样系统中对象的非黑即白,要么被引用,要么不被引用,这会带来一个问题:在GC的过程中如果新加入对象,这时候新加入的对象无论怎么设置都会带来问题,如果设置为白色,则如果处于回收阶段,则该对象会在没有遍历其关联对象的情况下被回收;如果标记为黑色,那么没有被扫描就被标记为不可回收,是不正确的。


为了降低一次性回收带来的性能问题以及双色算法的问题,在Lua5.1后,Lua都采用分布回收以及三色增量标记清除算法(Tri-color incremental mark and sweep)


其基本的原理伪代码,参考书中原文为:

每个新创建的对象颜色设置为白色

//初始化阶段

遍历root节点中引用的对象,从白色置为灰色,并且放入到灰色节点列表中

//标记阶段

while(灰色链表中还有未扫描的元素):

从中取出一个对象,将其置为黑色

遍历这个对象关联的其他所有对象:

if 为白色

标记为灰色,加入到灰色链表中(insert to the head)

//回收阶段

遍历所有对象:

if 为白色,

没有被引用的对象,执行回收

else

重新塞入到对象链表中,等待下一轮GC


二、GC的数据结构

分析Lua中对于需要GC的类型数据

#define iscollectable(o) (ttype(o) >= LUA_TSTRING)

都会有一个基本的定义CommonHeader,其定义为:

深入探究Lua的GC算法_第1张图片

next: GCObject链表指针,该指针用来将所有的GC对象都链接在一个表中;

tt: 数据类型:nil, boolean, number, string...

marked: 标记字段,byte表示的字段颜色定义为

深入探究Lua的GC算法_第2张图片

这儿特定解释一下为什么会有两种白色,前面提到,5.1后的Lua采用的是三色标记算法,其实质是四色标记算法,分为0型白色和1型白色,在GC回收的时候,会设置当前的白色为其中一种,详见globalstate中的currentwhite,这样在代码回收的时候,如果当前对象的白色不为currentwhite,则认为其不可回收,这样的对象需要等到下一次的GC才能决定是否回收,具体参看后面的,会有对应的应用。对于global_state的设计为:

深入探究Lua的GC算法_第3张图片

具体的参数的作用,详见注释,就不在一一解释了


三、GC的流程

1、数据的创建

想要了解GC的过程,首先看看数据是怎么在创建的时候被链接到GC链表中的,主要分为三种数据的创建

1) 一般数据的创建 luaC_link

深入探究Lua的GC算法_第4张图片

简单直接,直接insert to the head

2) upval的创建 luaC_linkupval

深入探究Lua的GC算法_第5张图片

3) userdata(udata)的创建 luaS_newdata

深入探究Lua的GC算法_第6张图片


2、开始GC

整个GC过程分为五个阶段,其定义为

深入探究Lua的GC算法_第7张图片

其执行GC的函数为singlestep,来看第一步的操作:

深入探究Lua的GC算法_第8张图片


进一步看看markroot的操作:

深入探究Lua的GC算法_第9张图片

其实就是reset一遍相关的变量,然后标记mainthread, G表,registry表,然后切换到下一个标记阶段。

参看定义:

最后都要执行reallymarkobject函数(此处需要展示一下我的竖屏截图便利了:D):

深入探究Lua的GC算法_第10张图片

基本的注释都解释了各个对象是如何的处理的,udata是不会引用其他类型的数据,所以一步到黑色,upvalue则根据是否为close来决定是否标记到黑色,open状态的upvalue变化较为频繁,需要在后面的remarkupvals中解决。


3、GC的扫描阶段 GCSpropagate

深入探究Lua的GC算法_第11张图片

只要处于这个阶段,就会分2种情况执行,一个是propagatemark,一个是atomic,让我们分别看其实现过程。

首先看处于灰色链表中一直都有对象的情况,在这步操作当中,是可以分步操作的,整个GC的分步操作,就是在这一步操作中,在每次扫描后,都会返回本次扫描标记的对象的大小之和,再下一个分步执行的时候再继续执行,而一旦进入atomic函数中,就需要一次性的执行,不能再分步执行了。

来看propagatemark函数是如何实现的:

深入探究Lua的GC算法_第12张图片

对于table,如果该表是weak表,则退回到灰色状态,否则遍历表的数组和散列表部分进行标记,详见traversetable函数;

对于func,traverseclosure主要对func中的upval进行标记;

对于thread, 则将其移植到grayagain中,放在atomic中进行处理;

对于proto,对其中的字符串、upvalue、局部变量等进行遍历标记;

注意,这儿没有处理string\udata类型数据,这是放在其他部分进行的,不需要进行相关的标记;


4、GC 扫描阶段的barrier操作

由于采用分步式增量扫描标记算法,所以会出现在分步操作过程中,新增加的对象与被扫描过的对象之间有引用关系的变化,未来确保黑色对象引用的对象中有白色对象,lua提供了两种操作设计:

1)标记过程向前走一步 luaC_barrierf

如果新建对象是白色,而它被一个黑色对象引用了,那么将这个新建对象颜色从白色变为灰色;

2)标记过程向后走一步 luaC_barrierback

类似于上,此时将引用的它的黑色对象的颜色从黑色变为灰色,使得其重新被扫描一次

深入探究Lua的GC算法_第13张图片

(或许你看出截图颜色变了,是的,回家了,又是新的编辑器了~)

从define可以看出,只有table需要进行luaC_barrierback,这是由于table本身设计,就是一个table可能会对应N个key或者value,这样如果新增一个key/value,如果将其置为灰色,然后将其加入gray链表中,这样多个添加会带来较大的性能。

采用向后,就是将该table对象退回到gray状态,这样添加多个,其实质都是只改变该table一次,注意这个gray不是改为gray链中,而是将该table加入到grayagain链中,在扫描完gray链后再扫描grayagain链即可。参考源码即可:

深入探究Lua的GC算法_第14张图片

对比向前比较简单了:直接调用reallymarkoject

深入探究Lua的GC算法_第15张图片


5、GC的atomic操作

当gray链表中对象都标记完成后,会执行一次atomic操作,注意这个操作是不能被打断的,所以叫原子操作,参考源码:

深入探究Lua的GC算法_第16张图片

首先处理上一篇文章中提到的对open状态的upvalues,然后处理一次gray链表;

然后处理整个弱表,将lua_State指针指向meta表,然后处理一次gray链表

然后处理grayagain链表,类似于上

然后处理udata,其处理函数为luaC_separateudata:

深入探究Lua的GC算法_第17张图片

注释很详细,注意放到tmudata链表中后,是在后续操作再集中处理一次;

处理完基本的几个数据后,atomic会把白色类型切换到下一个GC操作的白色类型,然后修改状态到回收阶段CGSsweepstring, 这儿对sweepstrgc进行了赋初值,是为了下面的字符串定位。


6、GC的回收阶段 GCSsweepstring/GCSsweep

首先进入的回收阶段是对字符串的处理

深入探究Lua的GC算法_第18张图片

虽然是case,但是其实质是一个循环,每次取出散列表中的一个字符串链表,进行一次遍历回收,sweepwholelist最终会调用到sweeplist,等一下给出源码。

当处理完所有的字符串后,切换到GCSsweep状态:

深入探究Lua的GC算法_第19张图片

关键操作是sweeplist,参看其源码:

深入探究Lua的GC算法_第20张图片

代码中也对前面说的多色标记中的两种白色的作用做了讲解,otherwhite就是本次不可回收的白色,如果处理的对象的白色就是otherwhite,是不会被回收的


7、结束阶段 GCSfinalize

这是整个GC的最后阶段了,来看看其操作的源码:

深入探究Lua的GC算法_第21张图片

首先处理,是否有前面提到的tmudata链表, 其操作函数为GCTM:

深入探究Lua的GC算法_第22张图片

注意,udata本身有GC方法,未来确保其GC方法的调用,实在这次GC中调用G方法,但是这个udata本身,是在下一次的GC中才会被回收的。udata的GC调用则是在fasttm中调用TM_GC来实现。

初看也会迷糊怎么循环的,其实结合上面的case中的 if(g->tmudata)可以理解,为什么每次GCTM都会执行 g->tmudata的移动赋值操作。


最终万事大吉,本次GC流程走完,设置到GCSpause状态,等待下一次GC调用。


8、GC的进度控制

其实GC的调用,可以分为两种,一种是自动调用,一个是手动调用

自动调用函数: luaC_checkGC

深入探究Lua的GC算法_第23张图片

一般不希望自动GC,可以采用setthreshold,将GCthreshold的值设置为非常大,这样不回自动触发GC

手动调用,则设置GC的相关参数 setthreshold:

estimate是对当前内存使用量的一个预估值,gcpause是一个百分比,通过lua_gc可以设置,另一个gc进度的参数是gcstepmul,其主要影响singlestep函数的调用次数,具体原因参看源码:

深入探究Lua的GC算法_第24张图片

整个流程都在注释中讲解了,其中关键是lim的设置,然后不断的调用singlestep, 然后处理GC状态即可,注意setthreshold是设置的两次GC之间的时间间隔。由于修改了threshold,对于关闭自动GC的情况,需要再次重新设置关闭自动GC一次。


9、总结

对于lua的GC的原理的探究就到这儿,熟悉一门语言的GC流程后,同理去推导理解其他语言的GC会有很大帮助,同时也可以在平时使用lua的时候,对于GC的一些操作更加知其所以然。大家共勉!


你可能感兴趣的:(lua)