wpstudio/majestic
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add assoc method into SqlResultCollection

Browse Source
This commit is contained in:
Alexander Demidov
2014-08-12 14:47:47 +04:00
parent 776859b3c8
commit 9e359e0c25
2 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Model/SqlResultCollection.php
View File

@ -37,4 +37,9 @@ class SqlResultCollection extends \ArrayIterator implements iSqlResultItems
{ {
return $this->offsetGet(0); return $this->offsetGet(0);
} }
public function assoc($field, $assoc_as_array = false) {
$sql_result_provider = new SqlResultProvider($this->items);
return $sql_result_provider->assoc($field, $assoc_as_array);
}
} }

10
Model/SqlResultProvider.php
View File

@ -19,7 +19,7 @@ class SqlResultProvider implements iSqlResultItems
private $result_items_base; private $result_items_base;
/** /**
* @param $result DbStatement * @param $result DbStatement|array
*/ */
public function __construct($result) public function __construct($result)
{ {
@ -29,8 +29,12 @@ class SqlResultProvider implements iSqlResultItems
private function defineResultItems() private function defineResultItems()
{ {
if (is_null($this->result_items_base)) { if (is_null($this->result_items_base)) {
$this->result_items_base = $this->result->fetchAll(); if (is_array($this->result)) {
$this->result_items = $this->result_items_base; $this->result_items_base = $this->result;
} else {
$this->result_items_base = $this->result->fetchAll();
}
$this->result_items = $this->result_items_base;
} }
} }