PHP PDOStatement：获取一行，作为数组的键的第一列[重复]

Question

我正在使用PDOStatement 来查询数据库。每当我得到一个返回的行时，我希望它被提取到一个数组中，$row[0] 作为键，行中的后续元素作为值。

当然，我可以编写foreach 循环和if 条件的组合来完成这项工作，如下所示：

private static function GetMySQLResult($dbname, $sqlString) {
    
    $dbh = self::ConstructPDOObject($dbname);
    $dbh->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
    
    $result=array();
    foreach ($dbh->query($sqlString) as $row) {
        // the simplest case for 2 columns, 
        // should add more to handle more columns
        $result[$row[0]][]=$row[1];
    }
    
    return $result;   
}

但我正在寻找现有的方法；有没有这样的方法已经存在了？

为什么要重新提出问题。

这里问的显然是 PDO::FETCH_GROUP|PDO::FETCH_ASSOC 的组合。 PDO::FETCH_KEY_PAIR 仅适用于 2 列结果。但这里的示例是一个 2 列结果，仅用于简化问题中的代码，并非适用于所有情况。

Answer 1

学分归 devdRew。 Check the other question here.

显然，您可以按照答案中的说明进行操作。我也检查过，效果很好。

$q = $db->query("SELECT `name` AS name, `value` AS value FROM `settings`;");
$r  = $q->fetchAll(PDO::FETCH_KEY_PAIR);

编辑

此答案要求您指定最多 2 列：1 个键和 1 个值。如果您需要从数据库中检索更多密钥，请查看下面的答案并阅读 @nullabilty 的评论。对于那些懒惰的人，这是他的方法：

$q->fetchAll(PDO::FETCH_UNIQUE);

Answer 2

$stmt->fetchAll(PDO::FETCH_UNIQUE); 

虽然有点小技巧，但实际上是唯一的方法，因为有人 decided FETCH_KEY_PAIR 应该只需要 2 列结果集。

注意：第一列是key，不会以任何其他形式在结果集中返回。

Answer 3

Afaik，在 PHP 中没有现有的方法可以做到这一点，但有几种方法可以实现你想要的。

第一个是 Xeoncross 所说但稍作修改的内容：

$pdostmt = $pdo->query("SELECT ... FROM your_table");

$res = array();

foreach ($pdostmt->fetchAll(PDO::FETCH_ASSOC) as $row)
{
    $res[array_shift($row)] = $row;
}

否则，您可以使用__set() 方法创建一个类来捕获变量赋值：

class at_res
{
  public $key;
  public $values = array();
  protected $flag = true;

  public function __set($var, $value)
  {
    if ($this->flag)
    {
      $this->key = $value;
      $this->flag = false;
    }
    else
    {
      $this->values[$var] = $value;
    }
  }

  public function extract()
  {
    return array($this->key => $this->values);
  }
}


$pdo = new PDO(...);

$pdostmt = $pdo->query("SELECT ... FROM your_table");

$res = $pdostmt->fetchObject('at_res');

var_dump($res->extract());

希望对你有帮助。

Answer 4

一些提示，您需要将正确的 fetch 样式传递给 PDOStatement->fetch() 方法，这样您就不会得到双重数据（数字和文本列名）。就像 $row[0] 和 $row['id'] 当你使用 PDO::FETCH_BOTH 时它们都包含相同的值。

$result = $dbh->query($sqlString);
while ($row = $result->fetch(PDO::FETCH_NUM)) {
...

至于您的问题，您必须获取所有结果，然后创建一个数组，其中 $row['id'] 作为键，结果行作为值 - 就像您正在做的那样。我围绕 PDO 构建了一个完整的 ORM 库，但我永远找不到任何东西可以自动执行此操作。

$result = $dbh->query($sqlString);

$results = array();
while ($row = $result->fetch(PDO::FETCH_NUM)) {
    $results[$row[0]] = $row;
}

return $results;

Answer 5

除了两列表场景之外，在 PDO 级别没有什么可以处理这个问题，但是您可以为它编写一个可重用的迭代器，如下所示：

class FirstColumnKeyIterator implements Iterator
{
    private $stmt;
    private $key;
    private $data;
    private $mode;

    public function __construct(PDOStatement $stmt, $fetch_mode = PDO::FETCH_NUM)
    {
        $this->stmt = $stmt;
        $this->mode = $fetch_mode;
        $this->fetch();
    }

    private function fetch()
    {
        if (false !== ($this->data = $this->stmt->fetch($this->mode))) {
            $this->key = current(array_splice($this->data, 0, 1));
        }
    }

    public function rewind()
    {
        // nil operation
    }

    public function key()
    {
        return $this->key;
    }

    public function current()
    {
        return $this->data;
    }

    public function next()
    {
        $this->fetch();
    }

    public function valid()
    {
        return false !== $this->data;
    }
}

构造函数采用PDOStatement 和可选的获取模式（默认为数字列，但可以将关联数组更改为PDO::FETCH_ASSOC）并允许您使用典型的foreach 迭代查询结果。

使用示例：

$dbh = new PDO(/* etc */);

$stmt = $dbh->query('SELECT * FROM accounts');
foreach (new FirstColumnKeyIterator($stmt, PDO::FETCH_ASSOC) as $key => $value) {
        echo $key, ' = ', print_r($value, true), PHP_EOL;
}