cast untyped start and step when inferring type for

`generate_series`
risingwavelabs · Oct 2, 2024 · 50f30f5 · 50f30f5
1 parent ef8c50f
commit 50f30f5
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/frontend/src/expr/type_inference/func.rs b/src/frontend/src/expr/type_inference/func.rs
@@ -704,10 +704,18 @@ fn infer_type_for_special_table_function(
 ) -> Result<Option<DataType>> {
     match func_type {
         PbTableFuncType::GenerateSeries => {
-            if inputs.len() < 3 {
+            if inputs.len() < 3 || !inputs[1].is_now() {
                 // let signature map handle this
                 return Ok(None);
             }
+            // Now we are inferring type for `generate_series(start, now(), step)`, which will
+            // be further handled by `GenerateSeriesWithNowRule`.
+            if inputs[0].is_untyped() {
+                inputs[0].cast_implicit_mut(DataType::Timestamptz)?;
+            }
+            if inputs[2].is_untyped() {
+                inputs[2].cast_implicit_mut(DataType::Interval)?;
+            }
             match (
                 inputs[0].return_type(),
                 inputs[1].return_type(),