File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -826,15 +826,17 @@ class TxPlanSerializer {
826
826
}
827
827
}
828
828
for (const auto & child : node->Children ()) {
829
- std::shared_ptr<TOptimizerStatistics> result;
830
829
if (child->IsLambda ()) {
831
- auto lambda = TExprBase (child).Cast <TCoLambda>();
832
- result = FindWrapStats (lambda.Body ().Ptr (), dataSourceNode);
830
+ // support wide lambda as well
831
+ for (size_t bodyIndex = 1 ; bodyIndex < child->ChildrenSize (); ++bodyIndex) {
832
+ if (auto result = FindWrapStats (child->ChildPtr (bodyIndex), dataSourceNode)) {
833
+ return result;
834
+ }
835
+ }
833
836
} else {
834
- result = FindWrapStats (child, dataSourceNode);
835
- }
836
- if (result) {
837
- return result;
837
+ if (auto result = FindWrapStats (child, dataSourceNode)) {
838
+ return result;
839
+ }
838
840
}
839
841
}
840
842
return nullptr ;
You can’t perform that action at this time.
0 commit comments