PinBuffer(BufferDesc *buf, BufferAccessStrategy strategy)
首先在PrivateRefCountArray中查找buffer,如果找到了则返回这个buffer
否则,在PrivateRefCountHash中查找,没找到,返回NULL
如果找到了,并且不需要将此buffer从hash移动到array,则返回
如果需要移动,则将这个buffer从hash中移除,存入PrivateRefCountArray
PrivateRefCountArray和PrivateRefCountHash中保存了buffer,记录了引用计数。
ReservePrivateRefCountEntry的意义
首先从PrivateRefCountArray中查找buffer是InvalidBuffer的成员,如果找到了则将这个成员赋值给ReservedRefCountEntry,ReservedRefCountEntry就指向了PrivateRefCountArray中的一个空的槽位。
如果PrivateRefCountArray中已经没有空槽位了,则需要牺牲掉一个成员,也就是将这个成员放到PrivateRefCountHash中。
这么是牺牲成员的算法,PrivateRefCountClock从0开始递增,也就是说牺牲掉的成员,从第一个成员开始,轮流到第八个,然后再从第一个成员开始牺牲,周而复始。
ReservedRefCountEntry =
&PrivateRefCountArray[PrivateRefCountClock++ % REFCOUNT_ARRAY_ENTRIES];
PrivateRefCountOverflowed中记录了这种牺牲掉的成员的数量,也就是溢出到hash的数量。
PinBuffer的流程
- GetPrivateRefCountEntry在PrivateRefCountArray和PrivateRefCountHash中查找buffer
- 如果找到了,直接将引用计数加1
- 如果没找到
3.1创建一个PrivateRefCountArray槽位,用于存放buffer和引用计数
3.2 如果buffer被锁,则等待锁释放
3.3 计算buffer的state的新值(即引用计数和使用计数)
3.4 将计算好的新值存入state(这里用到了cas操作)
4、引用计数加1
源码中的CAS操作举例
if (ux_atomic_compare_exchange_u32(&buf->state, &old_buf_state,
buf_state))
buf->state从解锁到赋值,中间有关于state的计算,在这个过程中,其值可能会被修改,CAS操作就是先检查state是不是还等于旧值old_buf_state,如果等于,说明没被修改,buf->state可以安全的被更新为buf_state,否则,其他进程修改了buf->state,本进程放弃更新buf->state的操作。
这种比较并更新的操作,对于CPU来将是一条指令,在执行完成前,不会进行其他操作,所以对于多进程来说,这种赋值是安全并且高效的。
关于SpinLockAcquire的实现,它的核心下面函数
tas(volatile slock_t *lock)
{
return __sync_lock_test_and_set(lock, 1);
}
C代码就要简单的多。如果锁没有被持有,那么锁对象的locked字段会是0,如果locked字段等于0,我们调用test-and-set将1写入locked字段,并且返回locked字段之前的数值0。如果返回0,那么意味着没有人持有锁,循环结束。如果locked字段之前是1,那么这里的流程是,先将之前的1读出,然后写入一个新的1,但是这不会改变任何数据,因为locked之前已经是1了。之后__sync_lock_test_and_set会返回1,表明锁之前已经被人持有了,这样的话,判断语句不成立,程序会持续循环(spin),直到锁的locked字段被设置回0。