使用PHP PDO导出PostgreSQL表CSV文件时首行数据丢失求助

嘿，我来帮你排查下CSV导出首行丢失的问题！这种情况我自己也踩过类似的坑，大概率和数据读取逻辑或者表头输出的时机有关，结合你给出的代码片段，我整理了几个常见原因和修复方案：

1. 忘记输出表头（字段名） #

很多时候所谓的“首行丢失”，其实是没把表的字段名作为CSV的第一行输出，直接开始输出数据行。你可以先获取结果集的列名，优先输出表头行：

function bb() {
    $servername = "localhost";
    $username = "postgres";
    $password = "mypassword";
    $dbname = "mydb";
    
    try {
        $conn = new PDO("pgsql:host=$servername;dbname=$dbname", $username, $password);
        $conn->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
        
        // 替换成你的实际查询语句
        $stmt = $conn->prepare("SELECT * FROM your_target_table");
        $stmt->execute();
        
        // 获取所有字段名作为CSV表头
        $columns = [];
        for ($i = 0; $i < $stmt->columnCount(); $i++) {
            $colMeta = $stmt->getColumnMeta($i);
            $columns[] = $colMeta['name'];
        }
        
        // 输出表头（处理引号转义，避免CSV格式错误）
        echo implode(',', array_map(function($item) {
            return '"' . str_replace('"', '""', $item) . '"';
        }, $columns)) . "\n";
        
        // 循环输出数据行
        while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
            echo implode(',', array_map(function($item) {
                return '"' . str_replace('"', '""', $item) . '"';
            }, $row)) . "\n";
        }
        
    } catch(PDOException $e) {
        echo "Error: " . $e->getMessage();
    }
    $conn = null;
}

2. 提前执行了一次fetch()导致首行被消耗 #

如果你的代码里在循环输出之前，不小心额外调用了一次$stmt->fetch()，那第一行数据就会被读取但没输出，直接从第二行开始循环：

// 错误示例：这个多余的fetch会吃掉首行数据
$row = $stmt->fetch(); 
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
    // 从第二行开始输出
}

修复方法很简单：删掉循环前多余的fetch()调用，确保循环从结果集的第一行开始读取。

3. 用PostgreSQL原生COPY命令（更高效的替代方案） #

如果数据量较大，推荐直接用PostgreSQL的COPY命令导出CSV，既避免PHP处理的性能瓶颈，也能完美保证表头和数据行的完整性：

function bb() {
    $servername = "localhost";
    $username = "postgres";
    $password = "mypassword";
    $dbname = "mydb";
    
    try {
        $conn = new PDO("pgsql:host=$servername;dbname=$dbname", $username, $password);
        $conn->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
        
        // 使用COPY命令直接导出，自动包含表头
        $stmt = $conn->prepare("COPY (SELECT * FROM your_target_table) TO STDOUT WITH (FORMAT CSV, HEADER)");
        $stmt->execute();
        
        // 直接输出COPY的结果到标准输出
        $stmt->fetchAll(PDO::FETCH_ASSOC);
        
    } catch(PDOException $e) {
        echo "Error: " . $e->getMessage();
    }
    $conn = null;
}

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