如何給MySQL添加自定義語法的方法示例
更新時間：2022年08月05日 15:22:50 作者：秋風五丈原
本文主要介紹了如何給MySQL添加自定義語法的方法示例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
1 背景
MySQL語法解析器用的bison（即yacc）來實現(xiàn)的，而詞法解析是自己來實現(xiàn)的，涉及到的token都在文件lex.h里面，然后通過Lex_input_stream 里面相關的函數(shù)，解析client的sql字節(jié)流（其中會通過提前構造好的hash表幫助快速找到對應symbol，相關代碼在sql_lex_hash.cc里面），轉換為token，交給bison進行語法解析。
為了給MySQL添加一個新的語法，我們必須添加新的token（如果有新增），以及增加新的語法（sql_yacc.yy）里面。本文以給create table增加一個新的options為例，來演示如何給MySQL新增一個語法。最終的效果如下：
create table t1 ( id int primary key, name varchar(100) ) global_partition by hash(id) partitions 10; //global_partition by為新增語法，global_partition為新增token
涉及到的修改文件如下：
sql/lex.h //token sql/parse_tree_nodes.cc sql/parse_tree_nodes.h sql/parse_tree_partitions.cc sql/parse_tree_partitions.h sql/parser_yystype.h sql/sql_yacc.yy
2 新增關鍵詞（token）
文件：sql/lex.h
static const SYMBOL symbols[] = {
/*
Insert new SQL keywords after that commentary (by alphabetical order):
*/
//省略部分代碼
{SYM("GLOBAL_PARTITION", GLOBAL_PARTITION_SYM)}, //注意按照字典序進行添加。
//省略部分代碼
};
按照上面的格式添加即可
3 新增語法
文件：sql/sql_yacc.yy
該文件為bison的語法，關于bison語法可以查看這里。下面凡是注釋標有###為新增部分，沒有標的注釋是為了方便理解
%token<lexer.keyword> GLOBAL_PARTITION_SYM 1309 /* seancheer */ //### 聲明上一步添加的token，聲明了才可以使用，編號1309選擇一個未使用的就行
%type <global_partition_clause> global_partition_clause //### 聲明新增加的數(shù)據(jù)結構，后面會介紹
create_table_stmt:
CREATE opt_temporary TABLE_SYM opt_if_not_exists table_ident
'(' table_element_list ')' opt_create_table_options_etc //最后一個標記在YYSTYPE中對應的是create_table_tail， 后面會介紹
{
$$= NEW_PTN PT_create_table_stmt(YYMEM_ROOT, $1, $2, $4, $5,
$7,
$9.opt_create_table_options,
$9.opt_partitioning,
$9.opt_global_partitioning, //### 賦值給對應參數(shù)，該構造函數(shù)需要新增，后面會介紹
$9.on_duplicate,
$9.opt_query_expression);
}
| CREATE opt_temporary TABLE_SYM opt_if_not_exists table_ident
opt_create_table_options_etc
{
$$= NEW_PTN PT_create_table_stmt(YYMEM_ROOT, $1, $2, $4, $5,
NULL,
$6.opt_create_table_options,
$6.opt_partitioning,
$6.opt_global_partitioning, //### 賦值給對應參數(shù)，該構造函數(shù)需要新增，后面會介紹
$6.on_duplicate,
$6.opt_query_expression);
//partition相關的語法
opt_create_partitioning_etc:
partition_clause opt_duplicate_as_qe //這里是原生的partition表語法
{
$$= $2;
$$.opt_partitioning= $1;
}
| global_partition_clause opt_duplicate_as_qe //### 此處是新增的global_partition語法，
{
$$= $2;
$$.opt_global_partitioning= $1;
}
| opt_duplicate_as_qe
;
//### 下面為重點，新增的global_partition語法，可以看到，用到了新增的token
global_partition_clause:
GLOBAL_PARTITION_SYM BY part_type_def opt_num_parts
{
$$= NEW_PTN PT_global_partition($3, @4, $4);
}
;
4 類似于PT_partition添加對應的數(shù)據(jù)結構global_partition_clause
文件：parser_yystype.h：該文件是bison（yacc）運行的一環(huán)，代替bison內(nèi)置的YYSTYPE的，當bison對相關語法解析后，需要構造相關的數(shù)據(jù)結構，通過對YYSTYPE的自定義，就可以實現(xiàn)構造自定義數(shù)據(jù)結構的目的了。添加我們自定義的數(shù)據(jù)結構代碼如下：
union YYSTYPE {
PT_sub_partition *opt_sub_part;
PT_part_type_def *part_type_def;
PT_partition *partition_clause;
PT_global_partition *global_partition_clause; //新加數(shù)據(jù)結構
struct {
Mem_root_array<PT_create_table_option *> *opt_create_table_options;
PT_partition *opt_partitioning;
PT_global_partition *opt_global_partitioning; //同時注意添加到create_table_tail里面，因為create table語法會有該操作
On_duplicate on_duplicate;
PT_query_primary *opt_query_expression;
} create_table_tail;
};
static_assert(sizeof(YYSTYPE) <= 40, "YYSTYPE is too big"); //因為struct里面添加了一個成員變量，所以該union需要的空間也會變大，因此注意修改這一行
下面內(nèi)容介紹PT_global_partition數(shù)據(jù)結構，為了保持和MySQL習慣一致，新增加的數(shù)據(jù)結構放在了
sql/parse_tree_nodes.cc sql/parse_tree_nodes.h sql/parse_tree_partitions.cc sql/parse_tree_partitions.h
四個文件里，理論上可以放在任何地方?？筛鶕?jù)自身需求添加對應數(shù)據(jù)結構：
文件：sql/parse_tree_partitions.h sql/parse_tree_partitions.cc
/**
新增數(shù)據(jù)結構
*/
class PT_global_partition : public Parse_tree_node {
typedef Parse_tree_node super;
PT_part_type_def *const part_type_def;
const POS part_defs_pos;
uint num_parts;
public:
partition_info part_info;
public:
PT_global_partition(PT_part_type_def *part_type_def, const POS &part_defs_pos,
uint opt_num_parts)
: part_type_def(part_type_def),
part_defs_pos(part_defs_pos),
num_parts(opt_num_parts) {}
bool contextualize(Parse_context *pc) override;
};
//模仿其原生的實現(xiàn)方式即可
bool PT_global_partition::contextualize(Parse_context *pc) {
if (super::contextualize(pc)) return true;
Partition_parse_context part_pc(pc->thd, &part_info, false);
if (part_type_def->contextualize(&part_pc)) return true;
if (part_info.part_type != partition_type::HASH) {
//only support hash partition for shard key
my_error(ER_PARTITIONS_MUST_BE_DEFINED_ERROR, MYF(0), "NOT HASH");
return true;
}
uint count_curr_parts = part_info.partitions.elements;
if (part_info.num_parts != 0) {
if (part_info.num_parts != count_curr_parts) {
error(&part_pc, part_defs_pos,
ER_THD(pc->thd, ER_PARTITION_WRONG_NO_PART_ERROR));
return true;
}
} else if (count_curr_parts > 0)
part_info.num_parts = count_curr_parts;
return false;
}
文件：sql/parse_tree_nodes.cc sql/parse_tree_nodes.h
接下來修改create table對應的數(shù)據(jù)結構，將新增的PT_global_partition添加到create table里面
class PT_create_table_stmt final : public PT_table_ddl_stmt_base {
PT_partition *opt_partitioning;
PT_global_partition *opt_global_partitioning; //添加成員變量
PT_create_table_stmt(
MEM_ROOT *mem_root, PT_hint_list *opt_hints, bool is_temporary,
bool only_if_not_exists, Table_ident *table_name,
const Mem_root_array<PT_table_element *> *opt_table_element_list,
const Mem_root_array<PT_create_table_option *> *opt_create_table_options,
PT_partition *opt_partitioning,
PT_global_partition *opt_global_partitioning, On_duplicate on_duplicate,
PT_query_primary *opt_query_expression)
: PT_table_ddl_stmt_base(mem_root),
m_opt_hints(opt_hints),
is_temporary(is_temporary),
only_if_not_exists(only_if_not_exists),
table_name(table_name),
opt_table_element_list(opt_table_element_list),
opt_create_table_options(opt_create_table_options),
opt_partitioning(opt_partitioning),
opt_global_partitioning(opt_global_partitioning), //添加構造函數(shù)，主要是為了增加對PT_global_partition的賦值操作
on_duplicate(on_duplicate),
opt_query_expression(opt_query_expression),
opt_like_clause(nullptr) {}
//在其對應的函數(shù)中增加相關邏輯，調(diào)用對應的初始化函數(shù)contextualize
Sql_cmd *PT_create_table_stmt::make_cmd(THD *thd) {
if (opt_global_partitioning){
if (opt_global_partitioning->contextualize(&pc)) return nullptr;
lex->part_info = &opt_global_partitioning->part_info;
}
}
到此這篇關于如何給MySQL添加自定義語法的方法示例的文章就介紹到這了,更多相關MySQL添加自定義語法內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
簡單的說，就是一組SQL語句集，功能強大，可以實現(xiàn)一些比較復雜的邏輯功能，類似于JAVA語言中的方法,這里就為大家簡單介紹一下,需要的朋友可以參考下2018-05-05
這篇文章主要為大家介紹了面試提問mysql一張表到底能存多少數(shù)據(jù)的問題分析，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步2022-03-03
MYSQL ERROR 1045 (28000): Access denied for user (using pass
Mysql中添加用戶之后可能出現(xiàn)登錄時提示ERROR 1045 (28000): Access denied for user的錯誤.2009-07-07
這篇文章主要介紹了MySQL無法存儲emoji表情解決方案,結合實例形式分析了Python爬蟲爬取文章中emoji表情存入數(shù)據(jù)庫的實現(xiàn)方法,涉及mysql utf8mb4編碼的修改相關操作技巧,需要的朋友可以參考下2018-07-07
2012-07-07
這篇文章主要給大家介紹了關于mysql中如何根據(jù)經(jīng)緯度計算距離的相關資料,經(jīng)緯度距離計算方法是一種用來計算地球上兩個點之間距離的方法,它可以在MySQL數(shù)據(jù)庫中輕松實現(xiàn),非常適用于需要進行地理定位的應用程序,需要的朋友可以參考下2023-08-08
這篇文章主要介紹了MySQL中count(*)執(zhí)行慢的解決方案，文章圍繞主題展開詳細的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-06-06
MySQL提示“too?many?connections“錯誤的解決過程
當大量的connect之后,就會出現(xiàn)Too many connections的錯誤,下面這篇文章主要給大家介紹了關于MySQL提示“too?many?connections“錯誤的解決過程,文中通過圖文介紹的非常詳細,需要的朋友可以參考下2022-04-04
這篇文章主要介紹了淺談MySQL大表優(yōu)化方案，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-11-11
這篇文章主要介紹了Mysql慢查詢優(yōu)化方法及優(yōu)化原則，本文給大家介紹的非常詳細，具有一定的參考借鑒價值 ,需要的朋友可以參考下2019-05-05