102 lines
1.9 KiB
C++
102 lines
1.9 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop application for the Telegram messaging service.
|
|
|
|
For license and copyright information please follow this link:
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
|
*/
|
|
#pragma once
|
|
|
|
#include "base/algorithm.h"
|
|
|
|
#include <atomic>
|
|
|
|
namespace base {
|
|
|
|
class binary_guard {
|
|
public:
|
|
binary_guard() = default;
|
|
binary_guard(binary_guard &&other);
|
|
binary_guard &operator=(binary_guard &&other);
|
|
~binary_guard();
|
|
|
|
binary_guard &operator=(std::nullptr_t);
|
|
|
|
bool alive() const;
|
|
binary_guard make_guard();
|
|
|
|
explicit operator bool() const;
|
|
|
|
private:
|
|
void destroy();
|
|
|
|
std::atomic<bool> *_bothAlive = nullptr;
|
|
|
|
};
|
|
|
|
inline binary_guard::binary_guard(binary_guard &&other)
|
|
: _bothAlive(base::take(other._bothAlive)) {
|
|
}
|
|
|
|
inline binary_guard &binary_guard::operator=(binary_guard &&other) {
|
|
if (this != &other) {
|
|
destroy();
|
|
_bothAlive = base::take(other._bothAlive);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
inline binary_guard::~binary_guard() {
|
|
destroy();
|
|
}
|
|
|
|
inline binary_guard &binary_guard::operator=(std::nullptr_t) {
|
|
destroy();
|
|
return *this;
|
|
}
|
|
|
|
inline binary_guard::operator bool() const {
|
|
return alive();
|
|
}
|
|
|
|
inline bool binary_guard::alive() const {
|
|
return _bothAlive && _bothAlive->load();
|
|
}
|
|
|
|
inline void binary_guard::destroy() {
|
|
if (const auto both = base::take(_bothAlive)) {
|
|
auto old = true;
|
|
if (!both->compare_exchange_strong(old, false)) {
|
|
delete both;
|
|
}
|
|
}
|
|
}
|
|
|
|
inline binary_guard binary_guard::make_guard() {
|
|
destroy();
|
|
|
|
auto result = binary_guard();
|
|
_bothAlive = result._bothAlive = new std::atomic<bool>(true);
|
|
return result;
|
|
}
|
|
|
|
} // namespace base
|
|
|
|
namespace crl {
|
|
|
|
template <typename T, typename Enable>
|
|
struct guard_traits;
|
|
|
|
template <>
|
|
struct guard_traits<base::binary_guard, void> {
|
|
static base::binary_guard create(base::binary_guard value) {
|
|
return value;
|
|
}
|
|
static bool check(const base::binary_guard &guard) {
|
|
return guard.alive();
|
|
}
|
|
|
|
};
|
|
|
|
} // namespace crl
|