std::call_once()的实现

std::call_once()的部分源码
pthread_cond_wait()为何先加锁后解锁又加锁
看完就会对pthread_cond_wait()和pthread_cond_broadcast()有了更好的理解。
由于不会写void* arg，就定了了call_once指定调用无参函数。哭哭。

#include<bits/stdc++.h>
#include<windows.h>
class once_flag{
public:
    once_flag()noexcept:state(0){}//noception表明这个函数不会出现异常
    friend void call_once(once_flag&,void(*func)(void));
private:
    once_flag(const once_flag&);
    once_flag& operator=(const once_flag&);
    unsigned long state;
};
static pthread_mutex_t mutex=PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t cond=PTHREAD_COND_INITIALIZER;
void _call_once(volatile unsigned long& flag,void(*func)(void)){
    pthread_mutex_lock(&mutex);
    while(flag==1) pthread_cond_wait(&cond,&mutex);
    if(flag==0){
#ifndef NO_EXCEPTION
        try{
#endif // NO_EXCEPTION
            flag=1;pthread_mutex_unlock(&mutex);
            func();
            //反正对条件变量flag操作都要加上mutex
            pthread_mutex_lock(&mutex);
            flag=~0ul;//0ul是无符号长整型的0
            pthread_mutex_unlock(&mutex);
            pthread_cond_broadcast(&cond);
#ifndef NO_EXCEPTION
        }catch(...){
            pthread_mutex_lock(&mutex);
            flag=~0ul;
            pthread_mutex_unlock(&mutex);
            pthread_cond_broadcast(&cond);
            //虽然是broadcast唤醒了所有线程,但这么多线程同时去检查条件变量,
            //需要获取到锁才能有检查条件变量的权利、只有一个检查到了条件变量推出了while
            //知道flag=1之后才解锁,但此时flag=1了,剩下被唤醒的线程还不能跳出while然后被挂起..
        }
#endif // NO_EXCEPTION
    }
    else pthread_mutex_unlock(&mutex);
}
void call_once(once_flag& flag,void(*func)(void)){
    _call_once(flag.state,func);
}
void do_once(){
    std::cout<<"pig pig get up!"<<std::endl;
}
static once_flag flag;
void task(){
    call_once(flag,do_once);
}
int main(){
    for(int i=0;i<10000;++i) new std::thread(task);
    Sleep(5000);
}