mirror of https://github.com/citusdata/citus.git
Annotate all queries included in a task
parent
81cf3cf408
commit
144915880d
|
@ -403,8 +403,7 @@ SetTaskQueryIfShouldLazyDeparse(Task *task, Query *query)
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
SetTaskQueryString(task, AnnotateQuery(DeparseTaskQuery(task, query),
|
SetTaskQueryString(task, DeparseTaskQuery(task, query));
|
||||||
task->partitionColumn, task->colocationId));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -424,7 +423,7 @@ SetTaskQueryString(Task *task, char *queryString)
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
task->taskQuery.queryType = TASK_QUERY_TEXT;
|
task->taskQuery.queryType = TASK_QUERY_TEXT;
|
||||||
task->taskQuery.data.queryStringLazy = queryString;
|
task->taskQuery.data.queryStringLazy = AnnotateQuery(queryString, task->partitionColumn, task->colocationId);
|
||||||
task->queryCount = 1;
|
task->queryCount = 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -438,6 +437,15 @@ SetTaskQueryStringList(Task *task, List *queryStringList)
|
||||||
{
|
{
|
||||||
Assert(queryStringList != NIL);
|
Assert(queryStringList != NIL);
|
||||||
task->taskQuery.queryType = TASK_QUERY_TEXT_LIST;
|
task->taskQuery.queryType = TASK_QUERY_TEXT_LIST;
|
||||||
|
|
||||||
|
|
||||||
|
char *queryString = NULL;
|
||||||
|
foreach_ptr(queryString, queryStringList)
|
||||||
|
{
|
||||||
|
queryString = AnnotateQuery(queryString, task->partitionColumn, task->colocationId);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
task->taskQuery.data.queryStringList = queryStringList;
|
task->taskQuery.data.queryStringList = queryStringList;
|
||||||
task->queryCount = list_length(queryStringList);
|
task->queryCount = list_length(queryStringList);
|
||||||
}
|
}
|
||||||
|
@ -552,5 +560,6 @@ TaskQueryString(Task *task)
|
||||||
char *queryString = DeparseTaskQuery(task, jobQueryReferenceForLazyDeparsing);
|
char *queryString = DeparseTaskQuery(task, jobQueryReferenceForLazyDeparsing);
|
||||||
MemoryContextSwitchTo(previousContext);
|
MemoryContextSwitchTo(previousContext);
|
||||||
SetTaskQueryString(task, queryString);
|
SetTaskQueryString(task, queryString);
|
||||||
|
|
||||||
return task->taskQuery.data.queryStringLazy;
|
return task->taskQuery.data.queryStringLazy;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue