Intermediate changes

commit_hash:9269fba6accab3db16a0bf72a8ca256583d95e72

1 files changed, 14 insertions, 1 deletions

diff --git a/library/cpp/containers/ordered_map/ordered_map.h b/library/cpp/containers/ordered_map/ordered_map.h
index 801a526217f..22422901a84 100644
--- a/library/cpp/containers/ordered_map/ordered_map.h
+++ b/library/cpp/containers/ordered_map/ordered_map.h
@@ -14,6 +14,7 @@ namespace NOrderedMap {
     struct TPairTraits {
         using TValue = Value;
         using Item = std::pair<const Key, Value>;
+        using StorageItem = Item;
         using TMutableItem = std::pair<Key, Value>;
 
         static const Key& GetSortProjector(const Item& x) {
@@ -29,6 +30,7 @@ namespace NOrderedMap {
     template<class Key>
     struct TSelfTraits {
         using Item = const Key;
+        using StorageItem = Key;
         using TMutableItem = Key;
         using TValue = Item;
 
@@ -44,12 +46,13 @@ namespace NOrderedMap {
     template <
         class TKey,
         class Traits,
-        class BaseToUse = std::list<typename Traits::Item> //bases: deque or list
+        class BaseToUse = std::list<typename Traits::StorageItem> //bases: deque or list
     >
     class TOrderedMapImpl : protected BaseToUse {
         using TBase = BaseToUse;
 
     public:
+        using TTraits = Traits;
         using typename TBase::const_iterator;
         using typename TBase::iterator;
         using typename TBase::reference;
@@ -62,6 +65,7 @@ namespace NOrderedMap {
         using TBase::end;
         using TBase::rbegin;
         using TBase::rend;
+
         operator bool() const {
             return empty();
         }
@@ -164,6 +168,11 @@ namespace NOrderedMap {
         }
 
         template <class... TArgs>
+        reference emplace(TArgs&&... args) {
+            return emplace_back(std::forward<TArgs>(args)...);
+        }
+
+        template <class... TArgs>
         reference try_emplace_back(TArgs&&... args) {
             auto& value = TBase::emplace_back(std::forward<TArgs>(args)...);
             auto [mapIt, ok] = Map_.try_emplace(&Traits::GetSortProjector(value), --TBase::end());
@@ -276,5 +285,9 @@ namespace NOrderedMap {
 
     template <class TKey>
     class TOrderedSet : public TOrderedMapImpl<TKey, TSelfTraits<TKey>> {
+    public:
+        auto operator==(const TOrderedSet& b) const {
+            return std::lexicographical_compare_three_way(this->begin(), this->end(), b.begin(), b.end()) == std::strong_ordering::equal;
+        }
     };
 }