mirror/mariadb
1
0
Fork 0
mirror of https://github.com/MariaDB/server.git synced 2025-09-27 03:19:20 +02:00
Code Issues Projects Releases Packages Wiki Activity
mariadb/sql/item_jsonfunc.h
Rucha Deodhar b8a7728963 MDEV-32854: Make JSON_DEPTH_LIMIT unlimited 
The initial hard capped limit on the depth was 32. It was implemented using
static arrays of relevant type and size 32.
Hence, to implement unlimited depth, dynamic array on mem_root was
implemented which grows by 3200 as needed. Relevant arrays were replaced
with this dynamic array.
2025-09-16 10:10:08 +05:30

1034 lines
29 KiB
C++
Raw Permalink Blame History

#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;
MEM_ROOT_DYNAMIC_ARRAY *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(MEM_ROOT *mem_root, CHARSET_INFO *i_cs, const uchar *str,
const uchar *end)
{
json_scan_start(this, i_cs, str, end);
}
Json_engine_scan(MEM_ROOT *mem_root, const String &str)
:Json_engine_scan(mem_root, str.charset(), (const uchar *) str.ptr(),
(const uchar *) str.end())
{ }
Json_engine_scan(){}
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:
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
String tmp_js, tmp_path;
Json_engine_scan je;
~Json_path_extractor() { }
virtual bool check_and_get_value(Json_engine_scan *je,
String *to, int *error)=0;
bool extract(MEM_ROOT *mem_root, String *to, Item *js,
Item *jp, CHARSET_INFO *cs,
MEM_ROOT_DYNAMIC_ARRAY *,
LEX_CSTRING *func_name);
void init_json_engine_stack(MEM_ROOT *mem_root);
public:
Json_path_extractor() {}
};
class Item_func_json_valid: public Item_bool_func
{
protected:
String tmp_value;
json_engine_t je;
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;
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
{
json_engine_t je1, je2, temp_je;
MEM_ROOT_DYNAMIC_ARRAY stack;
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:
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
json_path_with_flags path;
String tmp_js, tmp_path;
json_engine_t je;
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
{
LEX_CSTRING func_name_str;
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;
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
{
LEX_CSTRING func_name_str;
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;
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;
json_engine_t je;
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
{
private:
json_engine_t je;
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)
{}
~Item_json_str_multipath();
bool fix_fields(THD *thd, Item **ref) override;
bool fix_length_and_dec(THD* thd) override;
virtual uint get_n_paths() const = 0;
};
class Item_func_json_extract: public Item_json_str_multipath
{
protected:
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
String tmp_js;
json_path_t p;
json_engine_t je, sav_je;
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:
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
String tmp_js;
json_path_with_flags path;
String tmp_path;
bool a2_constant, a2_parsed;
String tmp_val, *val;
json_engine_t je, ve;
int loc_js_stack_buffer[JSON_DEPTH_LIMIT];
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); }
int check_contains(json_engine_t *je, json_engine_t *ve);
};
class Item_func_json_contains_path: public Item_bool_func
{
protected:
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
String tmp_js;
json_path_with_flags *paths;
String *tmp_paths;
bool mode_one;
bool ooa_constant, ooa_parsed;
bool *p_found;
json_path_t p;
json_engine_t je;
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;
json_engine_t je;
public:
Item_func_json_array(THD *thd):
Item_json_func(thd)
{result_limit= 0;}
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:
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
String tmp_js;
String tmp_val;
json_engine_t je;
MEM_ROOT current_mem_root;
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;
json_engine_t je2;
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;
}
bool fix_length_and_dec(THD *thd) override;
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
{
json_engine_t je, temp_je;
MEM_ROOT_DYNAMIC_ARRAY stack;
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;
json_engine_t je;
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;
json_engine_t je;
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
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;
json_engine_t je;
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;
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;
json_engine_t je;
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;
json_engine_t je;
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
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;
json_engine_t je;
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
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;
json_engine_t je;
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
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, temp;
bool mode_one;
bool ooa_constant, ooa_parsed;
int escape;
int n_path_found;
json_path_t p, sav_path;
json_engine_t je;
MEM_ROOT_DYNAMIC_ARRAY json_depth_array;
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;
json_engine_t je;
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
{
private:
/*
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;
json_engine_t je, ve, temp_je;
MEM_ROOT_DYNAMIC_ARRAY stack;
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;
json_engine_t je, ve;
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, tmp_val;
json_engine_t je, je_scan;
LEX_CSTRING func_name_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;
bool hash_inited, hash_root_inited;
HASH items;
MEM_ROOT hash_root;
bool parse_for_each_row;
json_engine_t je1, je2, res_je, temp_je;
MEM_ROOT_DYNAMIC_ARRAY stack;
public:
Item_func_json_array_intersect(THD *thd, Item *a, Item *b):
Item_str_func(thd, a, b)
{
hash_inited= hash_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 (hash_inited)
my_hash_free(&items);
if (hash_root_inited)
free_root(&hash_root, MYF(0));
Item_str_func::cleanup();
}
void prepare_json_and_create_hash(json_engine_t *je1, String *js);
};
class Item_func_json_object_filter_keys: public Item_str_func
{
protected:
String tmp_js1, tmp_js2;
bool hash_inited, hash_root_inited;
HASH items;
MEM_ROOT hash_root;
json_engine_t je1, je_res, temp_je, temp_je2;
MEM_ROOT_DYNAMIC_ARRAY stack;
public:
Item_func_json_object_filter_keys(THD *thd, Item *a, Item *b):
Item_str_func(thd, a, b)
{
hash_inited= hash_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
{
if (hash_inited)
my_hash_free(&items);
if (hash_root_inited)
free_root(&hash_root, MYF(0));
Item_str_func::cleanup();
}
};
#endif /* ITEM_JSONFUNC_INCLUDED */
Reference in a new issue View git blame Copy permalink