1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
---
source: prql-compiler/src/parser.rs
expression: "ast_of_string(r#\"\nfrom employees\nfilter country == \"USA\" # Each line transforms the previous result.\nderive [ # This adds columns / variables.\n gross_salary = salary + payroll_tax,\n gross_cost = gross_salary + benefits_cost # Variables can use other variables.\n]\nfilter gross_cost > 0\ngroup [title, country] (\naggregate [ # `by` are the columns to group by.\n average salary, # These are aggregation calcs run on each group.\n sum salary,\n average gross_salary,\n sum gross_salary,\n average gross_cost,\n sum_gross_cost = sum gross_cost,\n ct = count,\n] )\nsort sum_gross_cost\nfilter ct > 200\ntake 20\n \"#.trim(),\n Rule::query)?"
---
Query:
version: ~
dialect: Generic
nodes:
- Pipeline:
nodes:
- FuncCall:
name: from
args:
- Ident: employees
named_args: {}
- FuncCall:
name: filter
args:
- Binary:
left:
Ident: country
op: Eq
right:
Literal:
String: USA
named_args: {}
- FuncCall:
name: derive
args:
- List:
- Assign:
name: gross_salary
expr:
Binary:
left:
Ident: salary
op: Add
right:
Ident: payroll_tax
- Assign:
name: gross_cost
expr:
Binary:
left:
Ident: gross_salary
op: Add
right:
Ident: benefits_cost
named_args: {}
- FuncCall:
name: filter
args:
- Binary:
left:
Ident: gross_cost
op: Gt
right:
Literal:
Integer: 0
named_args: {}
- FuncCall:
name: group
args:
- List:
- Ident: title
- Ident: country
- FuncCall:
name: aggregate
args:
- List:
- FuncCall:
name: average
args:
- Ident: salary
named_args: {}
- FuncCall:
name: sum
args:
- Ident: salary
named_args: {}
- FuncCall:
name: average
args:
- Ident: gross_salary
named_args: {}
- FuncCall:
name: sum
args:
- Ident: gross_salary
named_args: {}
- FuncCall:
name: average
args:
- Ident: gross_cost
named_args: {}
- Assign:
name: sum_gross_cost
expr:
FuncCall:
name: sum
args:
- Ident: gross_cost
named_args: {}
- Assign:
name: ct
expr:
Ident: count
named_args: {}
named_args: {}
- FuncCall:
name: sort
args:
- Ident: sum_gross_cost
named_args: {}
- FuncCall:
name: filter
args:
- Binary:
left:
Ident: ct
op: Gt
right:
Literal:
Integer: 200
named_args: {}
- FuncCall:
name: take
args:
- Literal:
Integer: 20
named_args: {}