PolarDB-X最佳实践系列（五）：使用通义千问和存储过程快速生成测试数据-阿里云开发者社区

PolarDB-X最佳实践系列（五）：使用通义千问和存储过程快速生成测试数据

2024-03-13 411

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 我们在测试数据库性能的过程中，通常需要生成一批测试数据。以前，一般要写一段程序或者脚本来完成这项工作，但现在是2024年啦！时代变了！

文章来源：PolarDB知乎号

作者：梦实

我们在测试数据库性能的过程中，通常需要生成一批测试数据。

以前，一般要写一段程序或者脚本来完成这项工作，但现在是2024年啦！时代变了！

PolarDB-X目前是少有的支持存储过程的分布式数据库，结合大模型，我们可以非常简单的来制造测试数据：存储过程的原理和使用方法通义官网

例如，有这样的表结构：

CREATE TABLE t2 (
        id bigint(20) NOT NULL AUTO_INCREMENT,
        c1 varchar(64) NOT NULL ,
        c2 tinyint(4) NOT NULL ,
        c3 varchar(64) NOT NULL ,
        c4 tinyint(4) NOT NULL ,
        c5 varchar(64) NOT NULL ,
        c6 bigint(20) DEFAULT NULL ,
        c7 bigint(20) DEFAULT NULL ,
        c8 double DEFAULT NULL ,
        c9 bigint(20) DEFAULT NULL ,
        PRIMARY KEY (id),
        CLUSTERED INDEX gsi1 (c3)
                PARTITION BY KEY(c3)
                PARTITIONS 32,
        UNIQUE KEY uk1 (c5, c3, c1),
        KEY idx1 (c1),
        KEY idx2 (c5, c1, c8),
        KEY idx3 (c5, c3, c8)
) ENGINE = InnoDB PARTITION BY KEY(c1) PARTITIONS 32;

我们只需要对通义千问进行提问，描述以下需求，它就能自动生成一个存储过程：

写一个存储过程，为以下表生成100万条数据，其中，c1和c5是40位的十六进制字符串。（SQL内容省略了）

通义千问生成了这样的下面的存储过程：

DELIMITER ;;
CREATE PROCEDURE GenerateTestData()
BEGIN
    DECLARE v_counter INT DEFAULT 0;
    DECLARE v_hex_string CHAR(40);
    WHILE v_counter < 1000000 DO
        -- 生成40位十六进制字符串
        SET v_hex_string = LPAD(LOWER(REPLACE(UUID(), '-', '')), 40, '0');
        INSERT INTO t2 (
            c1,
            c2,
            c3,
            c4,
            c5,
            c6,
            c7,
            c8,
            c9
        ) VALUES (
            -- 生成c1的十六进制字符串
            v_hex_string,
            -- 随机生成tinyint值（例如，范围从0到255）
            FLOOR(RAND() * 256),
            -- 生成c3的随机字符串（也可以是十六进制，这里仅作示例）
            LPAD(LOWER(REPLACE(UUID(), '-', '')), 32, '0'),
            -- 随机生成tinyint值
            FLOOR(RAND() * 256),
            -- 生成c5的十六进制字符串
            v_hex_string,
            -- 随机生成bigint值
            FLOOR(RAND() * POW(2, 63)),
            -- 随机生成bigint值
            FLOOR(RAND() * POW(2, 63)),
            -- 随机生成double值
            RAND(),
            -- 随机生成bigint值
            FLOOR(RAND() * POW(2, 63))
        );
        SET v_counter = v_counter + 1;
    END WHILE;
END;;
DELIMITER ;
CALL GenerateTestData();