C++11 实现信号量Semaphore类

本文参考了http://blog.csdn.net/zdarks/article/details/46994767的代码,并作出适当修改。

Semaphore.h

 1 //Semaphore.h
 2 
 3 #pragma once
 4 #include <mutex>
 5 #include <condition_variable>
 6 class Semaphore
 7 {
 8 public:
 9     explicit Semaphore(unsigned int count); //用无符号数表示信号量资源
10     ~Semaphore();
11 public:
12     void wait();
13     void signal();
14 private:
15     int m_count; //计数器必须是有符号数
16     std::mutex m_mutex;
17     std::condition_variable m_condition_variable;
18 };

Semaphore.cpp

 1 //Semaphore.cpp
 2 
 3 #include "Semaphore.h"
 4 
 5 Semaphore::Semaphore(unsigned int count):m_count(count) {
 6 }
 7 
 8 Semaphore::~Semaphore() {
 9 }
10 
11 void Semaphore::wait() {
12     std::unique_lock<std::mutex> unique_lock(m_mutex);
13     --m_count;
14     while (m_count < 0) {
15         m_condition_variable.wait(unique_lock);
16     }
17 }
18 
19 void Semaphore::signal() {
20     std::lock_guard<std::mutex> lg(m_mutex);
21     if (++m_count < 1) {
22         m_condition_variable.notify_one();
23     }
24 }
25 
26 //解析:
27 //if (++m_count < 1) 说明m_count在++之前<0,而m_count的初始值,也即信号量的值为无符号数,只可能>=0。
28 //那么<0的情况也只可能是因为先调用了wait。因此,便可以确定此时有线程在等待,故可以调用notify_one了。

 

posted @ 2017-12-03 14:47  暗夜追月  阅读(1363)  评论(0编辑  收藏  举报