自旋锁的缺点
上一篇我们讲了一个简单自旋锁的实现方法。显而易见,这种锁是有缺点的,
当多个线程同时竞争一个锁时,就有可能出现某一个线程被“饿死”的情况,也就是该线程一直得不到锁,因为上一篇中的自旋锁没有公平性,并不能保证每一个线程都能得到锁。
线程如果得不到锁,他就会一直自旋(while循环判断),浪费资源。一个线程一直自旋,会浪费掉整个时间片。
如何改进
我们希望每一个线程都能加到锁,不会被饿死。
要保证一个线程在锁失败之后能够不浪费资源,能在锁释放之后继续竞争加锁操作。
本篇文章先解决第一个小问题,先保证每一个线程都能加到锁。下面是我们上一篇文章中实现的自旋锁
typedef struct mythread{
int mlock;
}mythread_t;
void mythread_init(mythread_t m)
{
m.mlock=0;//0代表没锁,1代表锁上了
}
int test_set(int* ptr,int x)//伪代码,实际为原子操作
{
int old=*ptr;
*ptr=x;
return old;
}
void mythread_lock(mythread_t* m)
{
while(test_set(&m->mlock,1)==1);
}
void mythread_unlock(mythread_t* m)
{
m->mlock=0;
}
下面是我们改进的新锁,也叫ticket锁,很有趣
typedef struct mythread{
int ticket;//票
int turn;//顺序
}mythread_t;
void mythread_init(mythread_t m)
{
m.ticket=0;
m.turn=0;
}
int FetchAndAdd(int* ptr)//伪代码,实际为原子操作
{
int old=*ptr;
*ptr=old+1;
return old;
}//每次返回旧的值,且给*ptr加1
void mythread_lock(mythread_t* m)
{
int myticket=FetchAndAdd(&m->ticket);
//获取一张票,来表示自己的顺序,并把票号加一,这样下一个获得的票号就比本次大一
while(m->turn!=myticket);
//判断顺序是否轮到自己的票了,如果没轮到自己,就自旋等待
}
void mythread_unlock(mythread_t* m)
{
FetchAndAdd(&m->turn);//顺序加一
}