ActionsDag: add allow_remove_inputs for other variants of removeUnusedActions()

This commit is contained in:
Azat Khuzhin 2021-10-22 00:15:59 +03:00
parent 874f7a5052
commit 248b689067
2 changed files with 6 additions and 6 deletions

View File

@ -325,7 +325,7 @@ std::string ActionsDAG::dumpNames() const
return out.str();
}
void ActionsDAG::removeUnusedActions(const NameSet & required_names)
void ActionsDAG::removeUnusedActions(const NameSet & required_names, bool allow_remove_inputs)
{
NodeRawConstPtrs required_nodes;
required_nodes.reserve(required_names.size());
@ -349,10 +349,10 @@ void ActionsDAG::removeUnusedActions(const NameSet & required_names)
}
index.swap(required_nodes);
removeUnusedActions();
removeUnusedActions(allow_remove_inputs);
}
void ActionsDAG::removeUnusedActions(const Names & required_names)
void ActionsDAG::removeUnusedActions(const Names & required_names, bool allow_remove_inputs)
{
NodeRawConstPtrs required_nodes;
required_nodes.reserve(required_names.size());
@ -372,7 +372,7 @@ void ActionsDAG::removeUnusedActions(const Names & required_names)
}
index.swap(required_nodes);
removeUnusedActions();
removeUnusedActions(allow_remove_inputs);
}
void ActionsDAG::removeUnusedActions(bool allow_remove_inputs)

View File

@ -162,8 +162,8 @@ public:
bool isInputProjected() const { return project_input; }
bool isOutputProjected() const { return projected_output; }
void removeUnusedActions(const Names & required_names);
void removeUnusedActions(const NameSet & required_names);
void removeUnusedActions(const Names & required_names, bool allow_remove_inputs = true);
void removeUnusedActions(const NameSet & required_names, bool allow_remove_inputs = true);
NameSet foldActionsByProjection(
const NameSet & required_columns,