[feat][coding] Timestamp pushdown.

Author: nokiaMS

coding/src/main/java/io/dingodb/expr/coding/ExprCoder.java

@@ -182,7 +182,8 @@ public CodingFlag visitUnaryOpExpr(@NonNull UnaryOpExpr expr, OutputStream obj)
                     Byte dstType = TypeCoder.INSTANCE.visit(expr.getType());
                     Byte srcType = TypeCoder.INSTANCE.visit((Type) expr.getOp().getKey());
                     if (dstType != null && srcType != null
-                        && dstType != TypeCoder.TYPE_DATE && srcType != TypeCoder.TYPE_DATE) {
+                        && dstType != TypeCoder.TYPE_DATE && srcType != TypeCoder.TYPE_DATE
+                        && dstType != TypeCoder.TYPE_TIMESTAMP && srcType != TypeCoder.TYPE_TIMESTAMP) {
                         obj.write(expr.getOp().doRangeChecking() ? CAST_C : CAST);
                         obj.write(dstType << 4 | srcType);
                         success = true;

coding/src/main/java/io/dingodb/expr/coding/TypeCoder.java

@@ -24,6 +24,7 @@
 import io.dingodb.expr.common.type.IntType;
 import io.dingodb.expr.common.type.LongType;
 import io.dingodb.expr.common.type.StringType;
+import io.dingodb.expr.common.type.TimestampType;
 import io.dingodb.expr.common.type.TypeVisitorBase;
 import lombok.AccessLevel;
 import lombok.RequiredArgsConstructor;
@@ -39,6 +40,7 @@ class TypeCoder extends TypeVisitorBase<Byte, Void> {
     public static final byte TYPE_DECIMAL = (byte) 0x06;
     public static final byte TYPE_STRING = (byte) 0x07;
     public static final byte TYPE_DATE = (byte) 0x08;
+    public static final byte TYPE_TIMESTAMP = (byte) 0x09;
     static final TypeCoder INSTANCE = new TypeCoder();
 
     @Override
@@ -81,4 +83,9 @@ public Byte visitStringType(@NonNull StringType type, Void obj) {
     public Byte visitDateType(@NonNull DateType type, Void obj) {
         return TYPE_DATE;
     }
+
+    @Override
+    public Byte visitTimestampType(@NonNull TimestampType type, Void obj) {
+        return TYPE_TIMESTAMP;
+    }
 }

coding/src/main/java/io/dingodb/expr/coding/ValCoder.java

@@ -24,6 +24,7 @@
 import io.dingodb.expr.common.type.IntType;
 import io.dingodb.expr.common.type.LongType;
 import io.dingodb.expr.common.type.StringType;
+import io.dingodb.expr.common.type.TimestampType;
 import io.dingodb.expr.common.type.TypeVisitorBase;
 import io.dingodb.expr.runtime.expr.Val;
 import io.dingodb.expr.runtime.utils.CodecUtils;
@@ -35,6 +36,7 @@
 import java.io.OutputStream;
 import java.nio.charset.StandardCharsets;
 import java.sql.Date;
+import java.sql.Timestamp;
 
 @RequiredArgsConstructor(access = AccessLevel.PACKAGE)
 class ValCoder extends TypeVisitorBase<CodingFlag, OutputStream> {
@@ -94,6 +96,20 @@ public CodingFlag visitDateType(@NonNull DateType type, OutputStream obj) {
         return CodingFlag.OK;
     }
 
+    @SneakyThrows
+    @Override
+    public CodingFlag visitTimestampType(@NonNull TimestampType type, OutputStream obj) {
+        Timestamp value = (Timestamp) val.getValue();
+        if (value != null) {
+            long milliseconds = value.getTime();
+            obj.write(CONST | TypeCoder.TYPE_TIMESTAMP);
+            CodecUtils.encodeVarInt(obj, milliseconds);
+        } else {
+            obj.write(NULL | TypeCoder.TYPE_TIMESTAMP);
+        }
+        return CodingFlag.OK;
+    }
+
     @SneakyThrows
     @Override
     public CodingFlag visitFloatType(@NonNull FloatType type, OutputStream obj) {