文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

分享一个批量产生随机测试数据的MySQL存储过程

2024-04-02 19:55

关注

批量产生随机测试数据:

内容如下:


# cat populate.sql 

DELIMITER $$


DROP PROCEDURE IF EXISTS populate $$

CREATE PROCEDURE populate(in_db varchar(50), in_table varchar(50), in_rows int, in_debug char(1)) 

BEGIN


DECLARE col_name VARCHAR(100);

DECLARE col_type VARCHAR(100); 

DECLARE col_datatype VARCHAR(100);

DECLARE col_maxlen VARCHAR(100); 

DECLARE col_extra VARCHAR(100);

DECLARE col_num_precision VARCHAR(100);

DECLARE col_num_scale VARCHAR(100);

DECLARE func_query VARCHAR(1000);

DECLARE i INT;


DECLARE done INT DEFAULT 0;

DECLARE cur_datatype cursor FOR

 SELECT column_name,COLUMN_TYPE,data_type,CHARACTER_MAXIMUM_LENGTH,EXTRA,NUMERIC_PRECISION,NUMERIC_SCALE FROM information_schema.columns WHERE table_name=in_table AND table_schema=in_db;

DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;



SET func_query='';

OPEN cur_datatype;

datatype_loop: loop


FETCH cur_datatype INTO col_name, col_type, col_datatype, col_maxlen, col_extra, col_num_precision, col_num_scale;

#SELECT CONCAT(col_name,"-", col_type,"-", col_datatype,"-", IFNULL(col_maxlen,'NULL'),"-", IFNULL(col_extra,'NULL')) AS VALS;

  IF (done = 1) THEN

    leave datatype_loop;

  END IF;


CASE 

WHEN col_extra='auto_increment' THEN SET func_query=concat(func_query,'NULL, ');

WHEN col_datatype in ('int','bigint') THEN SET func_query=concat(func_query,'get_int(), ');

WHEN col_datatype in ('varchar','char') THEN SET func_query=concat(func_query,'get_string(',ifnull(col_maxlen,0),'), ');

WHEN col_datatype in ('tinyint', 'smallint','year') or col_datatype='mediumint' THEN SET func_query=concat(func_query,'get_tinyint(), ');

WHEN col_datatype in ('datetime','timestamp') THEN SET func_query=concat(func_query,'get_datetime(), ');

WHEN col_datatype in ('date') THEN SET func_query=concat(func_query,'get_date(), ');

WHEN col_datatype in ('float', 'decimal') THEN SET func_query=concat(func_query,'get_float(',col_num_precision,',',col_num_scale,'), ');

WHEN col_datatype in ('enum','set') THEN SET func_query=concat(func_query,'get_enum("',col_type,'"), ');

WHEN col_datatype in ('GEOMETRY','POINT','LINESTRING','POLYGON','MULTIPOINT','MULTILINESTRING','MULTIPOLYGON','GEOMETRYCOLLECTION') THEN SET func_query=concat(func_query,'NULL, ');

ELSE SET func_query=concat(func_query,'get_varchar(',ifnull(col_maxlen,0),'), ');

END CASE;



end loop  datatype_loop;

close cur_datatype;


SET func_query=trim(trailing ', ' FROM func_query);

SET @func_query=concat("INSERT INTO ", in_db,".",in_table," VALUES (",func_query,");");

        IF in_debug='Y' THEN

                select @func_query;

        END IF;

SET i=in_rows;

populate :loop

        WHILE (i>0) DO

          PREPARE t_stmt FROM @func_query;

          EXECUTE t_stmt;

        SET i=i-1;

END WHILE;

LEAVE populate;

END LOOP populate;

SELECT "Kedar Vaijanapurkar" AS "Developed by";

END

$$

DELIMITER ;



       





## MySQL function to generate random string of specified length

DROP function if exists get_string;

delimiter $$

CREATE FUNCTION get_string(in_strlen int) RETURNS VARCHAR(500) DETERMINISTIC

BEGIN 

set @var:='';

while(in_strlen>0) do

set @var:=concat(@var,IFNULL(ELT(1+FLOOR(RAND() * 53), 'a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z',' ','A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z'),'Kedar'));

set in_strlen:=in_strlen-1;

end while;

RETURN @var;

END $$

delimiter ;



## MySQL function to generate random Enum-ID from specified enum definition

DELIMITER $$

DROP FUNCTION IF EXISTS get_enum $$

CREATE FUNCTION get_enum(col_type varchar(100)) RETURNS VARCHAR(100) DETERMINISTIC

        RETURN if((@var:=ceil(rand()*10)) > (length(col_type)-length(replace(col_type,',',''))+1),(length(col_type)-length(replace(col_type,',',''))+1),@var);

$$

DELIMITER ;



## MySQL function to generate random float value from specified precision and scale.

DELIMITER $$

DROP FUNCTION IF EXISTS get_float $$

CREATE FUNCTION get_float(in_precision int, in_scale int) RETURNS VARCHAR(100) DETERMINISTIC

        RETURN round(rand()*pow(10,(in_precision-in_scale)),in_scale) 

$$

DELIMITER ;




## MySQL function to generate random date (of year 2012).

DELIMITER $$

DROP FUNCTION IF EXISTS get_date $$

CREATE FUNCTION get_date() RETURNS VARCHAR(10) DETERMINISTIC

        RETURN DATE(FROM_UNIXTIME(RAND() * (1356892200 - 1325356200) + 1325356200))

#       Below will generate random data for random years

#       RETURN DATE(FROM_UNIXTIME(RAND() * (1577817000 - 946665000) + 1325356200))

$$

DELIMITER ;



## MySQL function to generate random time.

DELIMITER $$

DROP FUNCTION IF EXISTS get_time $$

CREATE FUNCTION get_time() RETURNS INTEGER DETERMINISTIC

        RETURN TIME(FROM_UNIXTIME(RAND() * (1356892200 - 1325356200) + 1325356200))

$$

DELIMITER ;


## MySQL function to generate random int.

DELIMITER $$

DROP FUNCTION IF EXISTS get_int $$

CREATE FUNCTION get_int() RETURNS INTEGER DETERMINISTIC

        RETURN floor(rand()*10000000) 

$$

DELIMITER ;


## MySQL function to generate random tinyint.

DELIMITER $$

DROP FUNCTION IF EXISTS get_tinyint $$

CREATE FUNCTION get_tinyint() RETURNS INTEGER DETERMINISTIC

        RETURN floor(rand()*100) 

$$

DELIMITER ;


## MySQL function to generate random varchar column of specified length(alpha-numeric string).

DELIMITER $$

DROP FUNCTION IF EXISTS get_varchar $$

CREATE FUNCTION get_varchar(in_length varchar(500)) RETURNS VARCHAR(500) DETERMINISTIC

        RETURN SUBSTRING(MD5(RAND()) FROM 1 FOR in_length)

$$

DELIMITER ;


## MySQL function to generate random datetime value (any datetime of year 2012).

DELIMITER $$

DROP FUNCTION IF EXISTS get_datetime $$

CREATE FUNCTION get_datetime() RETURNS VARCHAR(30) DETERMINISTIC

        RETURN FROM_UNIXTIME(ROUND(RAND() * (1356892200 - 1325356200)) + 1325356200)

$$

DELIMITER ;


使用方法:

mysql>use DBNAME


- Parameters are: `database-name`, `table-name`, `number-of-records`, `debug-mode`

- Setting `debug-mode` as `Y` will print all the insert statements that are being executed.


用法举例:


mysql>call populate('test','test',100,'N');

+---------------------+

| Developed by        |

+---------------------+

| Kedar Vaijanapurkar |

+---------------------+

1 row in set (2.38 sec)


Query OK, 0 rows affected (2.38 sec)


存储过程参数说明:

第一个参数应输入数据库名,第二个参数应输入表名,第三个参数应输入要插入的记录数,第四个参数是是否打开调试

模式,打开的话会打印出正在执行的SQL语句,例如:


mysql>use test


mysql> call populate('test','test',100,'Y');

+-----------------------------------------------------------+

| @func_query                                               |

+-----------------------------------------------------------+

| INSERT INTO test.test VALUES (get_int(), get_string(80)); |

+-----------------------------------------------------------+

1 row in set (0.01 sec)


+---------------------+

| Developed by        |

+---------------------+

| Kedar Vaijanapurkar |

+---------------------+

1 row in set (3.51 sec)


Query OK, 0 rows affected (3.51 sec)


该存储过程SQL语句下载地址为:

https://github.com/kedarvj/mysql-random-data-generator/blob/master/populate.sql


阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-数据库
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯