diff options
author | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-12-14 09:27:42 +0300 |
---|---|---|
committer | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-12-14 09:42:38 +0300 |
commit | 522749875c10848a48889ae6649c819defb4b61d (patch) | |
tree | b93768e19a813ac412eb543104a99e347305beee /yt | |
parent | 63778bd0984881908635fed0538928cab197c3e2 (diff) | |
download | ydb-522749875c10848a48889ae6649c819defb4b61d.tar.gz |
[yt provider] Fix YtReduce type annotation with aux sort columns
YQL-19373
commit_hash:8097d935a27fdc5e71ccee6855681514dcef892b
Diffstat (limited to 'yt')
-rw-r--r-- | yt/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/yt/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp b/yt/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp index 4f53f48424..b422af1bf2 100644 --- a/yt/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp +++ b/yt/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp @@ -250,7 +250,7 @@ private: return TStatus::Ok; } - static bool ValidateOutputType(const TTypeAnnotationNode* itemType, TPositionHandle positionHandle, TYtOutSection outTables, + static bool ValidateOutputType(const TTypeAnnotationNode* itemType, TPositionHandle positionHandle, const TYtOutSection& outTables, size_t beginIdx, size_t endIdx, bool useExtendedType, TExprContext& ctx) { YQL_ENSURE(beginIdx <= endIdx); @@ -352,13 +352,13 @@ private: return true; } - static bool ValidateOutputType(const TTypeAnnotationNode* itemType, TPositionHandle positionHandle, TYtOutSection outTables, + static bool ValidateOutputType(const TTypeAnnotationNode* itemType, TPositionHandle positionHandle, const TYtOutSection& outTables, TExprContext& ctx, bool useExtendedType = false) { return ValidateOutputType(itemType, positionHandle, outTables, 0, outTables.Ref().ChildrenSize(), useExtendedType, ctx); } - static bool ValidateOutputType(const TExprNode& list, TYtOutSection outTables, TExprContext& ctx, bool useExtendedType = false) { + static bool ValidateOutputType(const TExprNode& list, const TYtOutSection& outTables, TExprContext& ctx, bool useExtendedType = false) { const TTypeAnnotationNode* itemType = GetSequenceItemType(list.Pos(), list.GetTypeAnn(), true, ctx); if (nullptr == itemType) { return false; @@ -1144,7 +1144,7 @@ private: return TStatus::Error; } - if (!ValidateOutputType(*lambda, map.Output(), ctx, true)) { + if (!ValidateOutputType(*lambda, map.Output(), ctx, /*useExtendedType*/true)) { lambda->SetTypeAnn(nullptr); lambda->SetState(TExprNode::EState::Initial); return TStatus::Error; @@ -1260,7 +1260,7 @@ private: return TStatus::Error; } - if (!ValidateOutputType(*lambda, reduce.Output(), ctx)) { + if (!ValidateOutputType(*lambda, reduce.Output(), ctx, /*useExtendedType*/true)) { lambda->SetTypeAnn(nullptr); lambda->SetState(TExprNode::EState::Initial); return TStatus::Error; @@ -1638,7 +1638,7 @@ private: return TStatus::Error; } - if (!ValidateOutputType(lambda.Ref(), fill.Output(), ctx, true)) { + if (!ValidateOutputType(lambda.Ref(), fill.Output(), ctx, /*useExtendedType*/true)) { return TStatus::Error; } |