最近有一个需求,要求在常数区间 [S, E] 里为多个线程分配唯一的序号。
如: S, S+1, S+2, ... E-2, E-1, E, S, S+1,...
困难在于不使用锁的情况下实现线程安全。
最后使用内核提供的 atomic_t 实现。
代码如下:
int seq()
{
static atomic_t curr = ATOMIC_INIT(S);
int new = -1;
int old = -1;
while(1) {
new = atomic_read(&curr);
old = atomic_cmpxchg(&curr, new, new + 1);
if(old != new) {
//a race condition, get next one
continue;
}
if(new == E) {
atomic_set(&curr, S);
} else if(new > E){
//Rare but possible
continue;
}
break;
}
}
gcc 也提供了类似的原子操作(参考 http://gcc.gnu.org/onlinedocs/gcc-4.1.1/gcc/Atomic-Builtins.html),应该可以不依赖 atomic_t 库。