X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=folly%2FAtomicHashArray.h;h=16b8dad0c97dd0b48fe5110ea7b3c7dd78b5a5f0;hb=053fb02511ed385754f32987baa616221125036f;hp=96a647d1a3da6bcf0e0f907273a20da1c09f7a24;hpb=41365ea66d92749ba78f73d60325e5447beb04ab;p=folly.git diff --git a/folly/AtomicHashArray.h b/folly/AtomicHashArray.h index 96a647d1..16b8dad0 100644 --- a/folly/AtomicHashArray.h +++ b/folly/AtomicHashArray.h @@ -39,6 +39,7 @@ #include #include +#include namespace folly { @@ -66,16 +67,6 @@ struct AtomicHashArrayQuadraticProbeFcn // Enables specializing checkLegalKey without specializing its class. namespace detail { -// Local copy of folly::gen::Identity, to avoid heavy dependencies. -class AHAIdentity { - public: - template - auto operator()(Value&& value) const -> - decltype(std::forward(value)) { - return std::forward(value); - } -}; - template inline void checkLegalKeyIfKeyTImpl(NotKeyT /* ignored */, KeyT /* emptyKey */, @@ -89,22 +80,26 @@ inline void checkLegalKeyIfKeyTImpl(KeyT key_in, KeyT emptyKey, DCHECK_NE(key_in, lockedKey); DCHECK_NE(key_in, erasedKey); } -} // namespace detail - -template , - class EqualFcn = std::equal_to, - class Allocator = std::allocator, - class ProbeFcn = AtomicHashArrayLinearProbeFcn, - class KeyConvertFcn = detail::AHAIdentity> +} // namespace detail + +template < + class KeyT, + class ValueT, + class HashFcn = std::hash, + class EqualFcn = std::equal_to, + class Allocator = std::allocator, + class ProbeFcn = AtomicHashArrayLinearProbeFcn, + class KeyConvertFcn = Identity> class AtomicHashMap; -template , - class EqualFcn = std::equal_to, - class Allocator = std::allocator, - class ProbeFcn = AtomicHashArrayLinearProbeFcn, - class KeyConvertFcn = detail::AHAIdentity> +template < + class KeyT, + class ValueT, + class HashFcn = std::hash, + class EqualFcn = std::equal_to, + class Allocator = std::allocator, + class ProbeFcn = AtomicHashArrayLinearProbeFcn, + class KeyConvertFcn = Identity> class AtomicHashArray : boost::noncopyable { static_assert((std::is_convertible::value || std::is_convertible::value || @@ -132,7 +127,7 @@ class AtomicHashArray : boost::noncopyable { const KeyT kLockedKey_; const KeyT kErasedKey_; - template + template struct aha_iterator; typedef aha_iterator const_iterator; @@ -184,7 +179,6 @@ class AtomicHashArray : boost::noncopyable { uint32_t entryCountThreadCacheSize; size_t capacity; // if positive, overrides maxLoadFactor - public: // Cannot have constexpr ctor because some compilers rightly complain. Config() : emptyKey((KeyT)-1), lockedKey((KeyT)-2), @@ -215,17 +209,19 @@ class AtomicHashArray : boost::noncopyable { * * See folly/test/ArrayHashArrayTest.cpp for sample usage. */ - template + template < + typename LookupKeyT = key_type, + typename LookupHashFcn = hasher, + typename LookupEqualFcn = key_equal> iterator find(LookupKeyT k) { return iterator(this, findInternal(k).idx); } - template + template < + typename LookupKeyT = key_type, + typename LookupHashFcn = hasher, + typename LookupEqualFcn = key_equal> const_iterator find(LookupKeyT k) const { return const_cast(this)-> find(k); @@ -260,11 +256,12 @@ class AtomicHashArray : boost::noncopyable { * equal key is already present, this method converts 'key_in' to a key of * type KeyT using the provided LookupKeyToKeyFcn. */ - template + template < + typename LookupKeyT = key_type, + typename LookupHashFcn = hasher, + typename LookupEqualFcn = key_equal, + typename LookupKeyToKeyFcn = key_convert, + typename... ArgTs> std::pair emplace(LookupKeyT key_in, ArgTs&&... vCtorArgs) { SimpleRetT ret = insertInternal + template < + typename LookupKeyT = key_type, + typename LookupHashFcn = hasher, + typename LookupEqualFcn = key_equal, + typename LookupKeyToKeyFcn = Identity, + typename... ArgTs> SimpleRetT insertInternal(LookupKeyT key, ArgTs&&... vCtorArgs); - template + template < + typename LookupKeyT = key_type, + typename LookupHashFcn = hasher, + typename LookupEqualFcn = key_equal> SimpleRetT findInternal(const LookupKeyT key); template