feat: implement dateadd macro

UiPath · Nov 14, 2023 · de7beb2 · de7beb2
1 parent 8a89554
commit de7beb2
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/macros/multiple_databases/dateadd.sql b/macros/multiple_databases/dateadd.sql
@@ -0,0 +1,20 @@
+{%- macro dateadd(datepart, number, date_field) -%}
+
+{%- if target.type == 'databricks' -%}
+    {%- if datepart == 'millisecond' -%}
+        timestamp_millis(unix_millis({{ date_field }}) + {{ number }})
+    {%- elif datepart == 'second' -%}
+        timestamp_seconds(unix_seconds({{ date_field }}) + {{ number }})
+    {%- elif datepart == 'minute' -%}
+        timestamp_seconds(unix_seconds({{ date_field }}) + {{ number }}*60)
+    {%- elif datepart == 'hour' -%}
+        timestamp_seconds(unix_seconds({{ date_field }}) + {{ number }}*3600)
+    {%- elif datepart == 'day' -%}
+        date_add({{ datepart }}, {{ number }})
+    {%- elif datepart == 'week' -%}
+        date_add({{ datepart }}, {{ number }}*7)
+    {%- endif -%}
+{%- else -%}
+    dateadd({{ datepart }}, {{ number }}, {{ date_field }})
+{%- endif -%}
+{%- endmacro -%}