28.14 Using Queues: Sleeping Instead Of Spinning
1 typedef struct __lock_t {
2 int flag;
3 int guard;
4 queue_t *q;
5 } lock_t;
6
7 void lock_init(lock_t *m) {
8 m->flag = 0;
9 m->guard = 0;
10 queue_init(m->q);
11 }
12
13 void lock(lock_t *m) {
14 while (TestAndSet(&m->guard, 1) == 1)
15 ; //acquire guard lock by spinning
16 if (m->flag == 0) {
17 m->flag = 1; // lock is acquired
18 m->guard = 0;
19 } else {
20 queue_add(m->q, gettid());
21 m->guard = 0;
22 park();
23 }
24 }
25
26 void unlock(lock_t *m) {
27 while (TestAndSet(&m->guard, 1) == 1)
28 ; //acquire guard lock by spinning
29 if (queue_empty(m->q))
30 m->flag = 0; // let go of lock; no one wants it
31 else
32 unpark(queue_remove(m->q)); // hold lock
33 // (for next thread!)
34 m->guard = 0;
35 }
在原文中有一段说明了这段代码这一种情况会出现race现象,那就当一个线程在调用park之前被操作系统切换到另一个持有锁的线程,然后该线程很快的又释放了锁,这就导致第一个线程在调用park时,它的id已经在queue中被删除,然后调用park就会进入永远的sleep,这不符合我们对于一个锁的预期,所以有了以下改进
1 queue_add(m->q, gettid());
2 setpark(); // new code
3 m->guard = 0;
通过 setpark(),一个线程表明自己马上要 park。如果刚好另一个线程被调度,并且调用了 unpark,那么后续的 park调用就会直接返回,而不是一直睡眠。