PHP中如何高效安全拆分SQL命令以拆分为多个事务执行

最优方案：使用PostgreSQL原生扩展提供的pg_split_sql函数 #

该函数是PHP pgsql扩展自带的原生工具（PHP 8.1及以上版本默认支持），专门针对PostgreSQL语法设计，自动处理字符串内的分号、转义字符、注释、标识符引号等边界场景，无需自行实现解析逻辑，安全性和性能都有保障，完全适配仅含DDL语句的拆分需求。
示例代码如下：

// 初始化PG连接
$conn = pg_connect('host=localhost dbname=your_db user=your_user password=your_pwd');
// 读取SQL脚本全量内容
$sqlContent = file_get_contents('/path/to/your/ddl_script.sql');
// 直接调用原生函数完成拆分
$sqlStatements = pg_split_sql($sqlContent);

// 按批量大小拆分事务，可根据DDL复杂度自定义批量值
$batchSize = 10;
$batches = array_chunk($sqlStatements, $batchSize);

foreach ($batches as $batch) {
    pg_query($conn, 'BEGIN;');
    try {
        foreach ($batch as $stmt) {
            $trimStmt = trim($stmt);
            // 跳过拆分产生的空语句
            if (empty($trimStmt)) continue;
            pg_query($conn, $trimStmt);
        }
        pg_query($conn, 'COMMIT;');
    } catch (Exception $e) {
        pg_query($conn, 'ROLLBACK;');
        // 按需补充异常处理逻辑
        throw $e;
    }
}

兼容低版本PHP的自定义实现方案 #

如果使用的PHP版本低于8.1，可以自行实现轻量SQL解析逻辑，核心思路是遍历SQL字符，跟踪当前是否处于引号、注释范围内，仅拆分不在任何边界内的分号。
示例代码如下：

function splitPostgreSQLSql(string $sql): array {
    $statements = [];
    $currentStmt = '';
    $len = strlen($sql);
    // 定义边界状态标识
    $inSingleQuote = false;
    $inDoubleQuote = false;
    $inLineComment = false;
    $inBlockComment = false;
    $escaped = false;

    for ($i = 0; $i < $len; $i++) {
        $char = $sql[$i];
        $nextChar = $i < $len - 1 ? $sql[$i + 1] : '';

        // 处理转义字符
        if ($char === '\\' && !$escaped && !$inLineComment && !$inBlockComment) {
            $escaped = true;
            $currentStmt .= $char;
            continue;
        }

        // 处理单行注释 --
        if (!$inSingleQuote && !$inDoubleQuote && !$inBlockComment && $char === '-' && $nextChar === '-') {
            $inLineComment = true;
            $currentStmt .= $char . $nextChar;
            $i++;
            continue;
        }
        if ($inLineComment && $char === "\n") {
            $inLineComment = false;
        }

        // 处理块注释 /* */
        if (!$inSingleQuote && !$inDoubleQuote && !$inLineComment && $char === '/' && $nextChar === '*') {
            $inBlockComment = true;
            $currentStmt .= $char . $nextChar;
            $i++;
            continue;
        }
        if ($inBlockComment && $char === '*' && $nextChar === '/') {
            $inBlockComment = false;
            $currentStmt .= $char . $nextChar;
            $i++;
            continue;
        }

        if ($inLineComment || $inBlockComment) {
            $currentStmt .= $char;
            $escaped = false;
            continue;
        }

        // 处理引号边界
        if (!$escaped) {
            if ($char === "'" && !$inDoubleQuote) {
                $inSingleQuote = !$inSingleQuote;
            } elseif ($char === '"' && !$inSingleQuote) {
                $inDoubleQuote = !$inDoubleQuote;
            }
        }

        // 仅在无边界包裹时拆分分号
        if ($char === ';' && !$inSingleQuote && !$inDoubleQuote) {
            $currentStmt .= $char;
            $trimStmt = trim($currentStmt);
            if (!empty($trimStmt)) $statements[] = $trimStmt;
            $currentStmt = '';
        } else {
            $currentStmt .= $char;
        }

        $escaped = false;
    }

    // 处理末尾未加分号的语句
    $trimStmt = trim($currentStmt);
    if (!empty($trimStmt)) $statements[] = $trimStmt;
    return $statements;
}

拆分后的语句可以复用上面的批量事务执行逻辑完成操作。

额外注意事项 #

• 如果SQL涉及$$包裹的存储过程、自定义函数，可以在自定义实现中新增美元符引用的状态跟踪即可适配
• DDL批量执行时建议控制事务大小，避免长事务锁表影响线上业务
• 生产环境执行前可以先在测试环境验证拆分结果的准确性

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