/*
- * Copyright 2016 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
#pragma once
-#include <atomic>
-#include <type_traits>
#include <folly/Traits.h>
-#include <string.h>
+#include <folly/synchronization/detail/AtomicUtils.h>
#include <stdint.h>
+#include <string.h>
+#include <atomic>
+#include <type_traits>
namespace folly {
namespace detail {
template <int N> struct AtomicStructIntPick {};
-}
+} // namespace detail
/// AtomicStruct<T> work like C++ atomics, but can be used on any POD
/// type <= 8 bytes.
template <
typename T,
- template<typename> class Atom = std::atomic,
+ template <typename> class Atom = std::atomic,
typename Raw = typename detail::AtomicStructIntPick<sizeof(T)>::type>
class AtomicStruct {
static_assert(alignof(T) <= alignof(Raw),
}
bool compare_exchange_strong(
- T& v0, T v1,
- std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ T& v0,
+ T v1,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_strong(
+ v0, v1, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_strong(
+ T& v0,
+ T v1,
+ std::memory_order success,
+ std::memory_order failure) noexcept {
Raw d0 = encode(v0);
- bool rv = data.compare_exchange_strong(d0, encode(v1), mo);
+ bool rv = data.compare_exchange_strong(d0, encode(v1), success, failure);
if (!rv) {
v0 = decode(d0);
}
}
bool compare_exchange_weak(
- T& v0, T v1,
- std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ T& v0,
+ T v1,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_weak(
+ v0, v1, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_weak(
+ T& v0,
+ T v1,
+ std::memory_order success,
+ std::memory_order failure) noexcept {
Raw d0 = encode(v0);
- bool rv = data.compare_exchange_weak(d0, encode(v1), mo);
+ bool rv = data.compare_exchange_weak(d0, encode(v1), success, failure);
if (!rv) {
v0 = decode(d0);
}