[fix][runtime] Fix datetime extraction null failure

Author: wt0530

runtime/src/main/java/io/dingodb/expr/runtime/op/date/DayFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Timestamp;
@@ -49,6 +50,10 @@ static int extractDay(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractDay(date);
     }
 
+    static @Nullable Object extractDay(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/HourFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Time;
@@ -55,6 +56,10 @@ static int extractHour(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractHour(time);
     }
 
+    static @Nullable Object extractHour(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/MillisecondFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Timestamp;
@@ -39,6 +40,10 @@ static int extractMillisecond(@NonNull Timestamp value, ExprConfig config) {
         return TimestampUtils.extractMinuteSecond(value);
     }
 
+    static @Nullable Object extractMillisecond(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/MinuteFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Time;
@@ -52,6 +53,10 @@ static int extractMinute(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractMinute(time);
     }
 
+    static @Nullable Object extractMinute(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/MonthFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Timestamp;
@@ -48,6 +49,10 @@ static int extractMonth(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractMonth(date);
     }
 
+    static @Nullable Object extractMonth(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/QuarterFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.io.Serial;
 import java.sql.Date;
@@ -41,6 +42,18 @@ static int extractQuarter(@NonNull Timestamp value, ExprConfig config) {
         return TimestampUtils.extractQuarter(value);
     }
 
+    static int extractQuarter(String value, @NonNull ExprConfig config) {
+        Date date = DateTimeUtils.parseDate(value, config.getParseDateAndTimestampFormatters());
+        if (date == null) {
+            return 0;
+        }
+        return DateTimeUtils.extractQuarter(date);
+    }
+
+    static @Nullable Object extractQuarter(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/SecondFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Time;
@@ -52,6 +53,10 @@ static int extractSecond(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractSecond(time);
     }
 
+    static @Nullable Object extractSecond(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/WeekFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Timestamp;
@@ -48,6 +49,10 @@ static int extractWeek(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractWeek(date);
     }
 
+    static @Nullable Object extractWeek(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/date/YearFun.java

@@ -22,6 +22,7 @@
 import io.dingodb.expr.runtime.utils.DateTimeUtils;
 import io.dingodb.expr.runtime.utils.TimestampUtils;
 import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.sql.Date;
 import java.sql.Timestamp;
@@ -47,6 +48,10 @@ static int extractYear(String value, @NonNull ExprConfig config) {
         return DateTimeUtils.extractYear(date);
     }
 
+    static @Nullable Object extractYear(Void value, @NonNull ExprConfig config) {
+        return null;
+    }
+
     @Override
     public @NonNull String getName() {
         return NAME;

runtime/src/main/java/io/dingodb/expr/runtime/op/special/IfNullFun.java

@@ -76,6 +76,9 @@ static Timestamp ifNull(Timestamp value0, Timestamp value1) {
     }
 
     static Object ifNull(Object value0, Object value1) {
+        if (value0 instanceof Float && (value1 instanceof Double || value1 instanceof BigDecimal)) {
+            return new BigDecimal(value0.toString()).doubleValue();
+        }
         return value0 == null ? value1 : value0;
     }