TConcurrentHashMap::Exchange

commit_hash:09134890fc4d5d1267d6efa21bff48235205e6e5

1 files changed, 7 insertions, 0 deletions

diff --git a/library/cpp/containers/concurrent_hash/concurrent_hash.h b/library/cpp/containers/concurrent_hash/concurrent_hash.h
index 6fed13ee8d5..204434a9783 100644
--- a/library/cpp/containers/concurrent_hash/concurrent_hash.h
+++ b/library/cpp/containers/concurrent_hash/concurrent_hash.h
@@ -1,6 +1,7 @@
 #pragma once
 
 #include <util/generic/hash.h>
+#include <util/generic/utility.h>
 #include <util/system/spinlock.h>
 
 #include <array>
@@ -103,6 +104,12 @@ public:
         bucket.Map[key] = value;
     }
 
+    void Exchange(const K& key, V& value) {
+        TBucket& bucket = GetBucketForKey(key);
+        TBucketGuard guard(bucket.Mutex);
+        DoSwap(bucket.Map[key], value);
+    }
+
     void InsertUnique(const K& key, const V& value) {
         TBucket& bucket = GetBucketForKey(key);
         TBucketGuard guard(bucket.Mutex);