数据库|循环查询 DB2实现循环遍历数据的方法

数据库 | 循环查询：DB2实现循环遍历数据的方法

场景引入

假设你是一名DB2数据库开发人员,最近接到一个需求：需要批量处理一张包含数万条订单记录的表格，并对每条数据进行特定的业务逻辑计算，手动逐条操作显然不现实，这时候你就需要一种高效的方式来循环遍历数据。

在DB2中,循环遍历数据可以通过多种方式实现，比如使用游标（CURSOR）结合存储过程（Stored Procedure），或者利用递归SQL等方法，下面我们就来详细介绍几种常见的实现方式。

使用游标（CURSOR）循环遍历

游标是DB2中处理结果集的常用方式,特别适合逐行处理数据。

示例代码

-- 创建存储过程，使用游标循环处理数据
CREATE OR REPLACE PROCEDURE PROCESS_ORDERS()
LANGUAGE SQL
BEGIN
    -- 声明变量
    DECLARE v_order_id INT;
    DECLARE v_customer_name VARCHAR(100);
    DECLARE v_total_amount DECIMAL(10, 2);
    DECLARE v_done INT DEFAULT 0;
    -- 声明游标
    DECLARE order_cursor CURSOR FOR 
        SELECT order_id, customer_name, amount 
        FROM orders 
        WHERE status = 'PENDING';
    -- 声明异常处理器
    DECLARE CONTINUE HANDLER FOR NOT FOUND SET v_done = 1;
    -- 打开游标
    OPEN order_cursor;
    -- 循环处理数据
    read_loop: LOOP
        -- 获取当前行数据
        FETCH order_cursor INTO v_order_id, v_customer_name, v_total_amount;
        -- 如果游标读取完毕，退出循环
        IF v_done = 1 THEN
            LEAVE read_loop;
        END IF;
        -- 在这里编写业务逻辑，比如更新订单状态
        UPDATE orders 
        SET status = 'PROCESSED', 
            processed_time = CURRENT_TIMESTAMP 
        WHERE order_id = v_order_id;
        -- 可以打印日志（DB2的存储过程通常使用 CALL DBMS_OUTPUT.PUT_LINE）
        -- CALL DBMS_OUTPUT.PUT_LINE('Processed Order: ' || v_order_id);
    END LOOP;
    -- 关闭游标
    CLOSE order_cursor;
END;

适用场景

• 需要对数据进行逐行处理,并可能修改数据。
• 适用于复杂的业务逻辑,如条件判断、多表关联更新等。

优缺点

✅ 优点：灵活，适合复杂逻辑。
❌ 缺点：性能较低，大数据量时可能较慢。

使用递归SQL（WITH RECURSIVE）

如果不想用存储过程,DB2支持递归查询（WITH RECURSIVE），可以模拟循环遍历数据。

示例代码

-- 假设我们要处理所有未处理的订单，并更新状态
WITH RECURSIVE order_processor AS (
    -- 初始查询：获取第一条待处理订单
    SELECT order_id, customer_name, amount, ROW_NUMBER() OVER (ORDER BY order_id) AS rn
    FROM orders
    WHERE status = 'PENDING'
    ORDER BY order_id
    FETCH FIRST 1 ROW ONLY
    UNION ALL
    -- 递归部分：获取下一条订单
    SELECT o.order_id, o.customer_name, o.amount, op.rn + 1
    FROM orders o
    JOIN order_processor op ON o.order_id > op.order_id
    WHERE o.status = 'PENDING'
    ORDER BY o.order_id
    FETCH FIRST 1 ROW ONLY
)
-- 最终操作：更新所有匹配的记录
UPDATE orders
SET status = 'PROCESSED',
    processed_time = CURRENT_TIMESTAMP
WHERE order_id IN (SELECT order_id FROM order_processor);

适用场景

• 需要基于某些条件逐步处理数据,但不希望使用存储过程。
• 适用于数据量适中的情况。

优缺点

✅ 优点：纯SQL实现，无需存储过程。
❌ 缺点：递归查询可能性能较差，大数据量时需谨慎。

使用批量更新（更适合简单操作）

如果只是简单的更新操作,可以不用循环，直接用一条SQL完成：

-- 直接批量更新所有符合条件的记录
UPDATE orders
SET status = 'PROCESSED',
    processed_time = CURRENT_TIMESTAMP
WHERE status = 'PENDING';

适用场景

• 不需要逐行处理,只需批量更新或删除数据。

优缺点

✅ 优点：性能最高，单条SQL完成操作。
❌ 缺点：无法处理复杂逻辑，如逐行计算。

如果你的需求涉及复杂逻辑（如逐行计算、条件分支），推荐使用游标+存储过程，如果只是简单更新，直接用批量SQL更高效。

希望这篇文章能帮你在DB2中高效实现数据循环处理！