用atomic_t为多线程分配唯一的序号

最近有一个需求，要求在常数区间 [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 库。