[func](jsonb)support json_depth function

be/src/util/jsonb_document.h

@@ -554,6 +554,9 @@ class JsonbValue {
     //Get the number of jsonbvalue elements
     int length() const;
 
+    //Get the depth of jsonbvalue
+    int depth() const;
+
     //Whether to include the jsonbvalue rhs
     bool contains(JsonbValue* rhs) const;
 
@@ -1269,6 +1272,53 @@ inline int JsonbValue::length() const {
     }
 }
 
+inline int JsonbValue::depth() const {
+    switch (type_) {
+    case JsonbType::T_Int8:
+    case JsonbType::T_Int16:
+    case JsonbType::T_Int32:
+    case JsonbType::T_Int64:
+    case JsonbType::T_Double:
+    case JsonbType::T_Float:
+    case JsonbType::T_Int128:
+    case JsonbType::T_String:
+    case JsonbType::T_Binary:
+    case JsonbType::T_Null:
+    case JsonbType::T_True:
+    case JsonbType::T_False: {
+        return 1;
+    }
+    case JsonbType::T_Object: {
+        int base_depth = 1;
+        int numElem = ((ObjectVal*)this)->numElem();
+        if (numElem == 0) return base_depth;
+
+        int max_depth = base_depth;
+
+        for (int i = 0; i < numElem; ++i) {
+            JsonbKeyValue* key = ((ObjectVal*)this)->getJsonbKeyValue(i);
+            JsonbValue* value = ((ObjectVal*)this)->find(key->getKeyStr(), key->klen());
+            int current_depth = base_depth + value->depth();
+            if (current_depth > max_depth) max_depth = current_depth;
+        }
+        return max_depth;
+    }
+    case JsonbType::T_Array: {
+        int base_depth = 1;
+        int numElem = ((ArrayVal*)this)->numElem();
+        if (numElem == 0) return base_depth;
+        int max_depth = base_depth;
+        for (int i = 0; i < numElem; ++i) {
+            int current_depth = base_depth + ((ArrayVal*)this)->get(i)->depth();
+            if (current_depth > max_depth) max_depth = current_depth;
+        }
+        return max_depth;
+    }
+    default:
+        return 0;
+    }
+}
+
 inline bool JsonbValue::contains(JsonbValue* rhs) const {
     switch (type_) {
     case JsonbType::T_Int8:

be/src/vec/functions/function_jsonb.cpp

@@ -1167,7 +1167,6 @@ class FunctionJsonbContains : public IFunction {
     size_t get_number_of_arguments() const override {
         return get_variadic_argument_types_impl().size();
     }
-
     bool use_default_implementation_for_nulls() const override { return false; }
 
     Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
@@ -1289,6 +1288,53 @@ struct JsonbContainsAndPathImpl {
     }
 };
 
+class FunctionJsonbDepth : public IFunction {
+public:
+    static constexpr auto name = "json_depth";
+    String get_name() const override { return name; }
+    static FunctionPtr create() { return std::make_shared<FunctionJsonbDepth>(); }
+
+    DataTypePtr get_return_type_impl(const DataTypes& arguments) const override {
+        return std::make_shared<DataTypeInt32>();
+    }
+
+    size_t get_number_of_arguments() const override { return 1; }
+
+    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
+                        size_t result, size_t input_rows_count) const override {
+        DCHECK_GE(arguments.size(), 1);
+        auto jsonb_data_column = block.get_by_position(arguments[0]).column;
+
+        auto null_map = ColumnUInt8::create(input_rows_count, 0);
+        auto return_type = block.get_data_type(result);
+        auto col_result = ColumnVector<Int32>::create();
+        col_result->resize(input_rows_count);
+        auto& res_data = col_result->get_data();
+
+        for (size_t i = 0; i < input_rows_count; ++i) {
+            if (jsonb_data_column->is_null_at(i)) {
+                null_map->get_data()[i] = 1;
+                res_data[i] = 0;
+                continue;
+            }
+
+            auto jsonb_value = jsonb_data_column->get_data_at(i);
+            // doc is NOT necessary to be deleted since JsonbDocument will not allocate memory
+            JsonbDocument* doc = JsonbDocument::createDocument(jsonb_value.data, jsonb_value.size);
+            JsonbValue* value = doc->getValue();
+            if (UNLIKELY(jsonb_value.size == 0 || !value)) {
+                null_map->get_data()[i] = 1;
+                res_data[i] = 0;
+                continue;
+            }
+            res_data[i] = value->depth();
+        }
+
+        block.replace_by_position(result, std::move(col_result));
+        return Status::OK();
+    }
+};
+
 void register_function_jsonb(SimpleFunctionFactory& factory) {
     factory.register_function<FunctionJsonbParse>(FunctionJsonbParse::name);
     factory.register_alias(FunctionJsonbParse::name, FunctionJsonbParse::alias);
@@ -1354,6 +1400,7 @@ void register_function_jsonb(SimpleFunctionFactory& factory) {
     factory.register_function<FunctionJsonbLength<JsonbLengthAndPathImpl>>();
     factory.register_function<FunctionJsonbContains<JsonbContainsImpl>>();
     factory.register_function<FunctionJsonbContains<JsonbContainsAndPathImpl>>();
+    factory.register_function<FunctionJsonbDepth>();
 }
 
 } // namespace doris::vectorized

fe/fe-core/src/main/java/org/apache/doris/catalog/BuiltinScalarFunctions.java

@@ -218,6 +218,7 @@
 import org.apache.doris.nereids.trees.expressions.functions.scalar.IsIpv6String;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonArray;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonContains;
+import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonDepth;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonExtract;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonInsert;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonLength;
@@ -656,6 +657,7 @@ public class BuiltinScalarFunctions implements FunctionHelper {
             scalar(ToIpv6OrDefault.class, "to_ipv6_or_default"),
             scalar(ToIpv6OrNull.class, "to_ipv6_or_null"),
             scalar(JsonArray.class, "json_array"),
+            scalar(JsonDepth.class, "json_depth"),
             scalar(JsonObject.class, "json_object"),
             scalar(JsonQuote.class, "json_quote"),
             scalar(JsonUnQuote.class, "json_unquote"),

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/scalar/JsonDepth.java

@@ -0,0 +1,69 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package org.apache.doris.nereids.trees.expressions.functions.scalar;
+
+import org.apache.doris.catalog.FunctionSignature;
+import org.apache.doris.nereids.trees.expressions.Expression;
+import org.apache.doris.nereids.trees.expressions.functions.ExplicitlyCastableSignature;
+import org.apache.doris.nereids.trees.expressions.functions.PropagateNullable;
+import org.apache.doris.nereids.trees.expressions.shape.BinaryExpression;
+import org.apache.doris.nereids.trees.expressions.visitor.ExpressionVisitor;
+import org.apache.doris.nereids.types.IntegerType;
+import org.apache.doris.nereids.types.JsonType;
+
+import com.google.common.base.Preconditions;
+import com.google.common.collect.ImmutableList;
+
+import java.util.List;
+
+/**
+ * ScalarFunction 'json_depth'. This class is generated by GenerateFunction.
+ */
+public class JsonDepth extends ScalarFunction
+        implements BinaryExpression, ExplicitlyCastableSignature, PropagateNullable {
+
+    public static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
+            FunctionSignature.ret(IntegerType.INSTANCE).args(JsonType.INSTANCE)
+    );
+
+    /**
+     * constructor with 1 arguments.
+     */
+    public JsonDepth(Expression arg0) {
+        super("json_depth", arg0);
+    }
+
+    /**
+     * withChildren.
+     */
+    @Override
+    public JsonDepth withChildren(List<Expression> children) {
+        Preconditions.checkArgument(children.size() == 1);
+        return new JsonDepth(children.get(0));
+    }
+
+    @Override
+    public List<FunctionSignature> getSignatures() {
+        return SIGNATURES;
+    }
+
+    @Override
+    public <R, C> R accept(ExpressionVisitor<R, C> visitor, C context) {
+        return visitor.visitJsonDepth(this, context);
+    }
+}

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/visitor/ScalarFunctionVisitor.java

@@ -221,6 +221,7 @@
 import org.apache.doris.nereids.trees.expressions.functions.scalar.IsIpv6String;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonArray;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonContains;
+import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonDepth;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonExtract;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonInsert;
 import org.apache.doris.nereids.trees.expressions.functions.scalar.JsonLength;
@@ -1415,6 +1416,10 @@ default R visitJsonLength(JsonLength jsonLength, C context) {
         return visitScalarFunction(jsonLength, context);
     }
 
+    default R visitJsonDepth(JsonDepth jsonDepth, C context) {
+        return visitScalarFunction(jsonDepth, context);
+    }
+
     default R visitJsonContains(JsonContains jsonContains, C context) {
         return visitScalarFunction(jsonContains, context);
     }

gensrc/script/doris_builtins_functions.py

@@ -1787,6 +1787,7 @@
         [['json_contains'], 'BOOLEAN', ['JSONB', 'JSONB'], 'ALWAYS_NULLABLE'],
         [['json_contains'], 'BOOLEAN', ['JSONB', 'JSONB', 'VARCHAR'], 'ALWAYS_NULLABLE'],
         [['json_contains'], 'BOOLEAN', ['JSONB', 'JSONB', 'STRING'], 'ALWAYS_NULLABLE'],
+        [['json_depth'], 'INT', ['JSONB'], 'DEPEND_ON_ARGUMENT'],
 
         # Json functions
         [['get_json_int'], 'INT', ['VARCHAR', 'VARCHAR'], 'ALWAYS_NULLABLE'],

regression-test/data/jsonb_p0/test_jsonb_load_and_function.out

@@ -7726,3 +7726,65 @@ true
 30	-9223372036854775808	false
 31	18446744073709551615	false
 
+-- !sql_json_depth --
+\N
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+2
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+2
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+1	\N	\N
+2	null	1
+3	true	1
+4	false	1
+5	100	1
+6	10000	1
+7	1000000000	1
+8	1152921504606846976	1
+9	6.18	1
+10	"abcd"	1
+11	{}	1
+12	{"k1":"v31","k2":300}	2
+13	[]	1
+14	[123,456]	2
+15	["abc","def"]	2
+16	[null,true,false,100,6.18,"abc"]	2
+17	[{"k1":"v41","k2":400},1,"a",3.14]	3
+18	{"k1":"v31","k2":300,"a1":[{"k1":"v41","k2":400},1,"a",3.14]}	4
+26	\N	\N
+27	{"k1":"v1","k2":200}	2
+28	{"a.b.c":{"k1.a1":"v31","k2":300},"a":"niu"}	3
+29	12524337771678448270	1
+30	-9223372036854775808	1
+31	18446744073709551615	1
+

regression-test/data/jsonb_p0/test_jsonb_load_unique_key_and_function.out

@@ -6105,3 +6105,62 @@ true
 27	{"k1":"v1","k2":200}	false
 28	{"a.b.c":{"k1.a1":"v31","k2":300},"a":"niu"}	false
 
+-- !sql_json_depth --
+\N
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+1
+
+-- !sql_json_depth --
+2
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+2
+
+-- !sql_json_depth --
+3
+
+-- !sql_json_depth --
+1	\N	\N
+2	null	1
+3	true	1
+4	false	1
+5	100	1
+6	10000	1
+7	1000000000	1
+8	1152921504606846976	1
+9	6.18	1
+10	"abcd"	1
+11	{}	1
+12	{"k1":"v31","k2":300}	2
+13	[]	1
+14	[123,456]	2
+15	["abc","def"]	2
+16	[null,true,false,100,6.18,"abc"]	2
+17	[{"k1":"v41","k2":400},1,"a",3.14]	3
+18	{"k1":"v31","k2":300,"a1":[{"k1":"v41","k2":400},1,"a",3.14]}	4
+26	\N	\N
+27	{"k1":"v1","k2":200}	2
+28	{"a.b.c":{"k1.a1":"v31","k2":300},"a":"niu"}	3
+

regression-test/suites/jsonb_p0/test_jsonb_load_and_function.groovy

@@ -572,4 +572,18 @@ suite("test_jsonb_load_and_function", "p0") {
     qt_select_json_contains """SELECT id, j, json_contains(j, cast('{"k2":300}' as json)) FROM ${testTable} ORDER BY id"""
     qt_select_json_contains """SELECT id, j, json_contains(j, cast('{"k1":"v41","k2":400}' as json), '\$.a1') FROM ${testTable} ORDER BY id"""
     qt_select_json_contains """SELECT id, j, json_contains(j, cast('[123,456]' as json)) FROM ${testTable} ORDER BY id"""
+
+    qt_sql_json_depth """SELECT json_depth(NULL)"""
+    qt_sql_json_depth """SELECT json_depth('true')"""
+    qt_sql_json_depth """SELECT json_depth('null')"""
+    qt_sql_json_depth """SELECT json_depth('"abc"')"""
+    qt_sql_json_depth """SELECT json_depth('[]')"""
+    qt_sql_json_depth """SELECT json_depth('{}')"""
+    qt_sql_json_depth """SELECT json_depth('[1, 2]')"""
+    qt_sql_json_depth """SELECT json_depth('[1, {"x": 2}]')"""
+    qt_sql_json_depth """SELECT json_depth('{"x": 1, "y": [1, 2]}')"""
+    qt_sql_json_depth """SELECT json_depth('[1, [2, 3]]')"""
+    qt_sql_json_depth """SELECT json_depth('{"x": 1}')"""
+    qt_sql_json_depth """SELECT json_depth('{"x": {"y": 1}}')"""
+    qt_sql_json_depth """SELECT id, j, json_depth(j) FROM ${testTable} ORDER BY id"""
 }