Mysql自动填充测试数据
前言
最近写了两个小脚本，一个应用于Mysql的自动填充测试数据，另外一个是bash写的定期删除日志文件，两个脚本如何使用，在GitHub上面都有所说明，这里不再赘述，这里主要是想聊一下Mysql的存储过程以及自动填充测试数据。
为什么要写一个自动填充测试数据的脚本？
网上其实也有一些简单的给Mysql填充数据的博客，但是大多数都是针对于特定的表的特定数据来实现，写的过于简单，实用性不强，而这个脚本可以根据我们提供表的字段，来自动识别我们的字段并填充进入对应的内容，常用的结构都能够满足，当然还有进一步完善的空间。
存储过程
Mysql的存储过程可以帮助我们实现一些较为复杂的业务逻辑，就像我们在PHP或者其他语言中所写的逻辑一样，在Mysql中也同样可以执行，比如我们要循环写入1000行数据。
不过虽然Mysql可以实现，但是我们更希望把业务逻辑建立在我们的业务语言上面，而Mysql则专注于处理数据的CURD，其主要原因在于我们今后在修改或者要查找到这块的业务逻辑时，会相对麻烦一些。
因此，我更倾向于Mysql可以用来做一些与我们的业务逻辑无关，而又需要用到语言的逻辑性的项目，比如我在前面提到的自动填充数据。
形式
delimiter $$
drop procedure if exists test $$
create procedure test()
begin
-- do something
end $$
delimiter ;
这是一个自动识别表结构并填充数据的脚本，在第一行使用了delimiter $$用作分隔符，因为接下来的脚本里面很多地方会使用到;，所以为防止过早的识别结束，暂时修改了Mysql默认的分隔符。
此外，前面有begin和end包裹，用于识别开始和结束。
暂时先不讲逻辑怎么实现的(其实逻辑也非常简单)，我们先来了解一下Mysql的存储过程的几个部分。
注:通过show create procedure your_procedure_name可以查看创建的存储过程的代码内容。
变量
Mysql的变量分成三种:全局变量，用户变量，局部变量。
- 局部变量
局部变量存在于我们的存储过程中，在外面无法访问:
delimiter $$
drop procedure if exists test $$
create procedure test()
begin
declare myName varchar(8) default 'seven';
set myName = 'nine';
select myName;
end $$
delimiter ;
call test();
结果输出nine，因为在用户变量中需要@来声明，所以此时在外面无法使用select myName，否则报错。
- 用户变量
用户变量存在于全局，但是有效期仅限于会话期(所以也可以叫做会话变量)，即我们下次打开Mysql时，变量就会消失:
delimiter $$
drop procedure if exists test $$
create procedure test()
begin
set @myName = 'nine';
end $$
delimiter ;
call test();
select @myName ;
输出nine。
当然，赋值的形式是多样的，我们也可以结合查询语句来赋值:
select name, age from user limit 1 into @myName , @myAge;
select @myName , @myAge;
这个时候赋值的内容就是我们表中的数据，这里需要注意的是一一对应关系，切不可一对多或者多对一。
在某些时候，我们可能需要对查询的次数进行记录，其实这个时候我们可以完全使用变量来帮我们实现:
set @selectNum = 0;
select * , @selectNum := @selectNum + 1 from user limit 1;
这个时候@selectNum的结果为2。
- 全局变量
全局变量是设置在系统中的配置，我们可以通过show global variables来查看，也可以通过set global oneofvariable=value来设置我们已经存在的配置，这里我特意给已经存在这几个字眼加粗，因为我在网上看到有博文说可以设置自定义的全局变量，但是我尝试了之后发现报错了，我用版本5.5和5.7都尝试过了，提示这个变量不存在。
参数
参数分成IN，OUT以及INOUT三种情况:
其实从其字面上我们也能猜出他们的不同效果:IN是不会影响外面设置的结果(IN是默认方式)，OUT和INOUT是会影响到的，同时INOUT两边是相互影响的，我们还是以上面的test为例:
- in
set @num = 1;
delimiter $$
drop procedure if exists test $$
create procedure test(num int)
begin
set num = 2;
end $$
delimiter ;
call test(@num);
select @num;
结果为1。
- out
set @num = 1;
delimiter $$
drop procedure if exists test $$
create procedure test(OUT num int)
begin
set num = 2;
end $$
delimiter ;
call test(@num);
select @num;
结果为2。
语句块
这篇文章关于语句块已经阐述的足够详细，这里不再赘述。
实现
代码:
delimiter $$
drop procedure if exists fillTable $$
create procedure fillTable(in num int , in tbName varchar(16))
begin
-- 获取当前数据库
select (@dbName:=database());
set @tbName = tbName;
-- 获取表的字段总数
set @currSql = "select count(1) from information_schema.COLUMNS where table_name = ? and table_schema = ? into @columnSum";
prepare stmt from @currSql;
execute stmt using @tbName , @dbName;
deallocate prepare stmt;
set @currNum = 0;
-- 这里设置随机的字符串
set @chars = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
while @currNum < num do
-- 这里设置sql后面拼接
set @insertSql = concat("insert into " , @tbName , " values ( ");
set @columnNum = 1;
while @columnNum <= @columnSum do
set @value := '';
set @currSql = "select (@column := COLUMN_NAME) , (@length := CHARACTER_MAXIMUM_LENGTH) , (@key := COLUMN_KEY) , (@type := DATA_TYPE) from information_schema.COLUMNS where table_name = ? and table_schema = ? limit ?";
prepare stmt from @currSql;
execute stmt using @tbName , @dbName , @columnNum;
deallocate prepare stmt;
-- 根据类型来填充数据
if right(@type , 3) = 'int' then
if @type = 'int' then
set @value = 'default';
else
set @value = floor(rand() * 100);
end if;
elseif right(@type , 4) = 'char' then
set @counter = 0;
while @counter < @length do
set @value = concat(@value,substr(@chars,ceil(rand()*(length(@chars)-1)),1));
set @counter = @counter + 1;
end while;
set @value = concat("'" , @value , "'");
elseif @type = 'blob' or right(@type , 4) = 'text' then
set @counter = 0;
while @counter < 100 do
set @value = concat(@value,substr(@chars,ceil(rand()*(length(@chars)-1)),1));
set @counter = @counter + 1;
end while;
set @value = concat("'" , @value , "'");
elseif @type = 'float' or @type = 'decimal' then
set @value = round(rand() , 2);
else
set @value = 'nine';
end if;
-- 判断这个数是否是最后一个
if @columnNum = @columnSum then
set @insertSql = concat(@insertSql , @value , ' )');
else
set @insertSql = concat(@insertSql , @value , ' , ');
end if;
set @columnNum = @columnNum + 1;
end while;
-- 执行
prepare stmt from @insertSql;
execute stmt;
deallocate prepare stmt;
set @currNum = @currNum + 1;
end while;
end $$
delimiter ;
其实实现这个功能的逻辑非常简单，再各个步骤里面也附上了步骤，主要是利用了系统的information_schema.COLUMNS表来获取我们需要的一些基本信息，主要结构如下图所示:
