langzi_xl 发表于 2013-1-19 04:14:02

如何为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[]”里增加:
http://qa.taobao.com/DOCUME%7E1/wuyun.xl/LOCALS%7E1/Temp/moz-screenshot-1.png
http://qa.taobao.com/wp-content/uploads/2011/01/0.jpg
2. 在sql_lex.h中enum_sql_command中增加:
http://qa.taobao.com/wp-content/uploads/2011/01/00004.jpg
3. 在sql_yacc.yy中加上:
http://qa.taobao.com/wp-content/uploads/2011/01/005.jpg
和相应的语法信息:
http://qa.taobao.com/wp-content/uploads/2011/01/006.jpg
4.  然后就是到了比较熟悉的sql_parse.cc中增加相应的路由信息:
http://qa.taobao.com/wp-content/uploads/2011/01/007.jpg
5. 接着我们在sql_show.cc中加上show_disk_usage_command的实现:
http://qa.taobao.com/wp-content/uploads/2011/01/008.jpg
6. 对于windows用户来说,需要在sql_yacc.cc中增加几个声明:
http://qa.taobao.com/wp-content/uploads/2011/01/009.jpg
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测试下,果然可以了:)
http://qa.taobao.com/wp-content/uploads/2011/01/1.jpg
10. 剩下的事就很简单了, 把刚才打的桩子 show_disk_usage_command的实现再细化些即可
参考资料《Expert MySQL》
页: [1]
查看完整版本: 如何为mysql增加自定义命令