From c5ac5b79af1a34b914ac67de302f86744e23856b Mon Sep 17 00:00:00 2001
From: eshcherbin <eshcherbin@yandex-team.com>
Date: Mon, 3 Feb 2025 10:23:16 +0300
Subject: Make TCompactFlatMap formattable
 commit_hash:e5c092a9d0375881554f1d3bc905c662070956bf

---
 library/cpp/yt/compact_containers/compact_flat_map.h | 2 +-
 library/cpp/yt/string/format-inl.h                   | 5 +++++
 library/cpp/yt/string/unittests/format_ut.cpp        | 1 +
 3 files changed, 7 insertions(+), 1 deletion(-)

(limited to 'library/cpp')

diff --git a/library/cpp/yt/compact_containers/compact_flat_map.h b/library/cpp/yt/compact_containers/compact_flat_map.h
index d4e93668430..49c8628c46c 100644
--- a/library/cpp/yt/compact_containers/compact_flat_map.h
+++ b/library/cpp/yt/compact_containers/compact_flat_map.h
@@ -20,7 +20,7 @@ concept CComparisonAllowed = std::same_as<T, U> || CHasIsTransparentFlag<TCompar
 
 ///////////////////////////////////////////////////////////////////////////////
 
-//! A flat map implementation over TCompactTValueector that tries to keep data inline.
+//! A flat map implementation over TCompactVector that tries to keep data inline.
 /*!
  *  Similarly to SmallSet, this is implemented via binary search over a sorted
  *  vector. Unlike SmallSet, however, this one never falls back to std::map (or
diff --git a/library/cpp/yt/string/format-inl.h b/library/cpp/yt/string/format-inl.h
index 83d61986745..22da423f893 100644
--- a/library/cpp/yt/string/format-inl.h
+++ b/library/cpp/yt/string/format-inl.h
@@ -11,6 +11,7 @@
 #include <library/cpp/yt/assert/assert.h>
 
 #include <library/cpp/yt/compact_containers/compact_vector.h>
+#include <library/cpp/yt/compact_containers/compact_flat_map.h>
 
 #include <library/cpp/yt/containers/enum_indexed_array.h>
 
@@ -163,6 +164,10 @@ template <class... Ts>
 constexpr bool CKnownKVRange<THashMap<Ts...>> = true;
 template <class... Ts>
 constexpr bool CKnownKVRange<THashMultiMap<Ts...>> = true;
+template <class... Ts>
+constexpr bool CKnownKVRange<TCompactFlatMap<Ts...>> = true;
+template <class K, class V, size_t N>
+constexpr bool CKnownKVRange<TCompactFlatMap<K, V, N>> = true;
 
 // TODO(arkady-e1ppa): Uncomment me when
 // https://github.com/llvm/llvm-project/issues/58534 is shipped.
diff --git a/library/cpp/yt/string/unittests/format_ut.cpp b/library/cpp/yt/string/unittests/format_ut.cpp
index f783efcf11a..e2e23c737c2 100644
--- a/library/cpp/yt/string/unittests/format_ut.cpp
+++ b/library/cpp/yt/string/unittests/format_ut.cpp
@@ -70,6 +70,7 @@ static_assert(CFormattable<std::multimap<int, int>>);
 static_assert(CFormattable<THashSet<int>>);
 static_assert(CFormattable<THashMap<int, int>>);
 static_assert(CFormattable<THashMultiSet<int>>);
+static_assert(CFormattable<TCompactFlatMap<int, int, 2>>);
 static_assert(CFormattable<std::pair<int, int>>);
 static_assert(CFormattable<std::optional<int>>);
 static_assert(CFormattable<TDuration>);
-- 
cgit v1.3


From a19aaf0e98329e8fff4e16626c2f8212f7d87789 Mon Sep 17 00:00:00 2001
From: artyasen <artyasen@yandex-team.com>
Date: Mon, 3 Feb 2025 15:57:35 +0300
Subject: Получение XML аттрибута без исключений
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

add no exception methods
commit_hash:245a52ca795a16ad57a9ac642b0cd00ca0122a32
---
 library/cpp/xml/document/node-attr.h         | 12 ++++++++++++
 library/cpp/xml/document/xml-document-decl.h | 18 ++++++++++++++++++
 library/cpp/xml/document/xml-document_ut.cpp | 22 ++++++++++++++++++++++
 3 files changed, 52 insertions(+)

(limited to 'library/cpp')

diff --git a/library/cpp/xml/document/node-attr.h b/library/cpp/xml/document/node-attr.h
index 1378ffdfbff..6f5445d4c2d 100644
--- a/library/cpp/xml/document/node-attr.h
+++ b/library/cpp/xml/document/node-attr.h
@@ -44,6 +44,18 @@ namespace NXml {
         return t;
     }
 
+    template <class T>
+    TMaybe<T> TNode::TryAttr(TZtStringBuf name) const {
+        TCharPtr value(xmlGetProp(NodePointer, XMLCHAR(name.c_str())));
+        if (!value) {
+            return Nothing();
+        }
+
+        T t;
+        AttrInternal(value, t, name);
+        return t;
+    }
+
     template <class T>
     T TNode::Attr(TZtStringBuf name, const T& defvalue) const {
         TCharPtr attr(xmlGetProp(NodePointer, XMLCHAR(name.c_str())));
diff --git a/library/cpp/xml/document/xml-document-decl.h b/library/cpp/xml/document/xml-document-decl.h
index bfda1fb7e6e..643ba664680 100644
--- a/library/cpp/xml/document/xml-document-decl.h
+++ b/library/cpp/xml/document/xml-document-decl.h
@@ -325,6 +325,14 @@ namespace NXml {
         template <class T>
         T Attr(TZtStringBuf name) const;
 
+        /**
+        * get node attribute
+        * @param name: attribute name
+        * returns value if exists
+        */
+        template <class T>
+        TMaybe<T> TryAttr(TZtStringBuf name) const;
+
         /**
         * get node attribute
         * @param name: attribute name
@@ -597,6 +605,16 @@ namespace NXml {
             return ActualNode.NextSibling();
         }
 
+        /**
+        * get node attribute
+        * @param name: attribute name
+        * returns value if exists
+        */
+        template <class T>
+        TMaybe<T> TryAttr(TZtStringBuf name) const {
+            return ActualNode.TryAttr<T>(name);
+        }
+
         /**
         * get node attribute
         * @param name: attribute name
diff --git a/library/cpp/xml/document/xml-document_ut.cpp b/library/cpp/xml/document/xml-document_ut.cpp
index 9f537b75c4c..0ec1fc60838 100644
--- a/library/cpp/xml/document/xml-document_ut.cpp
+++ b/library/cpp/xml/document/xml-document_ut.cpp
@@ -1,5 +1,6 @@
 #include <library/cpp/testing/unittest/registar.h>
 #include <util/generic/map.h>
+#include <util/generic/yexception.h>
 
 #include "xml-document.h"
 
@@ -35,6 +36,27 @@ Y_UNIT_TEST_SUITE(TestXmlDocument) {
         NXml::TConstNode text = root.Node("text");
         UNIT_ASSERT_EQUAL(text.Value<TString>(), "Некоторый текст");
     }
+    Y_UNIT_TEST(GetAttributes) {
+        NXml::TDocument xml(R"(<?xml version="1.0"?>
+                            <root>
+                                <a><b len="15" correct="1">hello world</b></a>
+                                <text>Некоторый текст</text>
+                            </root>
+                            )",
+                            NXml::TDocument::String);
+
+        NXml::TConstNode root = xml.Root();
+        NXml::TConstNode b = root.Node("a/b");
+
+        UNIT_ASSERT_EXCEPTION_CONTAINS(b.Attr<int>("unknown attrib"), yexception, "@unknown attrib");
+
+        const auto unknownAttr = b.TryAttr<int>("unknown attrib");
+        UNIT_ASSERT(unknownAttr.Empty());
+
+        const auto knownAttr = b.TryAttr<int>("len");
+        UNIT_ASSERT(knownAttr.Defined());
+        UNIT_ASSERT_EQUAL(knownAttr.GetRef(), 15);
+    }
     Y_UNIT_TEST(SerializeString) {
         NXml::TDocument xml("frob", NXml::TDocument::RootName);
         xml.Root().SetAttr("xyzzy", "Frobozz");
-- 
cgit v1.3