互斥锁和条件变量的封装类

快来打我* 2022-04-04 12:18 253阅读 0赞

/*
    两个简单的实现类： 
    MutexLock，MutexLockGuard，Condition类
    MutexLock是封装了互斥量的封装类
    MutexLockGuard是管理MutexLock资源的警卫类
    Condition是条件变量的封装类
     */
    
    class MutexLock : boost::noncopyable {
        
    public:
    	MutexLock() : holder_(0) {
        
    		pthread_mutex_init(&mutex_,NULL);
    	}
    	~MutexLock(){
        
    		assert(holder_ == 0);
    		pthread_mutex_destory(&mutex_);
    	}
    
    	bool isLockedByThisThread() {
        
    		return holder_ == CurrentThread::tid();
    	}
    
    	void asswetLocked() {
        
    		assert(isLockedByThisThread());
    	}
    
    	void lock() {
        
    		pthread_mutex_lock(&mutex_);
    		holder_ = CurrentThread::tid();
    	}
    
    	void unlock() {
        
    		holder_ = 0;
    		pthread_mutex_unlock(&mutex_);
    	}
    
    	pthread_mutex_t* getPthreadMutex() {
        
    		return *mutex_;
    	}
    
    private:
    	pthread_mutex_t mutex_;
    	pid_t holder_;
    };
    
    
    class MutexLockGuard(MutexLock& mutex) {
        
    public:
    	explicit MutexLockGuard(MutexLock& mutex) :mutex_(mutex) {
        
    		mutex_.lock();
    	}
    
    	~MutexLockGuard() {
        
    		mutex_.unlock();
    	}
    
    private:
    	MutexLock& mutex_;
    };
    
    #define MutexLockGuard(x) static_assert(false,"missing mutex guard var name")
    
    
    // 正确的使用 MutexLockGuard lock(mutex);
    
    class Condition {
        
    public:
    	explicit Condition(MutexLock& mutex):mutex_(mutex){
        
    		pthread_cond_init(&pcond_,Null);
    	}
    
    	~Condition() {
        
    		pthread_cond_destory(&pcond_);
    	}
    
    	void wait() {
         pthread_cond_wait(&pcond_,mutex_.getPthreadMutex());}
    	void notify() {
         pthread_cond_signal(&pcond_);}
    	void notifyall() {
        pthread_cond_sbroadcast(&pcond_);}
    
    private:
    	MutexLock& mutex_;
    	pthread_cond_t pcond_;
    };