在MySQL存储过程中如何将查询语句赋值给变量实现复用

根据你SELECT id FROM audit的返回结果行数，可选择对应实现方案：

场景1：查询固定返回单条id（适配你现有示例的使用逻辑） #

直接用局部变量存储查询到的单值即可，实现代码如下：

CREATE PROCEDURE my_proc() 
BEGIN
    -- 声明局部变量存储id值，类型和audit表的id字段保持一致即可
    DECLARE audit_id INT DEFAULT NULL;
    -- 将查询结果赋值给变量，加LIMIT 1避免返回多条时报错
    SELECT id INTO audit_id FROM audit LIMIT 1;

    -- 后续所有场景直接复用变量即可
    UPDATE person SET status='Active' WHERE id = audit_id;
    SELECT COUNT(*) FROM audit WHERE id = audit_id;
END

注意：该方案仅适用于查询结果固定为单条的场景，返回多条时会触发报错。

场景2：查询返回多条id，需要复用整套查询逻辑 #

有两种常用实现方式：

方式A：用视图封装查询逻辑（性能最优，最推荐） #

提前创建视图固化查询逻辑，存储过程内直接调用视图即可：

-- 提前创建视图，一次创建永久复用
CREATE VIEW v_audit_id AS SELECT id FROM audit;

-- 存储过程内直接使用视图
CREATE PROCEDURE my_proc() 
BEGIN
    UPDATE person SET status='Active' WHERE id IN (SELECT id FROM v_audit_id);
    SELECT COUNT(*) FROM v_audit_id;
    -- 其他场景直接查询v_audit_id即可
END

方式B：用动态SQL拼接查询语句（和你期望的写法最接近） #

把查询字符串存入变量，后续拼接成完整SQL执行即可：

CREATE PROCEDURE my_proc() 
BEGIN
    -- 把需要复用的查询语句存入用户变量
    SET @audit_query = 'SELECT id FROM audit';

    -- 执行UPDATE逻辑
    SET @update_sql = CONCAT('UPDATE person SET status=''Active'' WHERE id IN (', @audit_query, ')');
    PREPARE stmt FROM @update_sql;
    EXECUTE stmt;
    DEALLOCATE PREPARE stmt;

    -- 执行COUNT查询
    SET @count_sql = CONCAT('SELECT COUNT(*) FROM (', @audit_query, ') AS t');
    PREPARE stmt FROM @count_sql;
    EXECUTE stmt;
    DEALLOCATE PREPARE stmt;

    -- 其他场景只要拼接@audit_query变量即可
END

注意：动态SQL里的单引号需要写两个完成转义，避免语法报错。

内容的提问来源于stack exchange，提问作者average.joe