fix: cast number param for dateadd to int for databricks

UiPath · Nov 17, 2023 · fec32aa · fec32aa
1 parent 3221bc2
commit fec32aa
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/macros/multiple_databases/dateadd.sql b/macros/multiple_databases/dateadd.sql
@@ -4,30 +4,33 @@
 {%- elif target.type == 'sqlserver' -%}
     dateadd({{ datepart }}, {{ number }}, try_convert(datetime2, {{ date_or_datetime_field }}))
 {%- elif target.type == 'databricks' -%}
+    {%- set number_int -%}
+        try_cast({{ number }} as INT)
+    {%- endset -%}
     {%- set datetime_field -%}
         try_to_timestamp({{ date_or_datetime_field }})
     {%- endset -%}
     {%- set time_field -%}
         unix_millis({{ datetime_field }}) - unix_millis(date_trunc('DD', {{ datetime_field }}))
     {%- endset -%}
     {%- if datepart == 'millisecond' -%}
-        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number }})
+        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number_int }})
     {%- elif datepart == 'second' -%}
-        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number }}*1000)
+        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number_int }}*1000)
     {%- elif datepart == 'minute' -%}
-        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number }}*60000)
+        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number_int }}*60000)
     {%- elif datepart == 'hour' -%}
-        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number }}*3600000)
+        timestamp_millis(unix_millis({{ datetime_field }}) + {{ number_int }}*3600000)
     {%- elif datepart == 'day' -%}
-        timestamp_millis(unix_millis(try_to_timestamp(date_add(to_date({{ datetime_field }}), {{ number }}))) + {{ time_field }})
+        timestamp_millis(unix_millis(try_to_timestamp(date_add(to_date({{ datetime_field }}), {{ number_int }}))) + {{ time_field }})
     {%- elif datepart == 'week' -%}
-        timestamp_millis(unix_millis(try_to_timestamp(date_add(to_date({{ datetime_field }}), {{ number }}*7))) + {{ time_field }})
+        timestamp_millis(unix_millis(try_to_timestamp(date_add(to_date({{ datetime_field }}), {{ number_int }}*7))) + {{ time_field }})
     {%- elif datepart == 'month' -%}
-        timestamp_millis(unix_millis(try_to_timestamp(add_months(to_date({{ datetime_field }}), {{ number }}))) + {{ time_field }})
+        timestamp_millis(unix_millis(try_to_timestamp(add_months(to_date({{ datetime_field }}), {{ number_int }}))) + {{ time_field }})
     {%- elif datepart == 'quarter' -%}
-        timestamp_millis(unix_millis(try_to_timestamp(add_months(to_date({{ datetime_field }}), {{ number }}*3))) + {{ time_field }})
+        timestamp_millis(unix_millis(try_to_timestamp(add_months(to_date({{ datetime_field }}), {{ number_int }}*3))) + {{ time_field }})
     {%- elif datepart == 'year' -%}
-        timestamp_millis(unix_millis(try_to_timestamp(add_months(to_date({{ datetime_field }}), {{ number }}*12))) + {{ time_field }})
+        timestamp_millis(unix_millis(try_to_timestamp(add_months(to_date({{ datetime_field }}), {{ number_int }}*12))) + {{ time_field }})
     {%- endif -%}
 {%- endif -%}
 {%- endmacro -%}