-
Notifications
You must be signed in to change notification settings - Fork 133
/
Copy pathupload_artifacts.sql
48 lines (35 loc) · 1.37 KB
/
upload_artifacts.sql
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
{% macro upload_dbt_artifacts(filenames) %}
{% set src_dbt_artifacts = source('dbt_artifacts', 'artifacts') %}
{% set remove_query %}
remove @{{ src_dbt_artifacts }} pattern='.*.json.gz';
{% endset %}
{% do log("Clearing existing files from Stage: " ~ remove_query, info=True) %}
{% do run_query(remove_query) %}
{% for filename in filenames %}
{% set file = filename ~ '.json' %}
{% set put_query %}
put file://target/{{ file }} @{{ src_dbt_artifacts }} auto_compress=true;
{% endset %}
{% do log("Uploading " ~ file ~ " to Stage: " ~ put_query, info=True) %}
{% do run_query(put_query) %}
{% set copy_query %}
begin;
copy into {{ src_dbt_artifacts }} from
(
select
$1 as data,
$1:metadata:generated_at::timestamp_ntz as generated_at,
metadata$filename as path,
regexp_substr(metadata$filename, '([a-z_]+.json)') as artifact_type
from @{{ src_dbt_artifacts }}
)
file_format=(type='JSON')
on_error='skip_file';
commit;
{% endset %}
{% do log("Copying " ~ file ~ " from Stage: " ~ copy_query, info=True) %}
{% do run_query(copy_query) %}
{% endfor %}
{% do log("Clearing new files from Stage: " ~ remove_query, info=True) %}
{% do run_query(remove_query) %}
{% endmacro %}