aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authoraneporada <aneporada@ydb.tech>2023-03-05 13:52:12 +0300
committeraneporada <aneporada@ydb.tech>2023-03-05 13:52:12 +0300
commit6fdca79661a504f37905a5c344167f9c0442992d (patch)
tree8bfde1a8cd5a638f96d63abf0140c1da2b7814dc
parentdc697e5cf6f0cd4d1ff44614a4b1c09a50583d94 (diff)
downloadydb-6fdca79661a504f37905a5c344167f9c0442992d.tar.gz
Add helper for strict block udfs
-rw-r--r--ydb/library/yql/public/udf/arrow/udf_arrow_helpers.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/ydb/library/yql/public/udf/arrow/udf_arrow_helpers.h b/ydb/library/yql/public/udf/arrow/udf_arrow_helpers.h
index 266e0f8a12d..50f33a629e5 100644
--- a/ydb/library/yql/public/udf/arrow/udf_arrow_helpers.h
+++ b/ydb/library/yql/public/udf/arrow/udf_arrow_helpers.h
@@ -373,6 +373,10 @@ public:
BEGIN_ARROW_UDF(udfName##_BlocksImpl, signatureFunc) \
UDF(udfName, builder.SimpleSignature<signatureFunc>().SupportsBlocks();)
+#define BEGIN_SIMPLE_STRICT_ARROW_UDF(udfName, signatureFunc) \
+ BEGIN_ARROW_UDF(udfName##_BlocksImpl, signatureFunc) \
+ UDF(udfName, builder.SimpleSignature<signatureFunc>().SupportsBlocks().IsStrict();)
+
#define END_ARROW_UDF(udfNameBlocks, exec) \
inline bool udfNameBlocks::DeclareSignature(\
const ::NYql::NUdf::TStringRef& name, \