mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-08-27 04:56:02 +02:00
git-svn-id: svn://svn.silverstripe.com/silverstripe/open/modules/sapphire/trunk@56212 467b73ca-7a2a-4603-9d3b-597d59a354a9
114 lines
2.5 KiB
PHP
Executable File
114 lines
2.5 KiB
PHP
Executable File
<?php
|
|
/**
|
|
* This is a class used to represent key->value pairs generated from database queries.
|
|
* The query isn't actually executed until you need it.
|
|
* @package sapphire
|
|
* @subpackage model
|
|
*/
|
|
class SQLMap extends Object implements IteratorAggregate {
|
|
/**
|
|
* The query used to generate the map.
|
|
* @var SQLQuery
|
|
*/
|
|
protected $query;
|
|
|
|
/**
|
|
* Construct a SQLMap.
|
|
* @param SQLQuery $query The query to generate this map. THis isn't executed until it's needed.
|
|
*/
|
|
public function __construct(SQLQuery $query) {
|
|
if(!$query) {
|
|
user_error('SQLMap constructed with null query.', E_USER_ERROR);
|
|
}
|
|
|
|
$this->query = $query;
|
|
}
|
|
|
|
/**
|
|
* Get the name of an item.
|
|
* @param string|int $id The id of the item.
|
|
* @return string
|
|
*/
|
|
public function getItem($id) {
|
|
if($id) {
|
|
$baseTable = reset($this->query->from);
|
|
$this->query->where[] = "$baseTable.ID = $id";
|
|
$record = $this->query->execute()->first();
|
|
if($record) {
|
|
$className = $record['ClassName'];
|
|
$obj = new $className($record);
|
|
return $obj->Title;
|
|
}
|
|
}
|
|
}
|
|
|
|
public function getIterator() {
|
|
$this->genItems();
|
|
return new SQLMap_Iterator($this->items->getIterator());
|
|
}
|
|
|
|
/**
|
|
* Get the items in this class.
|
|
* @return DataObjectSet
|
|
*/
|
|
public function getItems() {
|
|
$this->genItems();
|
|
return $this->items;
|
|
}
|
|
|
|
/**
|
|
* Generate the items in this map. This is used by
|
|
* getItems() if the items have not been generated already.
|
|
*/
|
|
protected function genItems() {
|
|
if(!isset($this->items)) {
|
|
$this->items = new DataObjectSet();
|
|
$items = $this->query->execute();
|
|
|
|
foreach($items as $item) {
|
|
$className = isset($item['RecordClassName']) ? $item['RecordClassName'] : $item['ClassName'];
|
|
|
|
if(!$className) {
|
|
user_error('SQLMap query could not retrieve className', E_USER_ERROR);
|
|
}
|
|
|
|
$this->items->push(new $className($item));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
class SQLMap_Iterator extends Object implements Iterator {
|
|
protected $items;
|
|
|
|
function __construct(Iterator $items) {
|
|
$this->items = $items;
|
|
}
|
|
|
|
|
|
/*
|
|
* Iterator functions - necessary for foreach to work
|
|
*/
|
|
public function rewind() {
|
|
return $this->items->rewind() ? $this->items->rewind()->Title : null;
|
|
}
|
|
|
|
public function current() {
|
|
return $this->items->current()->Title;
|
|
}
|
|
|
|
public function key() {
|
|
return $this->items->current()->ID;
|
|
}
|
|
|
|
public function next() {
|
|
$next = $this->items->next();
|
|
return isset($next->Title) ? $next->Title : null;
|
|
}
|
|
|
|
public function valid() {
|
|
return $this->items->valid();
|
|
}
|
|
}
|
|
|
|
?>
|