如何根据查询获取的QUANTITY列值重复插入数据库记录？

嘿，这个需求其实很好实现，核心就是把你从DB2查出来的每条记录，根据它的QUANTITY值重复插入对应次数到MySQL的placements_new表就行。结合你已经有可用的查询和插入逻辑，咱们一步步来：

你已经搞定了DB2记录查询和MySQL单条插入的核心逻辑，现在只需要在两者之间加一层循环控制，根据每条记录的QUANTITY值决定插入次数。下面分两种常见场景给你具体示例：

1. 基础循环插入（适合小数据量） #

如果你的数据量不大，直接遍历DB2的结果集，对每条记录循环QUANTITY次调用插入方法就可以。这里举两个主流语言的例子：

Java示例 #

假设你已经把DB2查询结果封装成了Db2Record实体类，并且有现成的MySQL插入方法：

// 调用你已有的方法获取DB2记录列表
List<Db2Record> db2RecordList = queryDb2Records();

// 遍历每条DB2记录
for (Db2Record record : db2RecordList) {
    int insertCount = record.getQuantity(); // 获取QUANTITY值
    // 先判断数量是否有效，避免无效循环
    if (insertCount <= 0) {
        continue;
    }
    // 取出要插入的字段值（根据你的表结构调整）
    String productNo = record.getProductNo();
    String locationCode = record.getLocationCode();

    // 循环插入对应次数
    for (int i = 0; i < insertCount; i++) {
        // 调用你已有的MySQL插入方法
        insertIntoPlacementsNew(productNo, locationCode);
    }
}

Python示例 #

假设你从DB2拿到的是字典格式的记录列表：

# 调用你已有的查询方法获取DB2数据
db2_records = get_db2_data()

for record in db2_records:
    insert_times = record["QUANTITY"]
    if insert_times <= 0:
        continue
    # 提取要插入的字段
    sku = record["sku"]
    warehouse = record["warehouse"]

    # 循环插入
    for _ in range(insert_times):
        # 调用你已有的插入函数
        insert_to_mysql(sku, warehouse)

2. 批量插入优化（适合大数据量） #

如果QUANTITY的值很大，或者DB2返回的记录很多，单条循环插入会很慢，这时候用批量插入能大幅提升性能。

Java批量插入示例 #

用JDBC的addBatch()和executeBatch()来批量提交：

String insertSql = "INSERT INTO placements_new (product_no, location_code) VALUES (?, ?)";

try (Connection mysqlConn = getMysqlConnection();
     PreparedStatement pstmt = mysqlConn.prepareStatement(insertSql)) {

    List<Db2Record> db2Records = queryDb2Records();
    int batchSize = 1000; // 每1000条提交一次，可根据内存调整
    int count = 0;

    for (Db2Record record : db2Records) {
        int insertCount = record.getQuantity();
        if (insertCount <= 0) continue;

        String productNo = record.getProductNo();
        String locationCode = record.getLocationCode();

        for (int i = 0; i < insertCount; i++) {
            pstmt.setString(1, productNo);
            pstmt.setString(2, locationCode);
            pstmt.addBatch();
            count++;

            // 达到批量阈值就提交
            if (count % batchSize == 0) {
                pstmt.executeBatch();
                pstmt.clearBatch();
            }
        }
    }
    // 提交剩余的批量数据
    pstmt.executeBatch();
    mysqlConn.commit();
} catch (SQLException e) {
    e.printStackTrace();
    // 出错时回滚事务
    if (mysqlConn != null) {
        mysqlConn.rollback();
    }
}

Python批量插入示例 #

用executemany()一次性提交所有插入数据：

insert_sql = "INSERT INTO placements_new (sku, warehouse) VALUES (%s, %s)"
insert_data = []

# 先收集所有要插入的数据
db2_records = get_db2_data()
for record in db2_records:
    insert_times = record["QUANTITY"]
    if insert_times <= 0:
        continue
    sku = record["sku"]
    warehouse = record["warehouse"]
    # 把当前记录重复insert_times次加入列表
    insert_data.extend([(sku, warehouse)] * insert_times)

# 批量插入到MySQL
with get_mysql_connection() as conn:
    cursor = conn.cursor()
    cursor.executemany(insert_sql, insert_data)
    conn.commit()

注意事项 #

• 数据有效性校验：一定要判断QUANTITY是否大于0，避免插入空记录或者无效循环
• 事务控制：插入过程中如果出现异常，要回滚事务，保证数据一致性
• 性能调优：批量插入的阈值（比如示例中的1000）可以根据你的服务器内存和数据库性能调整
• 类型匹配：确保DB2查询的字段类型和MySQL表的列类型一致，比如数字类型、字符串长度等，避免插入报错

内容的提问来源于stack exchange，提问作者Geoff_S