84 lines
1.7 KiB
C++
84 lines
1.7 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLToolKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLToolKit(https://github.com/ZLMediaKit/ZLToolKit).
|
||
*
|
||
* Use of this source code is governed by MIT license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#ifndef SEMAPHORE_H_
|
||
#define SEMAPHORE_H_
|
||
|
||
/*
|
||
* 目前发现信号量在32位的系统上有问题，
|
||
* 休眠的线程无法被正常唤醒，先禁用之
|
||
#if defined(__linux__)
|
||
#include <semaphore.h>
|
||
#define HAVE_SEM
|
||
#endif //HAVE_SEM
|
||
*/
|
||
|
||
#include <mutex>
|
||
#include <condition_variable>
|
||
|
||
namespace toolkit {
|
||
|
||
class semaphore {
|
||
public:
|
||
explicit semaphore(size_t initial = 0) {
|
||
#if defined(HAVE_SEM)
|
||
sem_init(&_sem, 0, initial);
|
||
#else
|
||
_count = 0;
|
||
#endif
|
||
}
|
||
|
||
~semaphore() {
|
||
#if defined(HAVE_SEM)
|
||
sem_destroy(&_sem);
|
||
#endif
|
||
}
|
||
|
||
void post(size_t n = 1) {
|
||
#if defined(HAVE_SEM)
|
||
while (n--) {
|
||
sem_post(&_sem);
|
||
}
|
||
#else
|
||
std::unique_lock<std::recursive_mutex> lock(_mutex);
|
||
_count += n;
|
||
if (n == 1) {
|
||
_condition.notify_one();
|
||
} else {
|
||
_condition.notify_all();
|
||
}
|
||
#endif
|
||
}
|
||
|
||
void wait() {
|
||
#if defined(HAVE_SEM)
|
||
sem_wait(&_sem);
|
||
#else
|
||
std::unique_lock<std::recursive_mutex> lock(_mutex);
|
||
while (_count == 0) {
|
||
_condition.wait(lock);
|
||
}
|
||
--_count;
|
||
#endif
|
||
}
|
||
|
||
private:
|
||
#if defined(HAVE_SEM)
|
||
sem_t _sem;
|
||
#else
|
||
size_t _count;
|
||
std::recursive_mutex _mutex;
|
||
std::condition_variable_any _condition;
|
||
#endif
|
||
};
|
||
|
||
} /* namespace toolkit */
|
||
#endif /* SEMAPHORE_H_ */
|