diff options
| -rw-r--r-- | src/libutil/cxx/util.hxx | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/src/libutil/cxx/util.hxx b/src/libutil/cxx/util.hxx index 3eeb6d20d..fa9aa1802 100644 --- a/src/libutil/cxx/util.hxx +++ b/src/libutil/cxx/util.hxx @@ -31,7 +31,7 @@ namespace rspamd { * Transparent smart pointers hashing */ template<typename T> -struct shared_ptr_equal { +struct smart_ptr_equal { using is_transparent = void; /* We want to find values in a set of shared_ptr by reference */ auto operator()(const std::shared_ptr<T> &a, const std::shared_ptr<T> &b) const { return (*a) == (*b); @@ -42,14 +42,26 @@ struct shared_ptr_equal { auto operator()(const T &a, const std::shared_ptr<T> &b) const { return a == (*b); } + auto operator()(const std::unique_ptr<T> &a, const std::unique_ptr<T> &b) const { + return (*a) == (*b); + } + auto operator()(const std::unique_ptr<T> &a, const T &b) const { + return (*a) == b; + } + auto operator()(const T &a, const std::unique_ptr<T> &b) const { + return a == (*b); + } }; template<typename T> -struct shared_ptr_hash { +struct smart_ptr_hash { using is_transparent = void; /* We want to find values in a set of shared_ptr by reference */ auto operator()(const std::shared_ptr<T> &a) const { return std::hash<T>()(*a); } + auto operator()(const std::unique_ptr<T> &a) const { + return std::hash<T>()(*a); + } auto operator()(const T &a) const { return std::hash<T>()(a); } |