PHP PDO 查询结果集添加常量值的两种实用方法

PHP PDO 查询：如何在结果集中塞入自定义常量值？

咱们在用 PHP 的 PDO 跟数据库打交道时，有时候会遇到这么个情况：从数据库查出来一批数据，想给每一行结果都加个固定的标记或者额外的字段，但这字段在数据库表里本来是没有的。就像下面这位朋友遇到的：

// 假设 $books_ids 是一个包含书籍 ID 的数组
if( !empty( $books_ids ) )
{
    // 准备 IN 查询的占位符
    $books_ids_in = implode(',', array_fill(0, count($books_ids), '?'));

    $query = "SELECT
        b.id,
        b.`name`,
        b.`year`,
        GROUP_CONCAT(DISTINCT a.`name`) AS author_names,
        GROUP_CONCAT(DISTINCT s.`name`) AS store_names
    FROM
        books AS b
        LEFT JOIN books_authors AS b_a ON b.id = b_a.book_id
        LEFT JOIN authors AS a ON a.id = b_a.author_id
        LEFT JOIN books_stores AS b_s ON b.id = b_s.book_id
        LEFT JOIN stores AS s ON s.id = b_s.store_id
    WHERE
        b.id IN (". $books_ids_in .") // IN 条件使用了占位符
    GROUP BY b.id
    ORDER BY b.id";

    $stmt = $conn->prepare($query);

    // 动态绑定 IN 条件的值
    foreach ($books_ids as $k => $id) {
        // 注意：占位符索引从 1 开始
        $stmt->bindValue(($k+1), $id);
    }

    $stmt->execute();
    // 获取所有结果行
    $results = $stmt->fetchAll(PDO::FETCH_ASSOC); // 建议使用 FETCH_ASSOC 获取关联数组

    // 现在 $results 是包含查询结果的数组
    // 但提问者想给 $results 里的每一行都加上一个类似 'param' => 'book' 的键值对
}

上面这段代码通过 PDO 查询了指定 ID 的书籍信息，还关联了作者和书店。现在的问题是，想在 $results 数组的每个元素（也就是每一行查询结果）里，都塞进去一个 param = "book" 这样的固定信息，用来标识这些数据是关于“书籍”的。这该怎么搞定呢？

为啥会有这需求？

你可能会想，数据库里本来没这玩意儿，为啥要硬加进去呢？这其实挺常见的，原因可能五花八门：

1. 数据来源区分： 可能你的应用需要同时处理多种类型的数据（比如书籍、文章、用户），查询后需要合并处理。在结果里加个标识，就能方便地区分“这批数据是书籍”、“那批数据是文章”。
2. 前端逻辑简化： 有时候，后端直接给前端一个规整、包含所有必要信息的数据结构，能省去前端不少判断和组装的功夫。加个常量字段可能就是为了适配前端组件的需要。
3. 统一数据格式： 在一些数据处理流程中，要求进入下一个环节的数据都必须包含某个字段，即使这个字段的值对于当前数据源是固定的，也得加上。
4. 临时标记： 可能在某个特定业务场景下，需要对一批查询结果打上临时的标记，用于后续的特殊处理。

知道了原因，咱们就来看看怎么解决这个问题。主要有两种路子：一种是在 SQL 查询层面直接动手脚，另一种是在 PHP 代码里对查询结果进行“后期加工”。

解决方案

方案一：直接在 SQL 里加戏

这是最直接的方法，在你的 SELECT 语句里，直接把这个常量加进去，就像你 SELECT 其他数据库字段一样。

原理和作用

数据库在执行 SELECT 语句时，不仅可以选取表里的字段，也可以直接选取一个你指定的常量值（字符串、数字等）。你给这个常量起个别名（使用 AS），它就会成为结果集中的一个新列，并且每一行的这个列都会是那个固定的值。

代码示例

修改上面例子中的 $query 字符串，在 SELECT 部分添加 param 字段：

if( !empty( $books_ids ) )
{
    $books_ids_in = implode(',', array_fill(0, count($books_ids), '?'));

    // 注意看 SELECT 子句的第一行
    $query = "SELECT
        'book' AS param, // <--- 在这里直接添加常量字符串，并给它取个别名 param
        b.id,
        b.`name`,
        b.`year`,
        GROUP_CONCAT(DISTINCT a.`name`) AS author_names,
        GROUP_CONCAT(DISTINCT s.`name`) AS store_names
    FROM
        books AS b
        LEFT JOIN books_authors AS b_a ON b.id = b_a.book_id
        LEFT JOIN authors AS a ON a.id = b_a.author_id
        LEFT JOIN books_stores AS b_s ON b.id = b_s.book_id
        LEFT JOIN stores AS s ON s.id = b_s.store_id
    WHERE
        b.id IN (". $books_ids_in .")
    GROUP BY b.id
    ORDER BY b.id";

    $stmt = $conn->prepare($query);
    foreach ($books_ids as $k => $id) {
        $stmt->bindValue(($k+1), $id);
    }

    $stmt->execute();
    // 现在 fetchAll 的结果里，每一行都会自动包含 'param' => 'book'
    $results = $stmt->fetchAll(PDO::FETCH_ASSOC);

    // 比如 $results[0] 可能看起来像这样：
    // [
    //   'param' => 'book',
    //   'id' => 123,
    //   'name' => 'PHP 核心技术与最佳实践',
    //   'year' => 2023,
    //   'author_names' => '张三,李四',
    //   'store_names' => '京东书城,当当网'
    // ]
}

就这样，执行查询后，fetchAll() 返回的 $results 数组里的每个元素，都会自动包含 'param' => 'book' 这个键值对了，是不是挺省事？

安全建议

这种方式通常很安全。因为你添加的是一个固定的字符串常量，不是用户输入，所以不存在 SQL 注入的风险（只要你的其他部分是安全的，比如这里用了预处理语句绑定 books_ids）。

从性能角度看，数据库处理一个常量列的开销非常小，几乎可以忽略不计。相比在 PHP 里循环处理，如果只是加个简单的固定值，这种方式通常效率更高，因为数据库在生成结果集的时候顺手就加上了。

进阶使用技巧

如果你想让这个“常量”稍微灵活一点，比如根据查询到的其他字段的值来决定这个附加列的值，你可以在 SQL 里使用 CASE 语句。

例如，假设你想根据书籍出版年份（b.year）来决定 param 的值，比如 2000 年后的书标记为 'new_book'，之前的标记为 'old_book'：

SELECT
    -- 使用 CASE 语句动态决定 param 的值
    CASE
        WHEN b.`year` >= 2000 THEN 'new_book'
        ELSE 'old_book'
    END AS param,
    b.id,
    b.`name`,
    b.`year`,
    -- ... 其他字段 ...
FROM
    -- ... 表连接和 WHERE 条件 ...

这样，虽然不是严格意义上的“常量”了，但它仍然是在 SQL 查询层面完成的列值添加，原理是相通的。

方案二：PHP 后期加工

如果你不想修改 SQL 语句（比如 SQL 是动态生成的，或者你觉得 SQL 就该只负责取数据，不掺杂这种“业务逻辑”），那就在 PHP 代码里处理 fetchAll 拿到的结果。

原理和作用

这个思路就是，先把数据库原始数据请回来（不包含你想加的那个 param 字段），拿到 PHP 数组 $results 之后，再写一段 PHP 代码，遍历这个数组，给里面的每个子数组（代表每一行数据）添加上新的键值对。

代码示例

接续原始代码的 fetchAll 之后：

// ... 前面的 PDO 查询和 fetchAll 部分保持不变 ...
$results = $stmt->fetchAll(PDO::FETCH_ASSOC);

// 现在 $results 是原始的查询结果数组，没有 'param' 字段

// 如果 $results 不为空，就开始加工
if (!empty($results)) {
    $param_to_add = 'book'; // 定义你要添加的常量值

    // 遍历 $results 数组，注意使用引用(&)，这样可以直接修改原数组元素
    foreach ($results as $key => &$row) {
        $row['param'] = $param_to_add; // 给当前行（子数组）添加新的键值对
        // 或者 $results[$key]['param'] = $param_to_add; 效果一样
    }
    // 记着 unset 引用，好习惯
    unset($row);
}

// 现在 $results 数组里的每一行也都包含了 'param' => 'book'
// 处理后的效果和方案一相同

使用 array_map 的方式

如果你喜欢函数式编程风格，也可以用 array_map 来实现同样的效果，代码看起来更紧凑：

// ... 前面的 PDO 查询和 fetchAll 部分保持不变 ...
$results = $stmt->fetchAll(PDO::FETCH_ASSOC);

if (!empty($results)) {
    $param_to_add = 'book';

    // 使用 array_map 对 $results 数组的每个元素执行一个回调函数
    $results = array_map(function($row) use ($param_to_add) {
        $row['param'] = $param_to_add; // 在回调函数里添加新键值对
        return $row; // 返回修改后的行
    }, $results);
}

// $results 现在也是加工后的结果了

array_map 会返回一个新数组，你需要将它重新赋给 $results。注意 use ($param_to_add) 是为了将外部变量引入匿名函数的作用域。

安全建议

这种方式在 PHP 层面操作，本身没有安全风险，因为它处理的是已经从数据库安全获取到的数据。

性能方面，如果结果集 $results 非常非常大（比如几十万、上百万行），在 PHP 里循环或使用 array_map 会消耗一些 CPU 时间和内存。相比方案一（SQL直接添加），这点开销可能会更明显。但对于绝大多数 Web 应用场景（几百、几千行数据），这点性能差异几乎可以忽略不计。

进阶使用技巧

• 按需添加： 在 PHP 里处理的好处是灵活性高。你可以根据更复杂的、不容易在 SQL 里表达的 PHP 逻辑来决定是否添加、添加什么值。
• 与其他处理结合： 你可能本来就需要遍历结果集做其他处理（比如格式化日期、计算衍生值等），那么顺手把这个常量字段加上，就不会增加额外的遍历开销。
• 性能考虑： 如果你的数据库服务器压力很大，而应用服务器（跑 PHP 的地方）相对空闲，那么把这种简单的加工任务挪到 PHP 来做，可以为数据库减负。反之亦然。

怎么选？

两种方法都能达到目的，选哪个呢？

• 如果你想加的就是个简单的固定值，或者能用 SQL 的 CASE 轻松表达的逻辑 ，并且你不介意稍微“污染”一点你的 SQL 语句，那么方案一（SQL 直接添加）通常更简洁、效率可能略高 。它让数据库干了该干的活儿，PHP 代码更干净。
• 如果你的 SQL 语句比较复杂，或者是由某个框架、ORM 生成的，你不想去动它 ；或者你需要根据非常复杂的 PHP 逻辑来决定添加什么值 ；又或者你反正也要在 PHP 里遍历结果做其他处理 ，那么方案二（PHP 后期加工）是更好的选择 。它保持了 SQL 的纯粹性，把表现层/业务逻辑相关的修饰放在了 PHP 里。

总的来说，对于“添加一个固定常量值”这种简单需求，在 SQL 里直接 SELECT '常量' AS 别名 是非常常用且推荐的做法。但了解 PHP 后处理的方法也很有用，因为它提供了更大的灵活性来应对更复杂的场景。