Helpers for Void & Set

1 files changed, 24 insertions, 0 deletions

diff --git a/ydb/library/yql/public/udf/udf_type_builder.h b/ydb/library/yql/public/udf/udf_type_builder.h
index 3b8c0f7fe2..8d7d5b4d4c 100644
--- a/ydb/library/yql/public/udf/udf_type_builder.h
+++ b/ydb/library/yql/public/udf/udf_type_builder.h
@@ -43,6 +43,11 @@ struct TDict {
     using ValueType = TValue;
 };
 
+template <typename TKey>
+struct TSetType {
+    using KeyType = TKey;
+};
+
 template <typename... TArgs>
 struct TTuple;
 
@@ -67,6 +72,8 @@ struct TBlockType { using ItemType = T; };
 template <typename T>
 struct TScalarType { using ItemType = T; };
 
+struct TVoid {};
+
 //////////////////////////////////////////////////////////////////////////////
 // ITypeBuilder
 //////////////////////////////////////////////////////////////////////////////
@@ -874,6 +881,23 @@ struct TTypeBuilderHelper<TDict<TKey, TValue>> {
     }
 };
 
+template <typename TKey>
+struct TTypeBuilderHelper<TSetType<TKey>> {
+    static TType* Build(const IFunctionTypeInfoBuilder& builder) {
+        return builder.Dict()->
+                Key(TTypeBuilderHelper<TKey>::Build(builder))
+                .Value(builder.Void())
+                .Build();
+    }
+};
+
+template <>
+struct TTypeBuilderHelper<TVoid> {
+    static TType* Build(const IFunctionTypeInfoBuilder& builder) {
+        return builder.Void();
+    }
+};
+
 template <typename T>
 struct TTypeBuilderHelper<TStream<T>> {
     static TType* Build(const IFunctionTypeInfoBuilder& builder) {