mirror of
https://github.com/MariaDB/server.git
synced 2025-11-08 00:28:29 +03:00
Analysis: So, there were two problems that needed to be fixed. 1) To fix the crash. 2) After fixing the crash, the result was wrong. Reason for crash: When we pass the hash to get_intersect_between_arrays(), We were initialially not passing it value, so the operations were not performed correctly. Reason for wrong result: The number of rows that it was returning were same as that in the table, but, only the first row had correct ouput, rest of them were NULL (it should also be the result of interection). This was because we modified the "items" HASH by deleting the "seen" elements. So for next rows, it did not have the elements it should have in the hash. Fix: 1) To fix the crash: pass the HASH by reference 2) To fix incorrect result: Maintain a separate "seen" hash, if an item is found the the "items" hash, delete it ony temporarily and put it in the seen hash. At then end, put the items from "seen" back into "items" and reset "seen".
949 lines
28 KiB
C++
949 lines
28 KiB
C++
#ifndef ITEM_JSONFUNC_INCLUDED
|
|
#define ITEM_JSONFUNC_INCLUDED
|
|
|
|
/* Copyright (c) 2016, 2021, MariaDB
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; version 2 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
|
|
|
|
|
|
/* This file defines all JSON functions */
|
|
|
|
|
|
#include <json_lib.h>
|
|
#include "item_cmpfunc.h" // Item_bool_func
|
|
#include "item_strfunc.h" // Item_str_func
|
|
#include "item_sum.h"
|
|
#include "sql_type_json.h"
|
|
#include "json_schema.h"
|
|
|
|
class json_path_with_flags
|
|
{
|
|
public:
|
|
json_path_t p;
|
|
bool constant;
|
|
bool parsed;
|
|
json_path_step_t *cur_step;
|
|
void set_constant_flag(bool s_constant)
|
|
{
|
|
constant= s_constant;
|
|
parsed= FALSE;
|
|
}
|
|
};
|
|
|
|
|
|
void report_path_error_ex(const char *ps, json_path_t *p,
|
|
const char *fname, int n_param,
|
|
Sql_condition::enum_warning_level lv);
|
|
void report_json_error_ex(const char *js, json_engine_t *je,
|
|
const char *fname, int n_param,
|
|
Sql_condition::enum_warning_level lv);
|
|
|
|
class Json_engine_scan: public json_engine_t
|
|
{
|
|
public:
|
|
Json_engine_scan(CHARSET_INFO *i_cs, const uchar *str, const uchar *end)
|
|
{
|
|
json_scan_start(this, i_cs, str, end);
|
|
}
|
|
Json_engine_scan(const String &str)
|
|
:Json_engine_scan(str.charset(), (const uchar *) str.ptr(),
|
|
(const uchar *) str.end())
|
|
{ }
|
|
bool check_and_get_value_scalar(String *res, int *error);
|
|
bool check_and_get_value_complex(String *res, int *error,
|
|
json_value_types cur_value_type=
|
|
JSON_VALUE_UNINITIALIZED);
|
|
};
|
|
|
|
|
|
class Json_path_extractor: public json_path_with_flags
|
|
{
|
|
protected:
|
|
String tmp_js, tmp_path;
|
|
virtual ~Json_path_extractor() { }
|
|
virtual bool check_and_get_value(Json_engine_scan *je,
|
|
String *to, int *error)=0;
|
|
bool extract(String *to, Item *js, Item *jp, CHARSET_INFO *cs);
|
|
};
|
|
|
|
|
|
class Item_func_json_valid: public Item_bool_func
|
|
{
|
|
protected:
|
|
String tmp_value;
|
|
|
|
public:
|
|
Item_func_json_valid(THD *thd, Item *json) : Item_bool_func(thd, json) {}
|
|
bool val_bool() override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_valid") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override
|
|
{
|
|
if (Item_bool_func::fix_length_and_dec(thd))
|
|
return TRUE;
|
|
set_maybe_null();
|
|
return FALSE;
|
|
}
|
|
bool set_format_by_check_constraint(Send_field_extended_metadata *to) const
|
|
override
|
|
{
|
|
static const Lex_cstring fmt(STRING_WITH_LEN("json"));
|
|
return to->set_format_name(fmt);
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_valid>(thd, this); }
|
|
enum Functype functype() const override { return JSON_VALID_FUNC; }
|
|
};
|
|
|
|
|
|
class Item_func_json_equals: public Item_bool_func
|
|
{
|
|
public:
|
|
Item_func_json_equals(THD *thd, Item *a, Item *b):
|
|
Item_bool_func(thd, a, b) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_equals") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_equals>(thd, this); }
|
|
bool val_bool() override;
|
|
};
|
|
|
|
|
|
class Item_func_json_exists: public Item_bool_func
|
|
{
|
|
protected:
|
|
json_path_with_flags path;
|
|
String tmp_js, tmp_path;
|
|
|
|
public:
|
|
Item_func_json_exists(THD *thd, Item *js, Item *i_path):
|
|
Item_bool_func(thd, js, i_path) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_exists") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_exists>(thd, this); }
|
|
bool val_bool() override;
|
|
};
|
|
|
|
|
|
class Item_json_func: public Item_str_func
|
|
{
|
|
public:
|
|
Item_json_func(THD *thd)
|
|
:Item_str_func(thd) { }
|
|
Item_json_func(THD *thd, Item *a)
|
|
:Item_str_func(thd, a) { }
|
|
Item_json_func(THD *thd, Item *a, Item *b)
|
|
:Item_str_func(thd, a, b) { }
|
|
Item_json_func(THD *thd, List<Item> &list)
|
|
:Item_str_func(thd, list) { }
|
|
const Type_handler *type_handler() const override
|
|
{
|
|
return Type_handler_json_common::json_type_handler(max_length);
|
|
}
|
|
};
|
|
|
|
|
|
class Item_func_json_value: public Item_str_func,
|
|
public Json_path_extractor
|
|
{
|
|
|
|
public:
|
|
Item_func_json_value(THD *thd, Item *js, Item *i_path):
|
|
Item_str_func(thd, js, i_path) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_value") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override ;
|
|
String *val_str(String *to) override
|
|
{
|
|
null_value= Json_path_extractor::extract(to, args[0], args[1],
|
|
collation.collation);
|
|
return null_value ? NULL : to;
|
|
}
|
|
bool check_and_get_value(Json_engine_scan *je,
|
|
String *res, int *error) override
|
|
{
|
|
return je->check_and_get_value_scalar(res, error);
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_value>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_query: public Item_json_func,
|
|
public Json_path_extractor
|
|
{
|
|
public:
|
|
Item_func_json_query(THD *thd, Item *js, Item *i_path):
|
|
Item_json_func(thd, js, i_path) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_query") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *to) override
|
|
{
|
|
null_value= Json_path_extractor::extract(to, args[0], args[1],
|
|
collation.collation);
|
|
return null_value ? NULL : to;
|
|
}
|
|
bool check_and_get_value(Json_engine_scan *je,
|
|
String *res, int *error) override
|
|
{
|
|
return je->check_and_get_value_complex(res, error, JSON_VALUE_UNINITIALIZED);
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_query>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_quote: public Item_str_func
|
|
{
|
|
protected:
|
|
String tmp_s;
|
|
|
|
public:
|
|
Item_func_json_quote(THD *thd, Item *s): Item_str_func(thd, s) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_quote") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_quote>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_unquote: public Item_str_func
|
|
{
|
|
protected:
|
|
String tmp_s;
|
|
String *read_json(json_engine_t *je);
|
|
public:
|
|
Item_func_json_unquote(THD *thd, Item *s): Item_str_func(thd, s) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_unquote") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_unquote>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_json_str_multipath: public Item_json_func
|
|
{
|
|
protected:
|
|
json_path_with_flags *paths;
|
|
String *tmp_paths;
|
|
private:
|
|
/**
|
|
Number of paths returned by calling virtual method get_n_paths() and
|
|
remembered inside fix_fields(). It is used by the virtual destructor
|
|
~Item_json_str_multipath() to iterate along allocated memory chunks stored
|
|
in the array tmp_paths and free every of them. The virtual method
|
|
get_n_paths() can't be used for this goal from within virtual destructor.
|
|
We could get rid of the virtual method get_n_paths() and store the number
|
|
of paths directly in the constructor of classes derived from the class
|
|
Item_json_str_multipath but presence of the method get_n_paths() allows
|
|
to check invariant that the number of arguments not changed between
|
|
sequential runs of the same prepared statement that seems to be useful.
|
|
*/
|
|
uint n_paths;
|
|
public:
|
|
Item_json_str_multipath(THD *thd, List<Item> &list):
|
|
Item_json_func(thd, list), paths(NULL), tmp_paths(0), n_paths(0) {}
|
|
virtual ~Item_json_str_multipath();
|
|
|
|
bool fix_fields(THD *thd, Item **ref) override;
|
|
virtual uint get_n_paths() const = 0;
|
|
};
|
|
|
|
|
|
class Item_func_json_extract: public Item_json_str_multipath
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
public:
|
|
String *read_json(String *str, json_value_types *type,
|
|
char **out_val, int *value_len);
|
|
Item_func_json_extract(THD *thd, List<Item> &list):
|
|
Item_json_str_multipath(thd, list) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_extract") };
|
|
return name;
|
|
}
|
|
enum Functype functype() const override { return JSON_EXTRACT_FUNC; }
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
longlong val_int() override;
|
|
double val_real() override;
|
|
my_decimal *val_decimal(my_decimal *) override;
|
|
uint get_n_paths() const override { return arg_count - 1; }
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_extract>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_contains: public Item_bool_func
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
json_path_with_flags path;
|
|
String tmp_path;
|
|
bool a2_constant, a2_parsed;
|
|
String tmp_val, *val;
|
|
public:
|
|
Item_func_json_contains(THD *thd, List<Item> &list):
|
|
Item_bool_func(thd, list) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_contains") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
bool val_bool() override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_contains>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_contains_path: public Item_bool_func
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
json_path_with_flags *paths;
|
|
String *tmp_paths;
|
|
bool mode_one;
|
|
bool ooa_constant, ooa_parsed;
|
|
bool *p_found;
|
|
|
|
public:
|
|
Item_func_json_contains_path(THD *thd, List<Item> &list):
|
|
Item_bool_func(thd, list), tmp_paths(0) {}
|
|
virtual ~Item_func_json_contains_path();
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_contains_path") };
|
|
return name;
|
|
}
|
|
bool fix_fields(THD *thd, Item **ref) override;
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
bool val_bool() override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_contains_path>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_array: public Item_json_func
|
|
{
|
|
protected:
|
|
String tmp_val;
|
|
ulong result_limit;
|
|
public:
|
|
Item_func_json_array(THD *thd):
|
|
Item_json_func(thd) {}
|
|
Item_func_json_array(THD *thd, List<Item> &list):
|
|
Item_json_func(thd, list) {}
|
|
String *val_str(String *) override;
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_array") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_array>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_array_append: public Item_json_str_multipath
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
String tmp_val;
|
|
public:
|
|
Item_func_json_array_append(THD *thd, List<Item> &list):
|
|
Item_json_str_multipath(thd, list) {}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
uint get_n_paths() const override { return arg_count/2; }
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_array_append") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_array_append>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_array_insert: public Item_func_json_array_append
|
|
{
|
|
public:
|
|
Item_func_json_array_insert(THD *thd, List<Item> &list):
|
|
Item_func_json_array_append(thd, list) {}
|
|
String *val_str(String *) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_array_insert") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_array_insert>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_object: public Item_func_json_array
|
|
{
|
|
public:
|
|
Item_func_json_object(THD *thd):
|
|
Item_func_json_array(thd) {}
|
|
Item_func_json_object(THD *thd, List<Item> &list):
|
|
Item_func_json_array(thd, list) {}
|
|
String *val_str(String *) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_object") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_object>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_merge: public Item_func_json_array
|
|
{
|
|
protected:
|
|
String tmp_js1, tmp_js2;
|
|
public:
|
|
Item_func_json_merge(THD *thd, List<Item> &list):
|
|
Item_func_json_array(thd, list) {}
|
|
String *val_str(String *) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_merge_preserve") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_merge>(thd, this); }
|
|
};
|
|
|
|
class Item_func_json_merge_patch: public Item_func_json_merge
|
|
{
|
|
public:
|
|
Item_func_json_merge_patch(THD *thd, List<Item> &list):
|
|
Item_func_json_merge(thd, list) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_merge_patch") };
|
|
return name;
|
|
}
|
|
String *val_str(String *) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_merge_patch>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_normalize: public Item_json_func
|
|
{
|
|
public:
|
|
Item_func_json_normalize(THD *thd, Item *a):
|
|
Item_json_func(thd, a) {}
|
|
String *val_str(String *) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_normalize") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_normalize>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_object_to_array: public Item_json_func
|
|
{
|
|
protected:
|
|
String tmp;
|
|
public:
|
|
Item_func_json_object_to_array(THD *thd, Item *a):
|
|
Item_json_func(thd, a) {}
|
|
String *val_str(String *) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_object_to_array") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_object_to_array>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_length: public Item_long_func
|
|
{
|
|
bool check_arguments() const override
|
|
{
|
|
const LEX_CSTRING name= func_name_cstring();
|
|
if (arg_count == 0 || arg_count > 2)
|
|
{
|
|
my_error(ER_WRONG_PARAMCOUNT_TO_NATIVE_FCT, MYF(0), name.str);
|
|
return true;
|
|
}
|
|
return args[0]->check_type_can_return_text(name) ||
|
|
(arg_count > 1 && args[1]->check_type_general_purpose_string(name));
|
|
}
|
|
protected:
|
|
json_path_with_flags path;
|
|
String tmp_js;
|
|
String tmp_path;
|
|
public:
|
|
Item_func_json_length(THD *thd, List<Item> &list):
|
|
Item_long_func(thd, list) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_length") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
longlong val_int() override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_length>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_depth: public Item_long_func
|
|
{
|
|
bool check_arguments() const override
|
|
{ return args[0]->check_type_can_return_text(func_name_cstring()); }
|
|
protected:
|
|
String tmp_js;
|
|
public:
|
|
Item_func_json_depth(THD *thd, Item *js): Item_long_func(thd, js) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_depth") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override { max_length= 10; return FALSE; }
|
|
longlong val_int() override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_depth>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_type: public Item_str_func
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
public:
|
|
Item_func_json_type(THD *thd, Item *js): Item_str_func(thd, js) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_type") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_type>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_insert: public Item_json_str_multipath
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
String tmp_val;
|
|
bool mode_insert, mode_replace;
|
|
public:
|
|
Item_func_json_insert(bool i_mode, bool r_mode, THD *thd, List<Item> &list):
|
|
Item_json_str_multipath(thd, list),
|
|
mode_insert(i_mode), mode_replace(r_mode) {}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
uint get_n_paths() const override { return arg_count/2; }
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING json_set= {STRING_WITH_LEN("json_set") };
|
|
static LEX_CSTRING json_insert= {STRING_WITH_LEN("json_insert") };
|
|
static LEX_CSTRING json_replace= {STRING_WITH_LEN("json_replace") };
|
|
return (mode_insert ?
|
|
(mode_replace ? json_set : json_insert) : json_replace);
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_insert>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_remove: public Item_json_str_multipath
|
|
{
|
|
protected:
|
|
String tmp_js;
|
|
public:
|
|
Item_func_json_remove(THD *thd, List<Item> &list):
|
|
Item_json_str_multipath(thd, list) {}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
uint get_n_paths() const override { return arg_count - 1; }
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_remove") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_remove>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_keys: public Item_str_func
|
|
{
|
|
protected:
|
|
json_path_with_flags path;
|
|
String tmp_js, tmp_path;
|
|
|
|
public:
|
|
Item_func_json_keys(THD *thd, List<Item> &list):
|
|
Item_str_func(thd, list) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_keys") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_keys>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_search: public Item_json_str_multipath
|
|
{
|
|
protected:
|
|
String tmp_js, tmp_path, esc_value;
|
|
bool mode_one;
|
|
bool ooa_constant, ooa_parsed;
|
|
int escape;
|
|
int n_path_found;
|
|
json_path_t sav_path;
|
|
|
|
int compare_json_value_wild(json_engine_t *je, const String *cmp_str);
|
|
|
|
public:
|
|
Item_func_json_search(THD *thd, List<Item> &list):
|
|
Item_json_str_multipath(thd, list) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_search") };
|
|
return name;
|
|
}
|
|
bool fix_fields(THD *thd, Item **ref) override;
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *) override;
|
|
uint get_n_paths() const override { return arg_count > 4 ? arg_count - 4 : 0; }
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_search>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_format: public Item_json_func
|
|
{
|
|
public:
|
|
enum formats
|
|
{
|
|
NONE,
|
|
COMPACT,
|
|
LOOSE,
|
|
DETAILED
|
|
};
|
|
protected:
|
|
formats fmt;
|
|
String tmp_js;
|
|
public:
|
|
Item_func_json_format(THD *thd, Item *js, formats format):
|
|
Item_json_func(thd, js), fmt(format) {}
|
|
Item_func_json_format(THD *thd, List<Item> &list):
|
|
Item_json_func(thd, list), fmt(DETAILED) {}
|
|
|
|
LEX_CSTRING func_name_cstring() const override;
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *str) override;
|
|
String *val_json(String *str) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_format>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_arrayagg : public Item_func_group_concat
|
|
{
|
|
protected:
|
|
/*
|
|
Overrides Item_func_group_concat::skip_nulls()
|
|
NULL-s should be added to the result as JSON null value.
|
|
*/
|
|
bool skip_nulls() const override { return false; }
|
|
String *get_str_from_item(Item *i, String *tmp) override;
|
|
String *get_str_from_field(Item *i, Field *f, String *tmp,
|
|
const uchar *key, size_t offset) override;
|
|
void cut_max_length(String *result,
|
|
uint old_length, uint max_length) const override;
|
|
public:
|
|
String m_tmp_json; /* Used in get_str_from_*.. */
|
|
Item_func_json_arrayagg(THD *thd, Name_resolution_context *context_arg,
|
|
bool is_distinct, List<Item> *is_select,
|
|
const SQL_I_List<ORDER> &is_order, String *is_separator,
|
|
bool limit_clause, Item *row_limit, Item *offset_limit):
|
|
Item_func_group_concat(thd, context_arg, is_distinct, is_select, is_order,
|
|
is_separator, limit_clause, row_limit, offset_limit)
|
|
{
|
|
}
|
|
Item_func_json_arrayagg(THD *thd, Item_func_json_arrayagg *item) :
|
|
Item_func_group_concat(thd, item) {}
|
|
const Type_handler *type_handler() const override
|
|
{
|
|
return Type_handler_json_common::json_type_handler_sum(this);
|
|
}
|
|
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_arrayagg(") };
|
|
return name;
|
|
}
|
|
bool fix_fields(THD *thd, Item **ref) override;
|
|
enum Sumfunctype sum_func() const override { return JSON_ARRAYAGG_FUNC; }
|
|
|
|
String* val_str(String *str) override;
|
|
|
|
Item *copy_or_same(THD* thd) override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_arrayagg>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_json_objectagg : public Item_sum
|
|
{
|
|
String result;
|
|
public:
|
|
Item_func_json_objectagg(THD *thd, Item *key, Item *value) :
|
|
Item_sum(thd, key, value)
|
|
{
|
|
quick_group= FALSE;
|
|
result.append('{');
|
|
}
|
|
|
|
Item_func_json_objectagg(THD *thd, Item_func_json_objectagg *item);
|
|
void cleanup() override;
|
|
|
|
enum Sumfunctype sum_func () const override { return JSON_OBJECTAGG_FUNC;}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_objectagg") };
|
|
return name;
|
|
}
|
|
const Type_handler *type_handler() const override
|
|
{
|
|
return Type_handler_json_common::json_type_handler_sum(this);
|
|
}
|
|
void clear() override;
|
|
bool add() override;
|
|
void reset_field() override { DBUG_ASSERT(0); } // not used
|
|
void update_field() override { DBUG_ASSERT(0); } // not used
|
|
bool fix_fields(THD *,Item **) override;
|
|
|
|
double val_real() override { return 0.0; }
|
|
longlong val_int() override { return 0; }
|
|
my_decimal *val_decimal(my_decimal *decimal_value) override
|
|
{
|
|
my_decimal_set_zero(decimal_value);
|
|
return decimal_value;
|
|
}
|
|
bool get_date(THD *thd, MYSQL_TIME *ltime, date_mode_t fuzzydate) override
|
|
{
|
|
return get_date_from_string(thd, ltime, fuzzydate);
|
|
}
|
|
String* val_str(String* str) override;
|
|
Item *copy_or_same(THD* thd) override;
|
|
void no_rows_in_result() override {}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_objectagg>(thd, this); }
|
|
};
|
|
|
|
extern bool is_json_type(const Item *item);
|
|
|
|
class Item_func_json_overlaps: public Item_bool_func
|
|
{
|
|
String tmp_js;
|
|
bool a2_constant, a2_parsed;
|
|
String tmp_val, *val;
|
|
public:
|
|
Item_func_json_overlaps(THD *thd, Item *a, Item *b):
|
|
Item_bool_func(thd, a, b)
|
|
{}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_overlaps") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
bool val_bool() override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_overlaps>(thd, this); }
|
|
};
|
|
|
|
class Item_func_json_schema_valid: public Item_bool_func
|
|
{
|
|
String tmp_js;
|
|
bool schema_parsed;
|
|
String tmp_val, *val;
|
|
List<Json_schema_keyword> keyword_list;
|
|
List<Json_schema_keyword> all_keywords;
|
|
|
|
public:
|
|
Item_func_json_schema_valid(THD *thd, Item *a, Item *b):
|
|
Item_bool_func(thd, a, b)
|
|
{
|
|
val= NULL;
|
|
schema_parsed= false;
|
|
set_maybe_null();
|
|
}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_schema_valid") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
bool val_bool() override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_schema_valid>(thd, this); }
|
|
void cleanup() override;
|
|
};
|
|
|
|
class Item_func_json_key_value: public Item_json_func,
|
|
public Json_path_extractor
|
|
{
|
|
|
|
String tmp_str;
|
|
|
|
public:
|
|
Item_func_json_key_value(THD *thd, Item *js, Item *i_path):
|
|
Item_json_func(thd, js, i_path) {}
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_key_value") };
|
|
return name;
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
String *val_str(String *to) override;
|
|
bool check_and_get_value(Json_engine_scan *je,
|
|
String *res, int *error) override
|
|
{
|
|
return je->check_and_get_value_complex(res, error, JSON_VALUE_OBJECT);
|
|
}
|
|
bool get_key_value(json_engine_t *je, String *str);
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_key_value>(thd, this); }
|
|
};
|
|
|
|
|
|
|
|
class Item_func_json_array_intersect: public Item_str_func
|
|
{
|
|
protected:
|
|
String tmp_js1, tmp_js2, temp_str;
|
|
bool item_hash_inited, seen_hash_inited, root_inited;
|
|
HASH items, seen;
|
|
MEM_ROOT hash_root;
|
|
bool parse_for_each_row;
|
|
public:
|
|
Item_func_json_array_intersect(THD *thd, Item *a, Item *b):
|
|
Item_str_func(thd, a, b)
|
|
{ item_hash_inited= seen_hash_inited= root_inited= parse_for_each_row= false; }
|
|
String *val_str(String *) override;
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_array_intersect") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_array_intersect>(thd, this); }
|
|
void cleanup() override
|
|
{
|
|
Item_str_func::cleanup();
|
|
if (item_hash_inited)
|
|
my_hash_free(&items);
|
|
if (seen_hash_inited)
|
|
my_hash_free(&seen);
|
|
if (root_inited)
|
|
free_root(&hash_root, MYF(0));
|
|
}
|
|
bool prepare_json_and_create_hash(json_engine_t *je1, String *js);
|
|
bool get_intersect_between_arrays(String *str, json_engine_t *value,
|
|
HASH *items, HASH *seen);
|
|
};
|
|
|
|
class Item_func_json_object_filter_keys: public Item_str_func
|
|
{
|
|
protected:
|
|
String tmp_js1, tmp_js2;
|
|
bool hash_inited, root_inited;
|
|
HASH items;
|
|
MEM_ROOT hash_root;
|
|
public:
|
|
Item_func_json_object_filter_keys(THD *thd, Item *a, Item *b):
|
|
Item_str_func(thd, a, b) { hash_inited= root_inited= false; }
|
|
String *val_str(String *) override;
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= {STRING_WITH_LEN("json_object_filter_keys") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_json_object_filter_keys>(thd, this); }
|
|
|
|
void cleanup() override
|
|
{
|
|
Item_str_func::cleanup();
|
|
if (hash_inited)
|
|
my_hash_free(&items);
|
|
if (root_inited)
|
|
free_root(&hash_root, MYF(0));
|
|
}
|
|
};
|
|
|
|
|
|
#endif /* ITEM_JSONFUNC_INCLUDED */
|