Fix assert in AssumeConstraints optimizer with _EnableYtDqProcessWriteConstraints flag

rvu1024 · rvu1024 · commit 53b0b8c5fa7f · 2025-04-11T20:45:28.000+03:00
commit_hash:c51295757f1f76396c25465f4e1f3b37ae47bc0b
diff --git a/yt/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_sort.cpp b/yt/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_sort.cpp
@@ -674,7 +674,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::AssumeConstraints(TExpr
         const size_t index = FromString(input.Cast<TYtOutput>().OutIndex().Value());
         TYtOutTableInfo outTable(op.Output().Item(index));
         if (builder) {
-            YQL_ENSURE(!builder->NeedMap());
+            YQL_ENSURE(!builder->NeedMap() || op.Maybe<TYtDqProcessWrite>());
             builder->FillRowSpecSort(*outTable.RowSpec);
         }
         outTable.RowSpec->SetConstraints(assume.Ref().GetConstraintSet());

Original file line number	Diff line number	Diff line change
`@@ -674,7 +674,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::AssumeConstraints(TExpr`
`674`	`674`	`const size_t index = FromString(input.Cast<TYtOutput>().OutIndex().Value());`
`675`	`675`	`TYtOutTableInfo outTable(op.Output().Item(index));`
`676`	`676`	`if (builder) {`
`677`		`- YQL_ENSURE(!builder->NeedMap());`
	`677`	`+ YQL_ENSURE(!builder->NeedMap() \|\| op.Maybe<TYtDqProcessWrite>());`
`678`	`678`	`builder->FillRowSpecSort(*outTable.RowSpec);`
`679`	`679`	`}`
`680`	`680`	`outTable.RowSpec->SetConstraints(assume.Ref().GetConstraintSet());`