def from_simple_query_plan(
cls, query_plan: ClickhouseQueryPlan) -> CompositeDataSourcePlan:
return CompositeDataSourcePlan(
translated_source=query_plan.query,
root_processors=SubqueryProcessors(
plan_processors=query_plan.plan_query_processors,
db_processors=query_plan.db_query_processors,
),
storage_set_key=query_plan.storage_set_key,
)
def visit_individual_node(
self, node: IndividualNode[Entity]) -> JoinDataSourcePlan:
assert isinstance(
node.data_source, ProcessableQuery
), "Invalid composite query. All nodes must be subqueries."
sub_query_plan = self.__plans[node.alias]
return JoinDataSourcePlan(
translated_source=IndividualNode(alias=node.alias,
data_source=sub_query_plan.query),
processors={
node.alias:
SubqueryProcessors(
plan_processors=sub_query_plan.plan_query_processors,
db_processors=sub_query_plan.db_query_processors,
)
},
storage_set_key=sub_query_plan.storage_set_key,
)
selected_columns=[
SelectedExpression(
"average",
FunctionCall(
"average", "avg",
(Column(None, None, "count_environment"), )),
),
],
),
CompositeExecutionStrategy(get_cluster(StorageSetKey.EVENTS), [],
{}, []),
StorageSetKey.EVENTS,
SubqueryProcessors(
[],
[
*events_storage.get_query_processors(),
MandatoryConditionApplier(),
MandatoryConditionEnforcer([]),
],
),
None,
),
CompositeQuery(
from_clause=ClickhouseQuery(
from_clause=events_table,
selected_columns=[
SelectedExpression("project_id",
Column(None, None, "project_id")),
SelectedExpression(
"count_environment",
FunctionCall(
"count_environment",
),
selected_columns=[
SelectedExpression(
"average",
FunctionCall("average", "avg",
(Column(None, None, "count_release"), )),
),
],
),
CompositeExecutionStrategy(get_cluster(StorageSetKey.EVENTS), [],
{}, []),
StorageSetKey.EVENTS,
SubqueryProcessors(
[],
[
*get_storage(StorageKey.EVENTS).get_query_processors(),
MandatoryConditionApplier(),
],
),
None,
),
CompositeQuery(
from_clause=ClickhouseQuery(
from_clause=events_table,
selected_columns=[
SelectedExpression("project_id",
Column(None, None, "project_id")),
SelectedExpression(
"count_release",
FunctionCall(
"count_release",