external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix #866, supports EXPLAIN on simple distributed statements

pull/899/head
Brian Cloutier 2016-10-21 15:19:21 +03:00
parent 97fa4d98ba
commit f3781ca66a
1 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
src/backend/distributed/executor/multi_utility.c
View File

@ -29,6 +29,7 @@
#include "catalog/pg_class.h"
#include "commands/defrem.h"
#include "commands/tablecmds.h"
#include "commands/prepare.h"
#include "distributed/citus_ruleutils.h"
#include "distributed/commit_protocol.h"
#include "distributed/connection_cache.h"
@ -181,23 +182,6 @@ multi_ProcessUtility(Node *parsetree,
return;
}
if (IsA(parsetree, ExplainStmt))
{
ExplainStmt *explainStmt = (ExplainStmt *) parsetree;
if (IsA(explainStmt->query, Query))
{
Query *query = (Query *) explainStmt->query;
if (query->commandType == CMD_UTILITY &&
IsA(query->utilityStmt, ExecuteStmt))
{
/* Due to a postgres limitation these cause crashes. Skip them for now */
ereport(ERROR, (errmsg("Citus does not support EXPLAIN EXECUTE")));
}
}
}
if (IsA(parsetree, CopyStmt))
{
parsetree = ProcessCopyStmt((CopyStmt *) parsetree, completionTag,
@ -316,6 +300,39 @@ multi_ProcessUtility(Node *parsetree,
" necessary users and roles.")));
}
/* due to an explain-hook limitation we have to special-case EXPLAIN EXECUTE */
if (IsA(parsetree, ExplainStmt) && IsA(((ExplainStmt*) parsetree)->query, Query))
{
ExplainStmt *explainStmt = (ExplainStmt *) parsetree;
Query *query = (Query *) explainStmt->query;
if (query->commandType == CMD_UTILITY &&
IsA(query->utilityStmt, ExecuteStmt))
{
ListCell *queryCell;
ExecuteStmt *execstmt = (ExecuteStmt *) query->utilityStmt;
PreparedStatement *entry = FetchPreparedStatement(execstmt->name, true);
/* copied from ExplainExecuteQuery, will never trigger if you used PREPARE */
if (!entry->plansource->fixed_result)
{
ereport(ERROR, (errmsg("EXPLAIN EXECUTE does not support variable-result"
" cached plans")));
}
foreach(queryCell, entry->plansource->query_list)
{
Query *query = (Query *) lfirst(queryCell);
explainStmt->query = (Node *) query;
standard_ProcessUtility(parsetree, query_string, context,
params, dest, completionTag);
}
return;
}
}
if (commandMustRunAsOwner)
{
GetUserIdAndSecContext(&savedUserId, &savedSecurityContext);