MCOL-785 Implement DISTRIBUTED JSON functions

The following functions are created: Create function JSON_VALID and test cases Create function JSON_DEPTH and test cases Create function JSON_LENGTH and test cases Create function JSON_EQUALS and test cases Create function JSON_NORMALIZE and test cases Create function JSON_TYPE and test cases Create function JSON_OBJECT and test cases Create function JSON_ARRAY and test cases Create function JSON_KEYS and test cases Create function JSON_EXISTS and test cases Create function JSON_QUOTE/JSON_UNQUOTE and test cases Create function JSON_COMPACT/DETAILED/LOOSE and test cases Create function JSON_MERGE and test cases Create function JSON_MERGE_PATCH and test cases Create function JSON_VALUE and test cases Create function JSON_QUERY and test cases Create function JSON_CONTAINS and test cases Create function JSON_ARRAY_APPEND and test cases Create function JSON_ARRAY_INSERT and test cases Create function JSON_INSERT/REPLACE/SET and test cases Create function JSON_REMOVE and test cases Create function JSON_CONTAINS_PATH and test cases Create function JSON_OVERLAPS and test cases Create function JSON_EXTRACT and test cases Create function JSON_SEARCH and test cases Note: Some functions output differs from MDB because session variables that affects functions output,e.g JSON_QUOTE/JSON_UNQUOTE This depends on MCOL-5212
2025-07-29 08:21:15 +03:00 · 2022-06-23 10:49:29 +08:00
parent b5d8e0324b
commit cdd41f05f3
82 changed files with 8645 additions and 2 deletions
--- a/utils/funcexp/func_json_contains_path.cpp
+++ b/utils/funcexp/func_json_contains_path.cpp
@ -0,0 +1,141 @@
+#include <string_view>
+#include <algorithm>
+using namespace std;
+
+#include "functor_json.h"
+#include "functioncolumn.h"
+#include "constantcolumn.h"
+#include "rowgroup.h"
+using namespace execplan;
+using namespace rowgroup;
+
+#include "dataconvert.h"
+
+#include "jsonhelpers.h"
+using namespace funcexp::helpers;
+
+namespace funcexp
+{
+CalpontSystemCatalog::ColType Func_json_contains_path::operationType(
+    FunctionParm& fp, CalpontSystemCatalog::ColType& resultType)
+{
+  return fp[0]->data()->resultType();
+}
+
+/**
+ * getBoolVal API definition
+ */
+bool Func_json_contains_path::getBoolVal(Row& row, FunctionParm& fp, bool& isNull,
+                                         CalpontSystemCatalog::ColType& type)
+{
+  const string_view js = fp[0]->data()->getStrVal(row, isNull);
+  if (isNull)
+    return false;
+
+#ifdef MYSQL_GE_1009
+  int arrayCounters[JSON_DEPTH_LIMIT];
+  bool hasNegPath = false;
+#endif
+  const int argSize = fp.size() - 2;
+
+  if (!isModeParsed)
+  {
+    if (!isModeConst)
+      isModeConst = (dynamic_cast<ConstantColumn*>(fp[1]->data()) != nullptr);
+
+    string mode = fp[1]->data()->getStrVal(row, isNull);
+    if (isNull)
+      return false;
+
+    transform(mode.begin(), mode.end(), mode.begin(), ::tolower);
+    if (mode != "one" && mode != "all")
+    {
+      isNull = true;
+      return false;
+    }
+
+    isModeOne = (mode == "one");
+    isModeParsed = isModeConst;
+  }
+
+  initJSPaths(paths, fp, 2, 1);
+  if (paths.size() == 0)
+    hasFound.assign(argSize, false);
+
+  for (size_t i = 2; i < fp.size(); i++)
+  {
+    JSONPath& path = paths[i - 2];
+
+    if (!path.parsed)
+    {
+      if (parseJSPath(path, row, fp[i]))
+      {
+        isNull = true;
+        return false;
+      }
+#ifdef MYSQL_GE_1009
+      hasNegPath |= path.p.types_used & JSON_PATH_NEGATIVE_INDEX;
+#endif
+    }
+  }
+
+  json_engine_t jsEg;
+  json_path_t p;
+  json_get_path_start(&jsEg, getCharset(fp[0]), (const uchar*)js.data(), (const uchar*)js.data() + js.size(),
+                      &p);
+
+  bool result = false;
+  int needFound = 0;
+
+  if (!isModeOne)
+  {
+    hasFound.assign(argSize, false);
+    needFound = argSize;
+  }
+
+  while (json_get_path_next(&jsEg, &p) == 0)
+  {
+#ifdef MYSQL_GE_1009
+    if (hasNegPath && jsEg.value_type == JSON_VALUE_ARRAY &&
+        json_skip_array_and_count(&jsEg, arrayCounters + (p.last_step - p.steps)))
+    {
+      result = true;
+      break;
+    }
+#endif
+
+    for (int restSize = argSize, curr = 0; restSize > 0; restSize--, curr++)
+    {
+      JSONPath& path = paths[curr];
+#ifdef MYSQL_GE_1009
+      int cmp = cmpJSPath(&path.p, &p, jsEg.value_type, arrayCounters);
+#else
+      int cmp = cmpJSPath(&path.p, &p, jsEg.value_type);
+#endif
+      if (cmp >= 0)
+      {
+        if (isModeOne)
+        {
+          result = true;
+          break;
+        }
+        /* mode_all */
+        if (hasFound[restSize - 1])
+          continue; /* already found */
+        if (--needFound == 0)
+        {
+          result = true;
+          break;
+        }
+        hasFound[restSize - 1] = true;
+      }
+    }
+  }
+
+  if (likely(jsEg.s.error == 0))
+    return result;
+
+  isNull = true;
+  return false;
+}
+}  // namespace funcexp