如何为mysql增加自定义命令

本文介绍如何在MySQL中新增自定义命令,包括修改源代码、词法及语法分析等步骤,并提供详细的操作流程。

最近在做MySQL源码测试,某同学有个新做的**Patch,会新增加一些自定义的命令,实现策略比较”环保”,主要是考虑到低侵入性,通过某 些回调函数做,所以有些地方看着不优雅,有种”跳线”的感觉,不爽, 我于是找时间顺道学习了下官方比较传统的实现策略。下文仅供学习参考。

其实新增加一条命令,从技术角度讲并不复杂,但比较锁碎。首先需要有一份mysql的源代码,我这边桌面开发机上用的是5.1.45版本的源码,工具Visual Studio 2005、Bison。

原理上讲,对于一条client发过来的query,server端会有一个线程被分配过来处理(如果线程缓存有的话会先从里面拿,否则新建一 个),然后这个SQL会交到parser处理,这个parser会做词法分析和语法分析。parser在mysql中主要是通过bison结合Lex、 YACC实现的。其中Lex主要负责词法分析,YACC负责语法分析。

假设我们现在要新增一条命令,类似于show authors;这样的自定义命令,比较show disk_usage;那么我们就需要先加些符合信息到lex中,再加些命令语法到YACC中(sql_yacc.yy),最后通过bison编译成一个 C文件,再用gen_lex_hash重新生成lex hash文件,下面是详细步骤:

1. 修改lex.h,增加新符号信息,在“static SYMBOL symbols[]”里增加:

2. 在sql_lex.h中enum_sql_command中增加:

3. 在sql_yacc.yy中加上:

和相应的语法信息:

4.  然后就是到了比较熟悉的sql_parse.cc中增加相应的路由信息:

5. 接着我们在sql_show.cc中加上show_disk_usage_command的实现:

6. 对于windows用户来说,需要在sql_yacc.cc中增加几个声明:

7. 代码部分结束。现在我们来更新下sql_yacc.cc & sql_yacc.h。其它这两个文件是由sql_yacc.yy通过bison生成的:) 所以让我们到/sql目录下:

bison -y -d sql_yacc.yy

会生成两个新文件y.tab.c 和y.tab.h,分别替换掉之前原有的sql_yacc.cc和sql_yacc.h即可

8. 现在重新生成下lex hash:

gen_lex_hash > lex_hash.h 然后替换掉/sql下相应的同名文件即可

9. 现在已经大功告成,重新编译一把mysqld工程,然后从client测试下,果然可以了:)

10. 剩下的事就很简单了, 把刚才打的桩子 show_disk_usage_command的实现再细化些即可

参考资料《Expert MySQL》

### MySQL 自定义变量赋值方法 在 MySQL 中,可以通过多种方式为自定义变量赋值。以下是几种常见的方法及其示例: #### 使用 `SET` 语句赋值 `SET` 是一种简单而直观的方式用于给用户自定义变量赋值。语法如下: ```sql SET @variable_name = value; ``` 或者也可以写成: ```sql SET @variable_name := value; ``` 例如,设置一个名为 `@myVar` 的变量并赋予整数值 `100`: ```sql SET @myVar = 100; SELECT @myVar; -- 输出:100 ``` 这种方法适用于直接赋值场景[^3]。 #### 使用 `SELECT ... INTO` 结构赋值 如果需要将查询的结果存储到变量中,则可以使用 `SELECT ... INTO` 语句。其基本形式为: ```sql SELECT col_name INTO variable_name FROM table_name WHERE condition; ``` 下面的例子展示了如何从表中读取数据并将结果存入变量: 假设有一个名为 `employees` 的表,其中包含员工姓名和工资字段 (`name`, `salary`)。 ```sql SELECT salary INTO @emp_salary FROM employees WHERE id = 1; SELECT @emp_salary; -- 假设返回该员工的薪水金额 ``` 此方法特别适合处理单行查询结果的情况[^4]。 #### 在 SQL 查询内部动态赋值 除了显式的 `SET` 或者 `SELECT ... INTO` 外,在某些情况下可以直接利用表达式来完成赋值操作。比如执行一条带有子查询的选择命令时顺便初始化几个局部参数: ```sql SELECT (@rownum:=@rownum+1) AS row_number, t.* FROM some_table t, (SELECT @rownum := 0) r; -- 这里我们创建了一个计数器(@rownum),随着每一行记录增加它的值,并将其作为新列(row_number)展示出来。 ``` 上述片段实现了类似数据库内置ROW_NUMBER()窗口功能的效果[^5]。 综上所述,MySQL 提供了灵活多样的手段让用户能够方便快捷地管理自己的临时状态信息——即所谓的“用户定义变量”。无论是简单的常量设定还是复杂的业务逻辑推导都可以借助这些工具轻松达成目标。 ---
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值