mariadb-columnstore-engine/utils/funcexp/func_json_equals.cpp

#include <string_view>
#include <memory>
using namespace std;

#include "functor_json.h"
#include "functioncolumn.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_equals::operationType(FunctionParm& fp,
                                                              CalpontSystemCatalog::ColType& /*resultType*/)
{
  return fp[0]->data()->resultType();
}

/**
 * getBoolVal API definition
 */
bool Func_json_equals::getBoolVal(Row& row, FunctionParm& fp, bool& isNull,
                                  CalpontSystemCatalog::ColType& /*type*/)
{
#if MYSQL_VERSION_ID >= 120100
  json_engine_t je;
  MEM_ROOT_DYNAMIC_ARRAY array;
  int je_stack[JSON_DEPTH_LIMIT];
  struct json_norm_value *buffer_array[JSON_DEPTH_LIMIT];

  initJsonArray(&je.stack, sizeof(int), &je_stack);
  initJsonArray(&array, sizeof(struct json_norm_value*), buffer_array);
#endif

  // auto release the DYNAMIC_STRING
  using DynamicString = unique_ptr<DYNAMIC_STRING, decltype(&dynstr_free)>;

  DynamicString str1{new DYNAMIC_STRING(), dynstr_free};
  if (init_dynamic_string(str1.get(), NULL, 0, 0))
  {
    isNull = true;
    return true;
  }

  DynamicString str2{new DYNAMIC_STRING(), dynstr_free};
  if (init_dynamic_string(str2.get(), NULL, 0, 0))
  {
    isNull = true;
    return true;
  }

  const auto js1_ns = fp[0]->data()->getStrVal(row, isNull);
  if (isNull)
    return false;

  const auto js2_ns = fp[1]->data()->getStrVal(row, isNull);
  if (isNull)
    return false;

  const string_view js1 = js1_ns.unsafeStringRef();
  const string_view js2 = js2_ns.unsafeStringRef();

  bool result = false;

#if MYSQL_VERSION_ID >= 120100
  if (json_normalize(str1.get(), js1.data(), js1.size(), getCharset(fp[0]), NULL, &je, &array))
#else
  if (json_normalize(str1.get(), js1.data(), js1.size(), getCharset(fp[0])))
#endif
  {
    isNull = true;
    return result;
  }

#if MYSQL_VERSION_ID >= 120100
  if (json_normalize(str2.get(), js2.data(), js2.size(), getCharset(fp[1]), NULL, &je, &array))
#else
  if (json_normalize(str2.get(), js2.data(), js2.size(), getCharset(fp[1])))
#endif
  {
    isNull = true;
    return result;
  }

  result = strcmp(str1->str, str2->str) ? false : true;
  return result;
}
}  // namespace funcexp