Fix spilling UT

4 files changed, 391 insertions, 2 deletions

diff --git a/library/cpp/json/converter/converter.cpp b/library/cpp/json/converter/converter.cpp
new file mode 100644
index 0000000000..f61fd9055c
--- /dev/null
+++ b/library/cpp/json/converter/converter.cpp
@@ -0,0 +1 @@
+#include "converter.h"
diff --git a/library/cpp/json/converter/converter.h b/library/cpp/json/converter/converter.h
new file mode 100644
index 0000000000..be86c31595
--- /dev/null
+++ b/library/cpp/json/converter/converter.h
@@ -0,0 +1,176 @@
+#include "library/cpp/json/writer/json_value.h"
+
+#include <limits>
+#include <util/generic/array_ref.h>
+#include <util/generic/deque.h>
+#include <util/generic/hash.h>
+#include <util/generic/list.h>
+#include <util/generic/map.h>
+#include <util/generic/maybe.h>
+
+
+namespace NJson {
+    template<typename T>
+    struct TConverter {
+    };
+
+    namespace {
+        template<typename T>
+        struct TDefaultEncoder {
+            static inline TJsonValue Encode(T value) {
+                return TJsonValue(value);
+            }
+        };
+
+        template<typename T, typename E>
+        struct TDefaultArrayEncoder {
+            static TJsonValue Encode(const T& value) {
+                TJsonValue result(NJson::JSON_ARRAY);
+                auto& encodedArray = result.GetArraySafe();
+                for (const auto& element : value) {
+                    encodedArray.push_back(TConverter<E>::Encode(element));
+                }
+                return result;
+            }
+        };
+
+        template<typename T, typename E>
+        struct TDefaultArrayDecoder {
+            static T Decode(const TJsonValue& value) {
+                T result;
+                for (const auto& element : value.GetArraySafe()) {
+                    result.push_back(TConverter<E>::Decode(element));
+                }
+                return result;
+            }
+        };
+
+        template<typename T, typename E>
+        struct TDefaultArrayConverter: public TDefaultArrayEncoder<T, E>, public TDefaultArrayDecoder<T, E> {
+        };
+
+        template<typename T, typename E>
+        struct TDefaultMapEncoder {
+            static TJsonValue Encode(const T& value) {
+                TJsonValue result(NJson::JSON_MAP);
+                auto& encodedMap = result.GetMapSafe();
+                for (const auto& [key, element] : value) {
+                    encodedMap[key] = TConverter<E>::Encode(element);
+                }
+                return result;
+            }
+        };
+
+        template<typename T, typename E>
+        struct TDefaultMapDecoder {
+            static T Decode(const TJsonValue& value) {
+                T result;
+                for (const auto& [key, element] : value.GetMapSafe()) {
+                    result[key] = TConverter<E>::Decode(element);
+                }
+                return result;
+            }
+        };
+
+        template<typename T, typename E>
+        struct TDefaultMapConverter: public TDefaultMapEncoder<T, E>, public TDefaultMapDecoder<T, E> {
+        };
+    }
+
+    template<>
+    struct TConverter<bool>: public TDefaultEncoder<bool> {
+        static inline bool Decode(const TJsonValue& value) {
+            return value.GetBooleanSafe();
+        }
+    };
+
+    template<typename T>
+    requires std::is_integral_v<T> && (!std::is_same_v<T, bool>)
+    struct TConverter<T>: public TDefaultEncoder<T> {
+        static T Decode(const TJsonValue& value) {
+            if constexpr (std::is_signed_v<T>) {
+                const auto decodedValue = value.GetIntegerSafe();
+                if (decodedValue < std::numeric_limits<T>::min() || std::numeric_limits<T>::max() < decodedValue) {
+                    ythrow yexception() << "Out of range (got " << decodedValue << ")";
+                }
+                return static_cast<T>(decodedValue);
+            } else {
+                const auto decodedValue = value.GetUIntegerSafe();
+                if (std::numeric_limits<T>::max() < decodedValue) {
+                    ythrow yexception() << "Out of range (got " << decodedValue << ")";
+                }
+                return static_cast<T>(decodedValue);
+            }
+        }
+    };
+
+    template<typename T>
+    requires std::is_floating_point_v<T>
+    struct TConverter<T>: public TDefaultEncoder<T> {
+        static inline T Decode(const TJsonValue& value) {
+            return static_cast<T>(value.GetDoubleSafe());
+        }
+    };
+
+    template<>
+    struct TConverter<TStringBuf>: public TDefaultEncoder<TStringBuf> {
+    };
+
+    template<>
+    struct TConverter<TString>: public TDefaultEncoder<TString> {
+        static inline TString Decode(const TJsonValue& value) {
+            return value.GetStringSafe();
+        }
+    };
+
+    template<typename T>
+    struct TConverter<TMaybe<T>> {
+        static TJsonValue Encode(const TMaybe<T>& value) {
+            if (value.Defined()) {
+                return TConverter<T>::Encode(*value);
+            } else {
+                return TJsonValue(NJson::JSON_NULL);
+            }
+        }
+
+        static TMaybe<T> Decode(const TJsonValue& value) {
+            if (value.IsDefined()) {
+                return TConverter<T>::Decode(value);
+            } else {
+                return Nothing();
+            }
+        }
+    };
+
+    template<typename T>
+    struct TConverter<TArrayRef<T>>: public TDefaultArrayEncoder<TArrayRef<T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<TVector<T>>: public TDefaultArrayConverter<TVector<T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<TList<T>>: public TDefaultArrayConverter<TList<T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<TDeque<T>>: public TDefaultArrayConverter<TDeque<T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<THashMap<TStringBuf, T>>: public TDefaultMapEncoder<THashMap<TStringBuf, T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<THashMap<TString, T>>: public TDefaultMapConverter<THashMap<TString, T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<TMap<TStringBuf, T>>: public TDefaultMapEncoder<TMap<TStringBuf, T>, T> {
+    };
+
+    template<typename T>
+    struct TConverter<TMap<TString, T>>: public TDefaultMapConverter<TMap<TString, T>, T> {
+    };
+}
diff --git a/library/cpp/json/converter/ut/test_conversion.cpp b/library/cpp/json/converter/ut/test_conversion.cpp
new file mode 100644
index 0000000000..c2c857835a
--- /dev/null
+++ b/library/cpp/json/converter/ut/test_conversion.cpp
@@ -0,0 +1,212 @@
+#include "library/cpp/json/converter/converter.h"
+#include <library/cpp/json/json_reader.h>
+#include <library/cpp/json/json_writer.h>
+#include "library/cpp/json/writer/json_value.h"
+#include <library/cpp/testing/unittest/registar.h>
+
+#include <util/generic/array_ref.h>
+#include <util/generic/deque.h>
+#include <util/generic/hash.h>
+#include <util/generic/list.h>
+#include <util/generic/map.h>
+#include <util/generic/maybe.h>
+
+
+namespace NJson {
+    void AssertJsonsEqual(const TJsonValue& actualJson, const TJsonValue& expectedJson) {
+        const auto actualString = WriteJson(actualJson, /*formatOutput*/ true, /*sortkeys*/ true, /*validateUtf8*/ true);
+        const auto expectedString = WriteJson(expectedJson, /*formatOutput*/ true, /*sortkeys*/ true, /*validateUtf8*/ true);
+        UNIT_ASSERT_NO_DIFF(actualString, expectedString);
+    }
+
+    void AssertJsonsEqual(const TJsonValue& actualJson, TStringBuf expectedString) {
+        const auto expectedJson = ReadJsonFastTree(expectedString);
+        AssertJsonsEqual(actualJson, expectedJson);
+    }
+
+    template<typename T>
+    struct TConverterTester {
+        using TValues = THashMap<TStringBuf, T>;
+
+        static void TestEncoding(const TValues& values) {
+            for (const auto& [serializedValue, value] : values) {
+                {
+                    const auto encodedValue = TConverter<T>::Encode(value);
+                    AssertJsonsEqual(encodedValue, serializedValue);
+                }
+            }
+        }
+
+        static void TestDecoding(const TValues& values) {
+            for (const auto& [serializedValue, value] : values) {
+                {
+                    const auto decodedValue = TConverter<T>::Decode(ReadJsonFastTree(serializedValue));
+                    UNIT_ASSERT_EQUAL(decodedValue, value);
+                }
+            }
+        }
+
+        static void TestDecodingException(TStringBuf serializedValue) {
+            try {
+                TConverter<T>::Decode(ReadJsonFastTree(serializedValue));
+                UNIT_ASSERT(false);
+            } catch (...) {
+            }
+        }
+
+        static void Test(const TValues& values) {
+            TestEncoding(values);
+            TestDecoding(values);
+
+            for (const auto& [serializedValue, value] : values) {
+                {
+                    const auto encodedValue = TConverter<T>::Encode(value);
+                    const auto decodedValue = TConverter<T>::Decode(encodedValue);
+                    UNIT_ASSERT_EQUAL(value, decodedValue);
+                }
+            }
+        }
+    };
+
+    template<typename T>
+    requires std::is_floating_point_v<T>
+    struct TConverterTester<T> {
+        using TValues = THashMap<TStringBuf, T>;
+
+        static void TestDecoding(const TValues& values) {
+            for (const auto& [serializedValue, value] : values) {
+                {
+                    const auto decodedValue = TConverter<T>::Decode(ReadJsonFastTree(serializedValue));
+                    UNIT_ASSERT_DOUBLES_EQUAL(decodedValue, value, 0.000001);
+                }
+            }
+        }
+
+        static void Test(const TValues& values) {
+            TestDecoding(values);
+
+            for (const auto& [serializedValue, value] : values) {
+                {
+                    const auto encodedValue = TConverter<T>::Encode(value);
+                    const auto decodedValue = TConverter<T>::Decode(encodedValue);
+                    UNIT_ASSERT_DOUBLES_EQUAL(decodedValue, value, 0.000001);
+                }
+            }
+        }
+    };
+
+    Y_UNIT_TEST_SUITE(ConversionTests) {
+        Y_UNIT_TEST(PrimitivesTest) {
+            TConverterTester<bool>::Test({{"true", true}, {"false", false}});
+
+            TConverterTester<ui8>::Test({{"0", 0}, {"255", 255}});
+            TConverterTester<i8>::Test({{"-128", -128}, {"127", 127}});
+            TConverterTester<ui16>::Test({{"0", 0}, {"65535", 65535}});
+            TConverterTester<i16>::Test({{"-32768", -32768}, {"32767", 32767}});
+            TConverterTester<ui32>::Test({{"0", 0}, {"4294967295", 4294967295}});
+            TConverterTester<i32>::Test({{"-2147483648", -2147483648}, {"2147483647", 2147483647}});
+            TConverterTester<ui64>::Test({{"0", 0}, {"18446744073709551615", 18446744073709551615u}});
+            TConverterTester<i64>::Test({
+                {"-9223372036854775808", -9223372036854775808u},
+                {"9223372036854775807", 9223372036854775807},
+            });
+
+            TConverterTester<i8>::TestDecodingException("128");
+            TConverterTester<i8>::TestDecodingException("-129");
+            TConverterTester<ui8>::TestDecodingException("256");
+            TConverterTester<i16>::TestDecodingException("32768");
+            TConverterTester<i16>::TestDecodingException("-32769");
+            TConverterTester<ui16>::TestDecodingException("65536");
+            TConverterTester<i32>::TestDecodingException("-2147483649");
+            TConverterTester<i32>::TestDecodingException("2147483649");
+            TConverterTester<ui32>::TestDecodingException("4294967296");
+
+            TConverterTester<unsigned char>::Test({{"0", 0}, {"255", 255}});
+            TConverterTester<signed char>::Test({{"-128", -128}, {"127", 127}});
+            TConverterTester<unsigned short int>::Test({{"0", 0}, {"65535", 65535}});
+            TConverterTester<signed short int>::Test({{"-32768", -32768}, {"32767", 32767}});
+            TConverterTester<unsigned int>::Test({{"0", 0}, {"65535", 65535}});
+            TConverterTester<signed int>::Test({{"-32768", -32768}, {"32767", 32767}});
+            TConverterTester<unsigned long int>::Test({{"0", 0}, {"4294967295", 4294967295}});
+            TConverterTester<signed long int>::Test({{"-2147483648", -2147483648}, {"2147483647", 2147483647}});
+            TConverterTester<unsigned long long int>::Test({{"0", 0}, {"18446744073709551615", 18446744073709551615u}});
+            TConverterTester<signed long long int>::Test({
+                {"-9223372036854775808", -9223372036854775808u},
+                {"9223372036854775807", 9223372036854775807},
+            });
+
+            TConverterTester<size_t>::Test({{"0", 0}, {"65535", 65535}});
+
+            TConverterTester<float>::Test({{"-1.12312", -1.12312}, {"3434.25674", 3434.25674}});
+            TConverterTester<double>::Test({{"-1.12312E+42", -1.12312E+42}, {"3.25E+120", 3.25E+120}});
+        }
+
+        Y_UNIT_TEST(StringsTest) {
+            TConverterTester<TStringBuf>::TestEncoding({
+                {R"("Let's permit using of Rust in Arcadia")", "Let's permit using of Rust in Arcadia"},
+            });
+            TConverterTester<TString>::Test({
+                {
+                    R"("Всякое непрерывное отображение замкнутого n-мерного шара в себя обладает неподвижной точкой")",
+                    "Всякое непрерывное отображение замкнутого n-мерного шара в себя обладает неподвижной точкой",
+                },
+            });
+        }
+
+        Y_UNIT_TEST(MaybeTest) {
+            TConverterTester<TMaybe<bool>>::Test({
+                {"true", TMaybe<bool>(true)},
+                {"null", Nothing()},
+                {"false", TMaybe<bool>(false)},
+            });
+        }
+
+        Y_UNIT_TEST(ArraysTest) {
+            TConverterTester<TVector<bool>>::Test({{"[true, true, false]", {true, true, false}}});
+            TConverterTester<TList<TString>>::Test({{R"(["a", "b"])", {"a", "b"}}});
+            TConverterTester<TDeque<bool>>::Test({{"[false, true, false]", {false, true, false}}});
+        }
+
+        Y_UNIT_TEST(MapsTest) {
+            TConverterTester<THashMap<TStringBuf, bool>>::TestEncoding({
+                {R"({"a": true, "b": false})", {{"a", true}, {"b", false}}},
+            });
+            TConverterTester<THashMap<TString, bool>>::Test({
+                {R"({"a": true, "b": false})", {{"a", true}, {"b", false}}},
+            });
+            TConverterTester<TMap<TStringBuf, TStringBuf>>::TestEncoding({
+                {R"({"a": "A", "b": "B"})", {{"a", "A"}, {"b", "B"}}},
+            });
+            TConverterTester<TMap<TString, TString>>::Test({
+                {R"({"a": "A", "b": "B"})", {{"a", "A"}, {"b", "B"}}},
+            });
+        }
+
+        Y_UNIT_TEST(NestedTest) {
+            TConverterTester<TVector<THashMap<TString, TVector<ui64>>>>::Test({
+                {
+                    R"([
+                        {
+                            "Three": [0, 1, 2],
+                            "Five": [0, 1, 2, 3, 4]
+                        },
+                        {
+                            "Four": [0, 1, 2, 3],
+                            "Six": [0, 1, 2, 3, 4, 5]
+                        },
+                    ])",
+                    {
+                        {
+                            {"Three", {0, 1, 2}},
+                            {"Five", {0, 1, 2, 3, 4}},
+                        },
+                        {
+                            {"Four", {0, 1, 2, 3}},
+                            {"Six", {0, 1, 2, 3, 4, 5}},
+                        },
+                    },
+                },
+            });
+        }
+    }
+}
diff --git a/ydb/core/kqp/runtime/kqp_spilling_file_ut.cpp b/ydb/core/kqp/runtime/kqp_spilling_file_ut.cpp
index c81956398f..6774ad85aa 100644
--- a/ydb/core/kqp/runtime/kqp_spilling_file_ut.cpp
+++ b/ydb/core/kqp/runtime/kqp_spilling_file_ut.cpp
@@ -225,7 +225,7 @@ Y_UNIT_TEST(MultipleFileParts) {
 
     WaitBootstrap(runtime);
 
-    const TString filePrefix = TStringBuilder() << NFs::CurrentWorkingDirectory() << "/kqp_spilling/node_1/1_test_";
+    const TString filePrefix = TStringBuilder() << NFs::CurrentWorkingDirectory() << "/kqp_spilling/node_" << runtime.GetNodeId() << "/1_test_";
 
     for (ui32 i = 0; i < 5; ++i) {
         // Cerr << "---- store blob #" << i << Endl;
@@ -268,7 +268,7 @@ Y_UNIT_TEST(SingleFilePart) {
 
     WaitBootstrap(runtime);
 
-    const TString filePrefix = TStringBuilder() << NFs::CurrentWorkingDirectory() << "/kqp_spilling/node_1/1_test_";
+    const TString filePrefix = TStringBuilder() << NFs::CurrentWorkingDirectory() << "/kqp_spilling/node_" << runtime.GetNodeId() << "/1_test_";
 
     for (ui32 i = 0; i < 5; ++i) {
         // Cerr << "---- store blob #" << i << Endl;