mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-10 01:25:21 +00:00
revert
This commit is contained in:
parent
0084a61c76
commit
530c9ae490
@ -504,6 +504,9 @@ QueryPipeline InterpreterExplainQuery::executeImpl()
|
||||
auto pipe = QueryPipelineBuilder::getPipe(std::move(*pipeline), resources);
|
||||
const auto & processors = pipe.getProcessors();
|
||||
|
||||
if (settings.compact)
|
||||
printPipelineCompact(processors, buf, settings.query_pipeline_options.header);
|
||||
else
|
||||
printPipeline(processors, buf);
|
||||
}
|
||||
else
|
||||
|
173
src/QueryPipeline/printPipeline.cpp
Normal file
173
src/QueryPipeline/printPipeline.cpp
Normal file
@ -0,0 +1,173 @@
|
||||
#include <map>
|
||||
#include <set>
|
||||
#include <Processors/QueryPlan/IQueryPlanStep.h>
|
||||
#include <QueryPipeline/printPipeline.h>
|
||||
|
||||
namespace DB
|
||||
{
|
||||
|
||||
void printPipelineCompact(const Processors & processors, WriteBuffer & out, bool with_header)
|
||||
{
|
||||
struct Node;
|
||||
|
||||
/// Group by processors name, QueryPlanStep and group in this step.
|
||||
struct Key
|
||||
{
|
||||
size_t group;
|
||||
IQueryPlanStep * step;
|
||||
std::string name;
|
||||
|
||||
auto getTuple() const { return std::forward_as_tuple(group, step, name); }
|
||||
|
||||
bool operator<(const Key & other) const { return getTuple() < other.getTuple(); }
|
||||
};
|
||||
|
||||
/// Group ports by header.
|
||||
struct EdgeData
|
||||
{
|
||||
Block header;
|
||||
size_t count;
|
||||
};
|
||||
|
||||
using Edge = std::vector<EdgeData>;
|
||||
|
||||
struct Node
|
||||
{
|
||||
size_t id = 0;
|
||||
std::map<Node *, Edge> edges = {};
|
||||
std::vector<const IProcessor *> agents = {};
|
||||
};
|
||||
|
||||
std::map<Key, Node> graph;
|
||||
|
||||
auto get_key = [](const IProcessor & processor) {
|
||||
return Key{processor.getQueryPlanStepGroup(), processor.getQueryPlanStep(), processor.getName()};
|
||||
};
|
||||
|
||||
/// Fill nodes.
|
||||
for (const auto & processor : processors)
|
||||
{
|
||||
auto res = graph.emplace(get_key(*processor), Node());
|
||||
auto & node = res.first->second;
|
||||
node.agents.emplace_back(processor.get());
|
||||
|
||||
if (res.second)
|
||||
node.id = graph.size();
|
||||
}
|
||||
|
||||
Block empty_header;
|
||||
|
||||
/// Fill edges.
|
||||
for (const auto & processor : processors)
|
||||
{
|
||||
auto & from = graph[get_key(*processor)];
|
||||
|
||||
for (auto & port : processor->getOutputs())
|
||||
{
|
||||
if (!port.isConnected())
|
||||
continue;
|
||||
|
||||
auto & to = graph[get_key(port.getInputPort().getProcessor())];
|
||||
auto & edge = from.edges[&to];
|
||||
|
||||
/// Use empty header for each edge if with_header is false.
|
||||
const auto & header = with_header ? port.getHeader() : empty_header;
|
||||
|
||||
/// Group by header.
|
||||
bool found = false;
|
||||
for (auto & item : edge)
|
||||
{
|
||||
if (blocksHaveEqualStructure(header, item.header))
|
||||
{
|
||||
found = true;
|
||||
++item.count;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!found)
|
||||
edge.emplace_back(EdgeData{header, 1});
|
||||
}
|
||||
}
|
||||
|
||||
/// Group processors by it's QueryPlanStep.
|
||||
std::map<IQueryPlanStep *, std::vector<const Node *>> steps_map;
|
||||
|
||||
for (const auto & item : graph)
|
||||
steps_map[item.first.step].emplace_back(&item.second);
|
||||
|
||||
out << "digraph\n{\n";
|
||||
out << " rankdir=\"LR\";\n";
|
||||
out << " { node [shape = rect]\n";
|
||||
|
||||
/// Nodes // TODO quoting and escaping
|
||||
size_t next_step = 0;
|
||||
for (const auto & item : steps_map)
|
||||
{
|
||||
/// Use separate clusters for each step.
|
||||
if (item.first != nullptr)
|
||||
{
|
||||
out << " subgraph cluster_" << next_step << " {\n";
|
||||
out << " label =\"" << item.first->getName() << "\";\n";
|
||||
out << " style=filled;\n";
|
||||
out << " color=lightgrey;\n";
|
||||
out << " node [style=filled,color=white];\n";
|
||||
out << " { rank = same;\n";
|
||||
|
||||
++next_step;
|
||||
}
|
||||
|
||||
for (const auto & node : item.second)
|
||||
{
|
||||
const auto & processor = node->agents.front();
|
||||
out << " n" << node->id << " [label=\"" << processor->getName();
|
||||
|
||||
if (node->agents.size() > 1)
|
||||
out << " × " << node->agents.size();
|
||||
|
||||
const auto & description = processor->getDescription();
|
||||
if (!description.empty())
|
||||
out << ' ' << description;
|
||||
|
||||
out << "\"];\n";
|
||||
}
|
||||
|
||||
if (item.first != nullptr)
|
||||
{
|
||||
out << " }\n";
|
||||
out << " }\n";
|
||||
}
|
||||
}
|
||||
|
||||
out << " }\n";
|
||||
|
||||
/// Edges
|
||||
for (const auto & item : graph)
|
||||
{
|
||||
for (const auto & edge : item.second.edges)
|
||||
{
|
||||
for (const auto & data : edge.second)
|
||||
{
|
||||
out << " n" << item.second.id << " -> "
|
||||
<< "n" << edge.first->id << " [label=\"";
|
||||
|
||||
if (data.count > 1)
|
||||
out << "× " << data.count;
|
||||
|
||||
if (with_header)
|
||||
{
|
||||
for (const auto & elem : data.header)
|
||||
{
|
||||
out << "\n";
|
||||
elem.dumpStructure(out);
|
||||
}
|
||||
}
|
||||
|
||||
out << "\"];\n";
|
||||
}
|
||||
}
|
||||
}
|
||||
out << "}\n";
|
||||
}
|
||||
|
||||
}
|
@ -64,4 +64,9 @@ void printPipeline(const Processors & processors, WriteBuffer & out)
|
||||
printPipeline(processors, std::vector<IProcessor::Status>(), out);
|
||||
}
|
||||
|
||||
/// Prints pipeline in compact representation.
|
||||
/// Group processors by it's name, QueryPlanStep and QueryPlanStepGroup.
|
||||
/// If QueryPlanStep wasn't set for processor, representation may be not correct.
|
||||
/// If with_header is set, prints block header for each edge.
|
||||
void printPipelineCompact(const Processors & processors, WriteBuffer & out, bool with_header);
|
||||
}
|
||||
|
@ -1,7 +0,0 @@
|
||||
-- The server does not crash after these queries:
|
||||
|
||||
DROP TABLE IF EXISTS t1;
|
||||
CREATE TABLE t1(ID UInt64, name String) engine=MergeTree order by ID;
|
||||
insert into t1(ID, name) values (1, 'abc'), (2, 'bbb');
|
||||
explain pipeline graph=1 select count(ID) from t1 FORMAT Null;
|
||||
DROP TABLE t1;
|
Loading…
Reference in New Issue
Block a user