[fix][dingo-exec] Support json_extract function

Author: guojn1

dingo-common/src/main/java/io/dingodb/common/type/converter/DataConverter.java

@@ -16,6 +16,7 @@
 
 package io.dingodb.common.type.converter;
 
+import io.dingodb.common.log.LogUtils;
 import io.dingodb.common.type.DingoType;
 import io.dingodb.expr.common.type.Type;
 import org.checkerframework.checker.nullness.qual.NonNull;
@@ -94,22 +95,59 @@ default Object convert(@NonNull Map<?, ?> value, @NonNull DingoType keyType, @No
     }
 
     default Integer convertIntegerFrom(@NonNull Object value) {
-        return (Integer) value;
+        if (value instanceof Integer) {
+            return (Integer) value;
+        } else {
+            try {
+                return new BigDecimal(value.toString()).intValue();
+            } catch (Exception e) {
+                return 0;
+            }
+        }
     }
 
     default Long convertLongFrom(@NonNull Object value) {
         if (value instanceof Integer) {
             return ((Integer) value).longValue();
+        } else if (value instanceof Long) {
+            return (Long) value;
+        } else {
+            try {
+                return new BigDecimal(value.toString()).longValue();
+            } catch (Exception e) {
+                return 0L;
+            }
         }
-        return (Long) value;
     }
 
     default Float convertFloatFrom(@NonNull Object value) {
-        return (Float) value;
+        if (value instanceof Float) {
+            return (Float) value;
+        } else if (value instanceof Double) {
+            Double valDouble = (Double) value;
+            return valDouble.floatValue();
+        } else {
+            try {
+                return new BigDecimal(value.toString()).floatValue();
+            } catch (Exception e) {
+                return 0F;
+            }
+        }
     }
 
     default Double convertDoubleFrom(@NonNull Object value) {
-        return (Double) value;
+        if (value instanceof Float) {
+            Float valFloat = (Float) value;
+            return valFloat.doubleValue();
+        } else if (value instanceof Double) {
+            return (Double) value;
+        } else {
+            try {
+                return new BigDecimal(value.toString()).doubleValue();
+            } catch (Exception e) {
+                return 0D;
+            }
+        }
     }
 
     default Boolean convertBooleanFrom(@NonNull Object value) {
@@ -128,28 +166,60 @@ default Boolean convertBooleanFrom(@NonNull Object value) {
     }
 
     default String convertStringFrom(@NonNull Object value) {
-        return (String) value;
+        if (value instanceof String) {
+            return (String) value;
+        } else {
+            return value.toString();
+        }
     }
 
     default BigDecimal convertDecimalFrom(@NonNull Object value) {
-        return (BigDecimal) value;
+        if (value instanceof BigDecimal) {
+            return (BigDecimal) value;
+        } else {
+            try {
+                return new BigDecimal(value.toString());
+            } catch (Exception e) {
+                return new BigDecimal(0);
+            }
+        }
     }
 
     default Date convertDateFrom(@NonNull Object value) {
         if (value instanceof Timestamp) {
             Timestamp timestamp = (Timestamp) value;
             LocalDateTime localDateTime = timestamp.toLocalDateTime();
             return new Date(localDateTime.atZone(ZoneId.of("UTC")).toInstant().toEpochMilli());
+        } else if (value instanceof Date) {
+            return (Date) value;
+        } else {
+            return new Date(System.currentTimeMillis());
         }
-        return (Date) value;
     }
 
     default Time convertTimeFrom(@NonNull Object value) {
-        return (Time) value;
+        if (value instanceof Time) {
+            return (Time) value;
+        } else if (value instanceof Date) {
+            Date date = (Date) value;
+            return new Time(date.getTime());
+        } else if (value instanceof Timestamp) {
+            Timestamp timestamp = (Timestamp) value;
+            return new Time(timestamp.getTime());
+        } else {
+            return new Time(System.currentTimeMillis());
+        }
     }
 
     default Timestamp convertTimestampFrom(@NonNull Object value) {
-        return (Timestamp) value;
+        if (value instanceof Timestamp) {
+            return (Timestamp) value;
+        } else if (value instanceof Date) {
+            Date date = (Date) value;
+            return new Timestamp(date.getTime());
+        } else {
+            return new Timestamp(System.currentTimeMillis());
+        }
     }
 
     default byte[] convertBinaryFrom(@NonNull Object value) {

dingo-exec/src/main/java/io/dingodb/exec/fun/mysql/JsonExtractFun.java

@@ -16,12 +16,19 @@
 
 package io.dingodb.exec.fun.mysql;
 
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
 import io.dingodb.expr.common.type.Type;
 import io.dingodb.expr.common.type.Types;
 import io.dingodb.expr.runtime.ExprConfig;
 import io.dingodb.expr.runtime.op.BinaryOp;
 import org.checkerframework.checker.nullness.qual.NonNull;
 
+import java.util.Arrays;
+import java.util.List;
+
 public class JsonExtractFun extends BinaryOp {
     private static final long serialVersionUID = -8343792468386621027L;
 
@@ -41,6 +48,68 @@ public Type getType() {
 
     @Override
     public Object evalValue(Object value0, Object value1, ExprConfig config) {
-        return "";
+        if (value0 == null) {
+            return null;
+        }
+        if (value1 == null) {
+            return value0;
+        }
+        String path = value1.toString();
+        String[] paths = path.split("\\.");
+        if (!paths[0].startsWith("$")) {
+            return null;
+        }
+        if (paths.length == 1 && paths[0].equals("$")) {
+            return value0;
+        }
+        List<String> pathList = Arrays.asList(paths);
+        ObjectMapper mapper = new ObjectMapper();
+        try {
+            JsonNode rootNode = mapper.readTree(value0.toString());
+            JsonNode node = rootNode;
+            for (int i = 0; i < pathList.size(); i ++) {
+                String pathItem = pathList.get(i);
+                if (i == 0) {
+                    if (pathItem.contains("[") && pathItem.contains("]")) {
+                        int start = pathItem.indexOf("[");
+                        int end = pathItem.indexOf("]");
+                        if (end >= start + 1) {
+                            int item = Integer.parseInt(pathItem.substring(start + 1, end));
+                            node = rootNode.get(item);
+                            if (node == null && !(rootNode instanceof ArrayNode) && item == 0) {
+                                node = rootNode;
+                            }
+                        } else {
+                            return null;
+                        }
+                    } else {
+                        node = rootNode;
+                    }
+                } else {
+                    if (pathItem.contains("[") && pathItem.contains("]")) {
+                        int start = pathItem.indexOf("[");
+                        int end = pathItem.indexOf("]");
+                        if (end >= start + 1) {
+                            int item = Integer.parseInt(pathItem.substring(start + 1, end));
+                            pathItem = pathItem.substring(0, start);
+                            node = node.get(pathItem).get(item);
+                        }
+                    } else {
+                        node = node.get(pathItem);
+                    }
+                }
+
+            }
+            if (node != null) {
+                String res = node.toString();
+                if (res.startsWith("\"") && res.endsWith("\"")) {
+                    return res.substring(1, res.length() - 1);
+                }
+                return node.toString();
+            }
+            return null;
+        } catch (JsonProcessingException e) {
+            return null;
+        }
     }
 }