用atomic_t爲多線程分配唯一的序號

2024年2月16日 171次阅读

最近有一個需求，要求在常數區間 [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 庫。

点赞