MySQL 性能测试：快速生成千万数据的方法

2023-11-28 10:42:25

在 MySQL 性能测试中，生成大量数据是必不可少的。通常情况下，我们需要生成数百万甚至数千万条数据来模拟真实的数据环境。然而，使用传统的代码或 SQL 语句逐条插入数据非常耗时。为了解决这个问题，我们可以使用一些快速生成数据的方法。

1. 使用代码生成数据

我们可以使用编程语言（如 Python 或 Java）编写代码来生成大量数据。这种方法的好处是我们可以灵活地控制数据的内容和格式。例如，我们可以生成具有特定分布的数据，或者生成包含特定字段的数据。

import random
import string

# 生成 1000 万条随机数据
for i in range(10000000):
    # 生成随机字符串作为姓名
    name = ''.join(random.choice(string.ascii_letters) for i in range(10))

    # 生成随机数字作为年龄
    age = random.randint(1, 100)

    # 生成随机字符串作为地址
    address = ''.join(random.choice(string.ascii_letters) for i in range(20))

    # 插入数据到数据库
    cursor.execute("INSERT INTO table_name (name, age, address) VALUES (%s, %s, %s)", (name, age, address))

# 提交事务
conn.commit()

2. 使用 SQL 生成数据

我们也可以使用 SQL 语句来生成大量数据。这种方法的好处是简单易用，而且速度也很快。

-- 生成 1000 万条随机数据
INSERT INTO table_name (name, age, address)
SELECT
    SUBSTR(MD5(RAND()), 1, 10) AS name,
    FLOOR(RAND() * 100) + 1 AS age,
    SUBSTR(MD5(RAND()), 1, 20) AS address
FROM
    information_schema.tables
LIMIT 10000000;