-
Notifications
You must be signed in to change notification settings - Fork 251
/
0019.yml
231 lines (230 loc) · 9.34 KB
/
0019.yml
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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
version: 19
migrationScript: 0019-migration.sql
downgradeScript: 0019-downgrade.sql
methods:
queue_artifacts_entities_load:
description: See taskcluster-lib-entities
mode: read
serviceName: queue
args: partition_key text, row_key text
returns: table (partition_key_out text, row_key_out text, value jsonb, version integer, etag uuid)
body: |-
declare
decoded_composite_key text[];
begin
decoded_composite_key := decode_composite_key(queue_artifacts_entities_load.partition_key);
return query
select
queue_artifacts_entities_load.partition_key,
queue_artifacts_entities_load.row_key,
entity_buf_encode(
jsonb_build_object(
'PartitionKey', queue_artifacts_entities_load.partition_key,
'RowKey', queue_artifacts_entities_load.row_key,
'taskId', slugid_to_uuid(task_id),
'runId', run_id,
'name', name,
'storageType', storage_type,
'contentType', content_type,
'present', present,
'expires', expires),
'details', details::text) as value,
1 as version,
queue_artifacts.etag as etag
from queue_artifacts
where
queue_artifacts.task_id = decoded_composite_key[1] and queue_artifacts.run_id = decoded_composite_key[2]::int and queue_artifacts.name = decode_string_key(queue_artifacts_entities_load.row_key);
end
queue_artifacts_entities_create:
serviceName: queue
description: See taskcluster-lib-entities
mode: write
args: pk text, rk text, properties jsonb, overwrite boolean, version integer
returns: uuid
body: |-
declare
new_row queue_artifacts%ROWTYPE;
begin
select
uuid_to_slugid(properties ->> 'taskId'),
(properties ->> 'runId')::integer,
(properties ->> 'name')::text,
(properties ->> 'storageType')::text,
(properties ->> 'contentType')::text,
entity_buf_decode(properties, 'details')::jsonb,
(properties ->> 'present')::boolean,
(properties ->> 'expires')::timestamptz,
public.gen_random_uuid()
into new_row;
if overwrite then
raise exception 'overwrite not implemented';
else
execute 'insert into queue_artifacts select $1.*' using new_row;
end if;
return new_row.etag;
end
queue_artifacts_entities_remove:
serviceName: queue
description: See taskcluster-lib-entities
mode: write
args: partition_key text, row_key text
returns: table (etag uuid)
body: |-
declare
decoded_composite_key text[];
begin
decoded_composite_key := decode_composite_key(queue_artifacts_entities_remove.partition_key);
return query delete from queue_artifacts
where
queue_artifacts.task_id = decoded_composite_key[1] and queue_artifacts.run_id = decoded_composite_key[2]::int and queue_artifacts.name = decode_string_key(queue_artifacts_entities_remove.row_key)
returning queue_artifacts.etag;
end
queue_artifacts_entities_modify:
serviceName: queue
description: See taskcluster-lib-entities
mode: write
args: partition_key text, row_key text, properties jsonb, version integer, old_etag uuid
returns: table (etag uuid)
body: |-
declare
new_row queue_artifacts%ROWTYPE;
decoded_composite_key text[];
begin
decoded_composite_key := decode_composite_key(queue_artifacts_entities_modify.partition_key);
select
(properties ->> 'taskId')::text,
(properties ->> 'runId')::integer,
(properties ->> 'name')::text,
(properties ->> 'storageType')::text,
(properties ->> 'contentType')::text,
entity_buf_decode(properties, 'details')::jsonb,
(properties ->> 'present')::boolean,
(properties ->> 'expires')::timestamptz,
public.gen_random_uuid()
into new_row;
update queue_artifacts
set (
storage_type,
content_type,
details,
present,
expires,
etag
) = (
new_row.storage_type,
new_row.content_type,
new_row.details,
new_row.present,
new_row.expires,
new_row.etag
)
where
queue_artifacts.task_id = decoded_composite_key[1] and
queue_artifacts.run_id = decoded_composite_key[2]::int and
queue_artifacts.name = decode_string_key(queue_artifacts_entities_modify.row_key) and
queue_artifacts.etag = queue_artifacts_entities_modify.old_etag;
if found then
return query select new_row.etag;
return;
end if;
perform queue_artifacts.etag from queue_artifacts
where
queue_artifacts.task_id = decoded_composite_key[1] and
queue_artifacts.run_id = decoded_composite_key[2]::int and
queue_artifacts.name = decode_string_key(queue_artifacts_entities_modify.row_key);
if found then
raise exception 'unsuccessful update' using errcode = 'P0004';
else
raise exception 'no such row' using errcode = 'P0002';
end if;
end
queue_artifacts_entities_scan:
description: See taskcluster-lib-entities
mode: read
serviceName: queue
args: pk text, rk text, condition text, size integer, page integer
returns: table (partition_key text, row_key text, value jsonb, version integer, etag uuid)
body: |-
declare
cond text[];
exp_cond_operator text;
exp_cond_operand timestamptz;
partition_key_var text;
row_key_var text;
decoded_composite_key text[];
begin
decoded_composite_key := decode_composite_key(queue_artifacts_entities_scan.pk);
if not condition is null then
cond := regexp_split_to_array(condition, '\s+');
exp_cond_operator := cond[4];
exp_cond_operand := cond[5] :: timestamptz;
return query select
encode_composite_key(queue_artifacts.task_id, queue_artifacts.run_id::text) as partition_key,
encode_string_key(queue_artifacts.name) as row_key,
entity_buf_encode(
jsonb_build_object(
'PartitionKey', encode_composite_key(queue_artifacts.task_id, queue_artifacts.run_id::text),
'RowKey', name,
'taskId', slugid_to_uuid(task_id),
'runId', run_id,
'name', name,
'storageType', storage_type,
'contentType', content_type,
'present', present,
'expires', expires
),
'details', details::text) as value,
1 as version,
queue_artifacts.etag as etag from queue_artifacts
where
(queue_artifacts_entities_scan.pk is null or queue_artifacts_entities_scan.pk = decoded_composite_key[1] || '~' || decoded_composite_key[2]) and
(queue_artifacts_entities_scan.rk is null or queue_artifacts_entities_scan.rk = name) and
case
when exp_cond_operator = '=' then expires = exp_cond_operand
when exp_cond_operator = '<' then expires < exp_cond_operand
when exp_cond_operator = '<=' then expires <= exp_cond_operand
when exp_cond_operator = '>' then expires > exp_cond_operand
when exp_cond_operator = '>=' then expires >= exp_cond_operand
else expires <> exp_cond_operand
end
order by queue_artifacts.task_id, queue_artifacts.run_id, queue_artifacts.name
limit case
when (size is not null and size > 0) then size + 1
else null
end
offset case
when (page is not null and page > 0) then page
else 0
end;
else
return query select
encode_composite_key(queue_artifacts.task_id, queue_artifacts.run_id::text) as partition_key,
encode_string_key(queue_artifacts.name) as row_key,
entity_buf_encode(
jsonb_build_object(
'PartitionKey', encode_composite_key(queue_artifacts.task_id, queue_artifacts.run_id::text),
'RowKey', name,
'taskId', slugid_to_uuid(task_id),
'runId', run_id,
'name', name,
'storageType', storage_type,
'contentType', content_type,
'present', present,
'expires', expires),
'details', details::text) as value,
1 as version,
queue_artifacts.etag as etag from queue_artifacts
where
(queue_artifacts_entities_scan.pk is null or (queue_artifacts.task_id = decoded_composite_key[1] and queue_artifacts.run_id = decoded_composite_key[2]::int)) and
(queue_artifacts_entities_scan.rk is null or queue_artifacts_entities_scan.rk = name)
order by queue_artifacts.task_id, queue_artifacts.run_id, queue_artifacts.name
limit case
when (size is not null and size > 0) then size + 1
else null
end
offset case
when (size is not null and size > 0 and page is not null and page > 0) then page
else 0
end;
end if;
end