def apdex(column: Column, satisfied: Literal) -> Expression:
tolerated = multiply(satisfied, Literal(None, 4))
return div(
plus(
countIf(
binary_condition(None, ConditionFunctions.LTE, column,
satisfied)),
div(
countIf(
binary_condition(
None,
BooleanFunctions.AND,
binary_condition(
None,
ConditionFunctions.GT,
column,
satisfied,
),
binary_condition(
None,
ConditionFunctions.LTE,
column,
tolerated,
),
), ),
Literal(None, 2),
),
),
count(),
)
def process_functions(exp: Expression) -> Expression:
if isinstance(exp, FunctionCall) and exp.function_name == "impact":
assert len(exp.parameters) == 3
column = exp.parameters[0]
satisfied = exp.parameters[1]
user_column = exp.parameters[2]
return plus(
minus(Literal(None, 1), apdex(column, satisfied)),
multiply(
minus(
Literal(None, 1),
div(
Literal(None, 1),
FunctionCall(
None,
"sqrt",
(FunctionCall(None, "uniq", user_column)),
),
),
),
Literal(None, 3),
),
)
return exp
def test_impact_format_expressions() -> None:
unprocessed = Query(
{},
TableSource("events", ColumnSet([])),
selected_columns=[
Column(None, "column2", None),
FunctionCall(
"perf",
"impact",
(
Column(None, "column1", None),
Literal(None, 300),
Column(None, "user", None),
),
),
],
)
expected = Query(
{},
TableSource("events", ColumnSet([])),
selected_columns=[
Column(None, "column2", None),
plus(
minus(
Literal(None, 1),
div(
plus(
countIf(
binary_condition(
None,
ConditionFunctions.LTE,
Column(None, "column1", None),
Literal(None, 300),
),
),
div(
countIf(
binary_condition(
None,
BooleanFunctions.AND,
binary_condition(
None,
ConditionFunctions.GT,
Column(None, "column1", None),
Literal(None, 300),
),
binary_condition(
None,
ConditionFunctions.LTE,
Column(None, "column1", None),
multiply(
Literal(None, 300), Literal(None, 4)
),
),
),
),
Literal(None, 2),
),
),
count(),
),
),
multiply(
minus(
Literal(None, 1),
div(
Literal(None, 1),
FunctionCall(
None,
"sqrt",
(
FunctionCall(
None,
"uniq",
Column(
alias=None,
column_name="user",
table_name=None,
),
)
),
),
),
),
Literal(None, 3),
),
),
],
)
ImpactProcessor().process_query(unprocessed, HTTPRequestSettings())
assert (
expected.get_selected_columns_from_ast()
== unprocessed.get_selected_columns_from_ast()
)
ret = unprocessed.get_selected_columns_from_ast()[1].accept(
ClickhouseExpressionFormatter()
)
assert ret == (
"plus(minus(1, div(plus(countIf(lessOrEquals(column1, 300)), "
"div(countIf(and(greater(column1, 300), lessOrEquals(column1, "
"multiply(300, 4)))), 2)), count())), "
"multiply(minus(1, div(1, sqrt(user, uniq(user)))), 3))"
)
def test_apdex_format_expressions() -> None:
unprocessed = Query(
{},
TableSource("events", ColumnSet([])),
selected_columns=[
Column(None, "column2", None),
FunctionCall(
"perf", "apdex", (Column(None, "column1", None), Literal(None, 300))
),
],
)
expected = Query(
{},
TableSource("events", ColumnSet([])),
selected_columns=[
Column(None, "column2", None),
div(
plus(
FunctionCall(
None,
"countIf",
(
binary_condition(
None,
ConditionFunctions.LTE,
Column(None, "column1", None),
Literal(None, 300),
),
),
),
div(
FunctionCall(
None,
"countIf",
(
binary_condition(
None,
BooleanFunctions.AND,
binary_condition(
None,
ConditionFunctions.GT,
Column(None, "column1", None),
Literal(None, 300),
),
binary_condition(
None,
ConditionFunctions.LTE,
Column(None, "column1", None),
multiply(Literal(None, 300), Literal(None, 4)),
),
),
),
),
Literal(None, 2),
),
),
FunctionCall(None, "count", (),),
),
],
)
ApdexProcessor().process_query(unprocessed, HTTPRequestSettings())
assert (
expected.get_selected_columns_from_ast()
== unprocessed.get_selected_columns_from_ast()
)
ret = unprocessed.get_selected_columns_from_ast()[1].accept(
ClickhouseExpressionFormatter()
)
assert ret == (
"div(plus(countIf(lessOrEquals(column1, 300)), "
"div(countIf(and(greater(column1, 300), "
"lessOrEquals(column1, multiply(300, 4)))), 2)), count())"
)