Adds int32 and int64 variants of date_add (#1291)

partiql · Dec 8, 2023 · ddd642d · ddd642d · github-actions · Dec 9, 2023
1 parent 1c3c79a
commit ddd642d
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/partiql-planner/src/main/kotlin/org/partiql/planner/PartiQLHeader.kt b/partiql-planner/src/main/kotlin/org/partiql/planner/PartiQLHeader.kt
@@ -590,23 +590,26 @@ public object PartiQLHeader : Header() {
     private fun extract(): List<FunctionSignature.Scalar> = emptyList()
 
     private fun dateAdd(): List<FunctionSignature.Scalar> {
+        val intervals = listOf(INT32, INT64, INT)
         val operators = mutableListOf<FunctionSignature.Scalar>()
         for (field in DatetimeField.values()) {
             for (type in types.datetime) {
                 if (field == DatetimeField.TIMEZONE_HOUR || field == DatetimeField.TIMEZONE_MINUTE) {
                     continue
                 }
-                val signature = FunctionSignature.Scalar(
-                    name = "date_add_${field.name.lowercase()}",
-                    returns = type,
-                    parameters = listOf(
-                        FunctionParameter("interval", INT),
-                        FunctionParameter("datetime", type),
-                    ),
-                    isNullable = false,
-                    isNullCall = true,
-                )
-                operators.add(signature)
+                for (interval in intervals) {
+                    val signature = FunctionSignature.Scalar(
+                        name = "date_add_${field.name.lowercase()}",
+                        returns = type,
+                        parameters = listOf(
+                            FunctionParameter("interval", interval),
+                            FunctionParameter("datetime", type),
+                        ),
+                        isNullable = false,
+                        isNullCall = true,
+                    )
+                    operators.add(signature)
+                }
             }
         }
         return operators