Newer
Older
<?php
/*********************************************************************
class.orm.php
Simple ORM (Object Relational Mapper) for PHP5 based on Django's ORM,
except that complex filter operations are not supported. The ORM simply
supports ANDed filter operations without any GROUP BY support.
Jared Hancock <jared@osticket.com>
Copyright (c) 2006-2013 osTicket
http://www.osticket.com
Released under the GNU General Public License WITHOUT ANY WARRANTY.
See LICENSE.TXT for details.
vim: expandtab sw=4 ts=4 sts=4:
**********************************************************************/
require_once INCLUDE_DIR . 'class.util.php';
class OrmException extends Exception {}
class OrmConfigurationException extends Exception {}
// Database fields/tables do not match codebase
class InconsistentModelException extends OrmException {
function __construct() {
// Drop the model cache (just incase)
ModelMeta::flushModelCache();
call_user_func_array(array('parent', '__construct'), func_get_args());
}
}
/**
* Meta information about a model including edges (relationships), table
* name, default sorting information, database fields, etc.
*
* This class is constructed and built automatically from the model's
* ::getMeta() method using a class's ::$meta array.
*/
class ModelMeta implements ArrayAccess {
static $base = array(
'pk' => false,
'table' => false,
'defer' => array(),
'select_related' => array(),
'joins' => array(),
'foreign_keys' => array(),
var $meta = array();
var $new;
var $subclasses = array();
function __construct($model) {
// Merge ModelMeta from parent model (if inherited)
$parent = get_parent_class($this->model);
$meta = $model::$meta;
if ($model::$meta instanceof self)
$meta = $meta->meta;
if (is_subclass_of($parent, 'VerySimpleModel')) {
$this->parent = $parent::getMeta();
$meta = $this->parent->extend($this, $meta);
$meta = $meta + self::$base;
// Short circuit the meta-data processing if APCu is available.
// This is preferred as the meta-data is unlikely to change unless
// osTicket is upgraded, (then the upgrader calls the
// flushModelCache method to clear this cache). Also, GIT_VERSION is
// used in the APC key which should be changed if new code is
// deployed.
if (function_exists('apcu_store')) {
$loaded = false;
$apc_key = SECRET_SALT.GIT_VERSION."/orm/{$this->model}";
$this->meta = apcu_fetch($apc_key, $loaded);
if ($loaded)
return;
}
if (!$meta['view']) {
if (!$meta['table'])
throw new OrmConfigurationException(
sprintf(__('%s: Model does not define meta.table'), $this->model));
elseif (!$meta['pk'])
throw new OrmConfigurationException(
sprintf(__('%s: Model does not define meta.pk'), $this->model));
}
// Ensure other supported fields are set and are arrays
foreach (array('pk', 'ordering', 'defer', 'select_related') as $f) {
if (!isset($meta[$f]))
$meta[$f] = array();
elseif (!is_array($meta[$f]))
$meta[$f] = array($meta[$f]);
}
// Break down foreign-key metadata
foreach ($meta['joins'] as $field => &$j) {
if ($j['local'])
$meta['foreign_keys'][$j['local']] = $field;
$this->meta = $meta;
if (function_exists('apcu_store')) {
apcu_store($apc_key, $this->meta, 1800);
/**
* Merge this class's meta-data into the recieved child meta-data.
* When a model extends another model, the meta data for the two models
* is merged to form the child's meta data. Returns the merged, child
* meta-data.
*/
function extend(ModelMeta $child, $meta) {
$this->subclasses[$child->model] = $child;
// Merge 'joins' settings (instead of replacing)
if (isset($this->meta['joins'])) {
$meta['joins'] = array_merge($meta['joins'] ?: array(),
$this->meta['joins']);
}
return $meta + $this->meta + self::$base;
}
function isSuperClassOf($model) {
if (isset($this->subclasses[$model]))
return true;
foreach ($this->subclasses as $M=>$meta)
if ($meta->isSuperClassOf($M))
return true;
}
function isSubclassOf($model) {
if (!isset($this->parent))
return false;
if ($this->parent->model === $model)
return true;
return $this->parent->isSubclassOf($model);
/**
* Adds some more information to a declared relationship. If the
* relationship is a reverse relation, then the information from the
* reverse relation is loaded into the local definition
*
* Compiled-Join-Structure:
* 'constraint' => array(local => array(foreign_field, foreign_class)),
* Constraint used to construct a JOIN in an SQL query
* 'list' => boolean
* TRUE if an InstrumentedList should be employed to fetch a list
* of related items
* 'broker' => Handler for the 'list' property. Usually a subclass of
* 'InstrumentedList'
* 'null' => boolean
* TRUE if relation is nullable
* 'fkey' => array(class, pk)
* Classname and field of the first item in the constraint that
* points to a PK field of a foreign model
* 'local' => string
* The local field corresponding to the 'fkey' property
*/
function processJoin(&$j) {
if (isset($j['reverse'])) {
list($fmodel, $key) = explode('.', $j['reverse']);
// NOTE: It's ok if the forein meta data is not yet inspected.
$info = $fmodel::$meta['joins'][$key];
if (!is_array($info['constraint']))
throw new OrmConfigurationException(sprintf(__(
// `reverse` here is the reverse of an ORM relationship
'%s: Reverse does not specify any constraints'),
$j['reverse']));
foreach ($info['constraint'] as $foreign => $local) {
list($L,$field) = is_array($local) ? $local : explode('.', $local);
$constraint[$field ?: $L] = array($fmodel, $foreign);
}
if (!isset($j['list']))
$j['list'] = true;
if (!isset($j['null']))
// By default, reverse releationships can be empty lists
$j['null'] = true;
else {
foreach ($j['constraint'] as $local => $foreign) {
list($class, $field) = $constraint[$local]
= is_array($foreign) ? $foreign : explode('.', $foreign);
if ($j['list'] && !isset($j['broker'])) {
$j['broker'] = 'InstrumentedList';
}
if ($j['broker'] && !class_exists($j['broker'])) {
throw new OrmException($j['broker'] . ': List broker does not exist');
}
foreach ($constraint as $local => $foreign) {
list($class, $field) = $foreign;
if ($local[0] == "'" || $field[0] == "'" || !class_exists($class))
continue;
function addJoin($name, array $join) {
$this->meta['joins'][$name] = $join;
$this->processJoin($this->meta['joins'][$name]);
/**
* Fetch ModelMeta instance by following a join path from this model
*/
function getByPath($path) {
if (is_string($path))
$path = explode('__', $path);
$root = $this;
foreach ($path as $P) {
if (!($root = $root['joins'][$P]['fkey'][0]))
break;
$root = $root::getMeta();
}
return $root;
}
function offsetGet($field) {
return $this->meta[$field];
}
function offsetSet($field, $what) {
$this->meta[$field] = $what;
}
function offsetExists($field) {
return isset($this->meta[$field]);
}
function offsetUnset($field) {
throw new Exception('Model MetaData is immutable');
}
/**
* Fetch the column names of the table used to persist instances of this
* model in the database.
*/
function getFieldNames() {
if (!isset($this->fields))
$this->fields = self::inspectFields();
return $this->fields;
}
/**
* Create a new instance of the model, optionally hydrating it with the
* given hash table. The constructor is not called, which leaves the
* default constructor free to assume new object status.
*
* Three methods were considered, with runtime for 10000 iterations
* * unserialze('O:9:"ModelBase":0:{}') - 0.0671s
* * new ReflectionClass("ModelBase")->newInstanceWithoutConstructor()
* - 0.0478s
* * and a hybrid by cloning the reflection class instance - 0.0335s
*/
function newInstance($props=false) {
if (!isset($this->new)) {
$rc = new ReflectionClass($this->model);
$this->new = $rc->newInstanceWithoutConstructor();
$instance = clone $this->new;
// Hydrate if props were included
if (is_array($props)) {
$instance->ht = $props;
}
return $instance;
}
function inspectFields() {
if (!isset(self::$model_cache))
self::$model_cache = function_exists('apcu_fetch');
$key = SECRET_SALT.GIT_VERSION."/orm/{$this['table']}";
if ($fields = apcu_fetch($key)) {
return $fields;
}
}
$fields = DbEngine::getCompiler()->inspectTable($this['table']);
static function flushModelCache() {
if (self::$model_cache)
@apcu_clear_cache('user');
class VerySimpleModel {
static $meta = array(
'table' => false,
'ordering' => false,
'pk' => false
);
var $dirty = array();
var $__deferred__ = array();
function __construct($row=false) {
if (is_array($row))
foreach ($row as $field=>$value)
if (!is_array($value))
$this->set($field, $value);
$this->__new__ = true;
/**
* Creates a new instance of the model without calling the constructor.
* If the constructor is required, consider using the PHP `new` keyword.
* The instance returned from this method will not be considered *new*
* and will now result in an INSERT when sent to the database.
*/
static function __hydrate($row=false) {
return static::getMeta()->newInstance($row);
}
function get($field, $default=false) {
if (array_key_exists($field, $this->ht))
return $this->ht[$field];
elseif (($joins = static::getMeta('joins')) && isset($joins[$field])) {
$j = $joins[$field];
// Support instrumented lists and such
if (isset($j['list']) && $j['list']) {
$class = $j['fkey'][0];
$fkey = array();
// Localize the foreign key constraint
foreach ($j['constraint'] as $local=>$foreign) {
list($_klas,$F) = $foreign;
$fkey[$F ?: $_klas] = ($local[0] == "'")
? trim($local, "'") : $this->ht[$local];
}
$v = $this->ht[$field] = new $j['broker'](
// Send Model, [Foriegn-Field => Local-Id]
array($class, $fkey)
);
return $v;
}
// Support relationships
elseif (isset($j['fkey'])) {
$criteria = array();
foreach ($j['constraint'] as $local => $foreign) {
if (class_exists($klas))
$class = $klas;
if ($local[0] == "'") {
$criteria[$F] = trim($local,"'");
}
// Does not affect the local model
continue;
}
else {
$criteria[$F] = $this->ht[$local];
}
}
$v = $this->ht[$field] = $class::lookup($criteria);
}
catch (DoesNotExist $e) {
$v = null;
}
elseif (isset($this->__deferred__[$field])) {
// Fetch deferred field
$row = static::objects()->filter($this->getPk())
// XXX: Seems like all the deferred fields should be fetched
->values_flat($field)
->one();
if ($row)
return $this->ht[$field] = $row[0];
}
elseif ($field == 'pk') {
return $this->getPk();
}
if (isset($default))
return $default;
// For new objects, assume the field is NULLable
if ($this->__new__)
return null;
// Check to see if the column referenced is actually valid
if (in_array($field, static::getMeta()->getFieldNames()))
throw new OrmException(sprintf(__('%s: %s: Field not defined'),
get_class($this), $field));
}
function __get($field) {
return $this->get($field, null);
}
function getByPath($path) {
if (is_string($path))
$path = explode('__', $path);
$root = $this;
foreach ($path as $P)
$root = $root->get($P);
return $root;
}
function __isset($field) {
|| isset(static::$meta['joins'][$field]);
function __unset($field) {
if ($this->__isset($field))
unset($this->ht[$field]);
else
unset($this->{$field});
// Update of foreign-key by assignment to model instance
$joins = static::getMeta('joins');
if (isset($joins[$field])) {
$j = $joins[$field];
if ($j['list'] && ($value instanceof InstrumentedList)) {
// Magic list property
$this->ht[$field] = $value;
return;
}
if (in_array($j['local'], static::$meta['pk'])) {
// Reverse relationship — don't null out local PK
return;
}
// Pass. Set local field to NULL in logic below
}
elseif ($value instanceof VerySimpleModel) {
// Ensure that the model being assigned as a relationship is
// an instance of the foreign model given in the
// relationship, or is a super class thereof. The super
// class case is used primary for the xxxThread classes
// which all extend from the base Thread class.
if (!$value instanceof $j['fkey'][0]
&& !$value::getMeta()->isSuperClassOf($j['fkey'][0])
) {
throw new InvalidArgumentException(
sprintf(__('Expecting NULL or instance of %s. Got a %s instead'),
$j['fkey'][0], is_object($value) ? get_class($value) : gettype($value)));
}
// Capture the object under the object's field name
$this->ht[$field] = $value;
$value = $value->get($j['fkey'][1]);
// Fall through to the standard logic below
}
// Capture the foreign key id value
$field = $j['local'];
}
// elseif $field is in a relationship, adjust the relationship
elseif (isset(static::$meta['foreign_keys'][$field])) {
// meta->foreign_keys->{$field} points to the property of the
// foreign object. For instance 'object_id' points to 'object'
$related = static::$meta['foreign_keys'][$field];
}
$old = isset($this->ht[$field]) ? $this->ht[$field] : null;
if ($old != $value) {
// isset should not be used here, because `null` should not be
// replaced in the dirty array
if (!array_key_exists($field, $this->dirty))
$this->dirty[$field] = $old;
if ($related)
// $related points to a foreign object propery. If setting a
// new object_id value, the relationship to object should be
// cleared and rebuilt
unset($this->ht[$related]);
$this->ht[$field] = $value;
}
function __set($field, $value) {
return $this->set($field, $value);
}
function setAll($props) {
foreach ($props as $field=>$value)
$this->set($field, $value);
}
function serialize() {
return $this->getPk();
}
function unserialize($data) {
$this->ht = $data;
$this->refetch();
}
static function getMeta($key=false) {
if (!static::$meta instanceof ModelMeta
|| get_called_class() != static::$meta->model
) {
static::$meta = new ModelMeta(get_called_class());
}
$M = static::$meta;
return ($key) ? $M->offsetGet($key) : $M;
static function getOrmFields($recurse=false) {
$fks = $lfields = $fields = array();
$myname = get_called_class();
foreach (static::getMeta('joins') as $name=>$j) {
$fks[$j['local']] = true;
if (!$j['reverse'] && !$j['list'] && $recurse) {
foreach ($j['fkey'][0]::getOrmFields($recurse - 1) as $name2=>$f) {
$fields["{$name}__{$name2}"] = "{$name} / $f";
}
}
}
foreach (static::getMeta('fields') as $f) {
if (isset($fks[$f]))
continue;
if (in_array($f, static::getMeta('pk')))
continue;
$lfields[$f] = "{$f}";
}
return $lfields + $fields;
}
/**
* objects
*
* Retrieve a QuerySet for this model class which can be used to fetch
* models from the connected database. Subclasses can override this
* method to apply forced constraints on the QuerySet.
*/
static function objects() {
return new QuerySet(get_called_class());
}
/**
* lookup
*
* Retrieve a record by its primary key. This method may be short
* circuited by model caching if the record has already been loaded by
* the database. In such a case, the database will not be consulted for
* the model's data.
*
* This method can be called with an array of keyword arguments matching
* the PK of the object or the values of the primary key. Both of these
* usages are correct:
*
* >>> User::lookup(1)
* >>> User::lookup(array('id'=>1))
*
* For composite primary keys and the first usage, pass the values in
* the order they are given in the Model's 'pk' declaration in its meta
* data.
*
* Parameters:
* $criteria - (mixed) primary key for the sought model either as
* arguments or key/value array as the function's first argument
*
* Returns:
* (Object<Model>|null) a single instance of the sought model or null if
* no such instance exists.
// Model::lookup(1), where >1< is the pk value
if (!is_array($criteria)) {
$criteria = array();
$pk = static::getMeta('pk');
// Only consult cache for PK lookup, which is assumed if the
// values are passed as args rather than an array
if ($cached = ModelInstanceManager::checkCache(get_called_class(),
$criteria))
return $cached;
try {
return static::objects()->filter($criteria)->one();
}
catch (DoesNotExist $e) {
return null;
}
$ex = DbEngine::delete($this);
try {
$ex->execute();
if ($ex->affected_rows() != 1)
return false;
Signal::send('model.deleted', $this);
}
catch (OrmException $e) {
return false;
}
if ($this->__deleted__)
throw new OrmException('Trying to update a deleted object');
$pk = static::getMeta('pk');
$wasnew = $this->__new__;
// First, if any foreign properties of this object are connected to
// another *new* object, then save those objects first and set the
// local foreign key field values
foreach (static::getMeta('joins') as $prop => $j) {
if (isset($this->ht[$prop])
&& ($foreign = $this->ht[$prop])
&& $foreign instanceof VerySimpleModel
&& !in_array($j['local'], $pk)
&& null === $this->get($j['local'])
) {
if ($foreign->__new__ && !$foreign->save())
return false;
$this->set($j['local'], $foreign->get($j['fkey'][1]));
}
}
// If there's nothing in the model to be saved, then we're done
$ex = DbEngine::save($this);
try {
$ex->execute();
if ($ex->affected_rows() != 1) {
// This doesn't really signify an error. It just means that
// the database believes that the row did not change. For
// inserts though, it's a deal breaker
if ($this->__new__)
return false;
else
// No need to reload the record if requested — the
// database didn't update anything
$refetch = false;
}
catch (OrmException $e) {
return false;
// XXX: Ensure AUTO_INCREMENT is set for the field
$this->ht[$pk[0]] = $ex->insert_id();
Signal::send('model.created', $this);
}
else {
$data = array('dirty' => $this->dirty);
Signal::send('model.updated', $this, $data);
}
# Refetch row from database
if ($refetch) {
// Preserve non database information such as list relationships
// across the refetch
}
if ($wasnew) {
// Attempt to update foreign, unsaved objects with the PK of
// this newly created object
foreach (static::getMeta('joins') as $prop => $j) {
if (isset($this->ht[$prop])
&& ($foreign = $this->ht[$prop])
&& in_array($j['local'], $pk)
) {
if ($foreign instanceof VerySimpleModel
&& null === $foreign->get($j['fkey'][1])
) {
$foreign->set($j['fkey'][1], $this->get($j['local']));
}
elseif ($foreign instanceof InstrumentedList) {
foreach ($foreign as $item) {
if (null === $item->get($j['fkey'][1]))
$item->set($j['fkey'][1], $this->get($j['local']));
}
}
}
}
private function refetch() {
$this->ht =
static::objects()->filter($this->getPk())->values()->one()
+ $this->ht;
}
private function getPk() {
$pk = array();
foreach ($this::getMeta('pk') as $f)
$pk[$f] = $this->ht[$f];
return $pk;
}
/**
* Create a new clone of this model. The primary key will be unset and the
* object will be set as __new__. The __clone() magic method is reserved
* by the buildModel() system, because it clone's a single instance when
* hydrating objects from the database.
*/
function copy() {
// Drop the PK and set as unsaved
$dup = clone $this;
foreach ($dup::getMeta('pk') as $f)
$dup->__unset($f);
$dup->__new__ = true;
return $dup;
}
/**
* AnnotatedModel
*
* Simple wrapper class which allows wrapping and write-protecting of
* annotated fields retrieved from the database. Instances of this class
* will delegate most all of the heavy lifting to the wrapped Model instance.
*/
class AnnotatedModel {
static function wrap(VerySimpleModel $model, $extras=array(), $class=false) {
static $classes;
$class = $class ?: get_class($model);
if ($extras instanceof VerySimpleModel) {
$extra = "Writeable";
}
if (!isset($classes[$class])) {
$classes[$class] = eval(<<<END_CLASS
class {$extra}AnnotatedModel___{$class}
protected \$__overlay__;
use {$extra}AnnotatedModelTrait;
static function __hydrate(\$ht=false, \$annotations=false) {
\$instance = parent::__hydrate(\$ht);
\$instance->__overlay__ = \$annotations;
return \$instance;
return "{$extra}AnnotatedModel___{$class}";
return $classes[$class]::__hydrate($model->ht, $extras);
if (isset($this->__overlay__[$what]))
return $this->__overlay__[$what];
return parent::get($what);
function set($what, $to) {
if (isset($this->__overlay__[$what]))
throw new OrmException('Annotated fields are read-only');
return parent::set($what, $to);
if (isset($this->__overlay__[$what]))
return true;
return parent::__isset($what);
function getDbFields() {
return $this->__overlay__ + parent::getDbFields();
}
}
/**
* Slight variant on the AnnotatedModelTrait, except that the overlay is
* another model. Its fields are preferred over the wrapped model's fields.
* Updates to the overlayed fields are tracked in the overlay model and
* therefore kept separate from the annotated model's fields. ::save() will
* call save on both models. Delete will only delete the overlay model (that
* is, the annotated model will remain).
*/
trait WriteableAnnotatedModelTrait {
if ($this->__overlay__->__isset($what))
return $this->__overlay__->get($what);
return parent::get($what);
}
function set($what, $to) {
if (isset($this->__overlay__)
&& $this->__overlay__->__isset($what)) {
return $this->__overlay__->set($what, $to);
}
return parent::set($what, $to);
}
function __isset($what) {
if (isset($this->__overlay__) && $this->__overlay__->__isset($what))
return true;
return parent::__isset($what);
}
function getDbFields() {
return $this->__overlay__->getDbFields() + parent::getDbFields();
function save($refetch=false) {
$this->__overlay__->save($refetch);
return parent::save($refetch);
}
function delete() {
if ($rv = $this->__overlay__->delete())
// Mark the annotated object as deleted
$this->__deleted__ = true;
return $rv;
$this->func = $name;
$this->args = array_slice(func_get_args(), 1);
}
function input($what, $compiler, $model) {
if ($what instanceof SqlFunction)
$A = $what->toSql($compiler, $model);
elseif ($what instanceof Q)
$A = $compiler->compileQ($what, $model);
else
$A = $compiler->input($what);
return $A;
}
function toSql($compiler, $model=false, $alias=false) {
$args = array();
foreach ($this->args as $A) {
$args[] = $this->input($A, $compiler, $model);
return sprintf('%s(%s)%s', $this->func, implode(', ', $args),
$alias && $this->alias ? ' AS '.$compiler->quote($this->alias) : '');
}
function getAlias() {
return $this->alias;
}
function setAlias($alias) {
$this->alias = $alias;
static function __callStatic($func, $args) {
$I = new static($func);
$I->args = $args;
return $I;
}
function __call($operator, $other) {
array_unshift($other, $this);
return SqlExpression::__callStatic($operator, $other);
}
class SqlCase extends SqlFunction {
var $cases = array();
var $else = false;
static function N() {
return new static('CASE');
}
function when($expr, $result) {
if (is_array($expr))
$expr = new Q($expr);
$this->cases[] = array($expr, $result);
return $this;
}
function otherwise($result) {
$this->else = $result;
return $this;
}
function toSql($compiler, $model=false, $alias=false) {
$cases = array();
foreach ($this->cases as $A) {
list($expr, $result) = $A;
$expr = $this->input($expr, $compiler, $model);
$result = $this->input($result, $compiler, $model);
$cases[] = "WHEN {$expr} THEN {$result}";
}
if ($this->else) {
$else = $this->input($this->else, $compiler, $model);
$cases[] = "ELSE {$else}";
}
return sprintf('CASE %s END%s', implode(' ', $cases),
$alias && $this->alias ? ' AS '.$compiler->quote($this->alias) : '');
}
}
class SqlExpr extends SqlFunction {
function __construct($args) {
$this->args = func_get_args();
if (count($this->args) == 1 && is_array($this->args[0]))
$this->args = $this->args[0];
}
function toSql($compiler, $model=false, $alias=false) {
$O = array();
foreach ($this->args as $field=>$value) {
$ex = $compiler->compileQ($value, $model, false);
$O[] = $ex->text;
}
else {
list($field, $op) = $compiler->getField($field, $model);
if (is_callable($op))
$O[] = call_user_func($op, $field, $value, $model);
else
$O[] = sprintf($op, $field, $compiler->input($value));
}
return implode(' ', $O) . ($alias ? ' AS ' . $compiler->quote($alias) : '');
class SqlExpression extends SqlFunction {
var $operator;
var $operands;
function toSql($compiler, $model=false, $alias=false) {
$O = array();
$O[] = $this->input($operand, $compiler, $model);
return '('.implode(' '.$this->func.' ', $O)
. ($alias ? ' AS '.$compiler->quote($alias) : '')
. ')';
}
static function __callStatic($operator, $operands) {
switch ($operator) {
case 'minus':
$operator = '-'; break;
case 'plus':
$operator = '+'; break;
case 'times':
$operator = '*'; break;
case 'bitand':
$operator = '&'; break;
case 'bitor':
$operator = '|'; break;
throw new InvalidArgumentException($operator.': Invalid operator specified');
}
return parent::__callStatic($operator, $operands);
}
function __call($operator, $operands) {
array_unshift($operands, $this);
return SqlExpression::__callStatic($operator, $operands);
}
class SqlInterval extends SqlFunction {
var $type;
function toSql($compiler, $model=false, $alias=false) {
$A = $this->args[0];
if ($A instanceof SqlFunction)
$A = $A->toSql($compiler, $model);
else
$A = $compiler->input($A);
return sprintf('INTERVAL %s %s',
$A,
$this->func)
. ($alias ? ' AS '.$compiler->quote($alias) : '');
}
static function __callStatic($interval, $args) {
if (count($args) != 1) {
throw new InvalidArgumentException("Interval expects a single interval value");
}
return parent::__callStatic($interval, $args);
}
}
class SqlField extends SqlExpression {
var $level;
function __construct($field, $level=0) {
}
function toSql($compiler, $model=false, $alias=false) {
$L = $this->level;
while ($L--)
$compiler = $compiler->getParent();
list($field) = $compiler->getField($this->field, $model);
return $field;
}
}
class SqlCode extends SqlFunction {
function __construct($code) {
$this->code = $code;
}
function toSql($compiler, $model=false, $alias=false) {
return $this->code.($alias ? ' AS '.$alias : '');
}
}
class SqlAggregate extends SqlFunction {
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
var $func;
var $expr;
var $distinct=false;
var $constraint=false;
function __construct($func, $expr, $distinct=false, $constraint=false) {
$this->func = $func;
$this->expr = $expr;
$this->distinct = $distinct;
if ($constraint instanceof Q)
$this->constraint = $constraint;
elseif ($constraint)
$this->constraint = new Q($constraint);
}
static function __callStatic($func, $args) {
$distinct = @$args[1] ?: false;
$constraint = @$args[2] ?: false;
return new static($func, $args[0], $distinct, $constraint);
}
function toSql($compiler, $model=false, $alias=false) {
$options = array('constraint' => $this->constraint, 'model' => true);
// For DISTINCT, require a field specification — not a relationship
// specification.
$E = $this->expr;
if ($E instanceof SqlFunction) {
$field = $E->toSql($compiler, $model);
}
else {
list($field, $rmodel) = $compiler->getField($E, $model, $options);
if ($this->distinct) {
$pk = false;
$fpk = $rmodel::getMeta('pk');
foreach ($fpk as $f) {
$pk |= false !== strpos($field, $f);
}
if (!$pk) {
// Try and use the foriegn primary key
list($field) = $compiler->getField(
$this->expr . '__' . $fpk[0],
$model, $options);
}
else {
throw new OrmException(
sprintf('%s :: %s', $rmodel, $field) .
': DISTINCT aggregate expressions require specification of a single primary key field of the remote model'
);
}
}
}
return sprintf('%s(%s%s)%s', $this->func,
$this->distinct ? 'DISTINCT ' : '', $field,
$alias && $this->alias ? ' AS '.$compiler->quote($this->alias) : '');
}
function getFieldName() {
return strtolower(sprintf('%s__%s', $this->args[0], $this->func));
}
}
class QuerySet implements IteratorAggregate, ArrayAccess, Serializable, Countable {
var $model;
var $constraints = array();
var $path_constraints = array();
var $ordering = array();
var $limit = false;
var $offset = 0;
var $related = array();
var $values = array();
var $defer = array();
var $aggregated = false;
var $extra = array();
var $distinct = array();
var $options = array();
const LOCK_EXCLUSIVE = 1;
const LOCK_SHARED = 2;
const ASC = 'ASC';
const DESC = 'DESC';
const OPT_NOSORT = 'nosort';
const OPT_NOCACHE = 'nocache';
const OPT_MYSQL_FOUND_ROWS = 'found_rows';
const ITER_MODELS = 1;
const ITER_HASH = 2;
const ITER_ROW = 3;
var $iter = self::ITER_MODELS;
var $compiler = 'MySqlCompiler';
var $query;
function __construct($model) {
$this->model = $model;
}
function filter() {
// Multiple arrays passes means OR
foreach (func_get_args() as $Q) {
$this->constraints[] = $Q instanceof Q ? $Q : new Q($Q);
return $this;
}
function exclude() {
foreach (func_get_args() as $Q) {
$this->constraints[] = $Q instanceof Q ? $Q->negate() : Q::not($Q);
/**
* Add a path constraint for the query. This is different from ::filter
* in that the constraint is added to a join clause which is normally
* built from the model meta data. The ::filter() method on the other
* hand adds the constraint to the where clause. This is generally useful
* for aggregate queries and left join queries where multiple rows might
* match a filter in the where clause and would produce incorrect results.
*
* Example:
* Find users with personal email hosted with gmail.
* >>> $Q = User::objects();
* >>> $Q->constrain(['user__emails' => new Q(['type' => 'personal']))
* >>> $Q->filter(['user__emails__address__contains' => '@gmail.com'])
*/
function constrain() {
foreach (func_get_args() as $I) {
foreach ($I as $path => $Q) {
if (!is_array($Q) && !$Q instanceof Q) {
// ->constrain(array('field__path__op' => val));
$Q = array($path => $Q);
list(, $path) = SqlCompiler::splitCriteria($path);
$path = implode('__', $path);
}
$this->path_constraints[$path][] = $Q instanceof Q ? $Q : Q::all($Q);
}
}
return $this;
}
function defer() {
foreach (func_get_args() as $f)
$this->defer[$f] = true;
return $this;
}
function order_by($order, $direction=false) {
if ($order === false)
return $this->options(array('nosort' => true));
$args = func_get_args();
if (in_array($direction, array(self::ASC, self::DESC))) {
$args = array($args[0]);
}
else
$direction = false;
$new = is_array($order) ? $order : $args;
if ($direction) {
foreach ($new as $i=>$x) {
$new[$i] = array($x, $direction);
}
}
$this->ordering = array_merge($this->ordering, $new);
function getSortFields() {
$ordering = $this->ordering;
if ($this->extra['order_by'])
$ordering = array_merge($ordering, $this->extra['order_by']);
return $ordering;
}
function lock($how=false) {
$this->lock = $how ?: self::LOCK_EXCLUSIVE;
return $this;
}
function limit($count) {
$this->limit = $count;
return $this;
}
function offset($at) {
$this->offset = $at;
return $this;
}
function isWindowed() {
return $this->limit || $this->offset || (count($this->values) + count($this->annotations) + @count($this->extra['select'])) > 1;
/**
* Fetch related fields with the query. This will result in better
* performance as related items are fetched with the root model with
* only one trip to the database.
*
* Either an array of fields can be sent as one argument, or the list of
* fields can be sent as the arguments to the function.
*
* Example:
* >>> $q = User::objects()->select_related('role');
*/
$args = func_get_args();
if (is_array($args[0]))
$args = $args[0];
$this->related = array_merge($this->related, $args);
function extra(array $extra) {
foreach ($extra as $section=>$info) {
$this->extra[$section] = array_merge($this->extra[$section] ?: array(), $info);
}
return $this;
}
function distinct() {
foreach (func_get_args() as $D)
$this->distinct[] = $D;
return $this;
}
function models() {
$this->iter = self::ITER_MODELS;
$this->values = $this->related = array();
return $this;
}
foreach (func_get_args() as $A)
$this->values[$A] = $A;
$this->iter = self::ITER_HASH;
// This disables related models
$this->related = false;
function values_flat() {
$this->values = func_get_args();
$this->iter = self::ITER_ROW;
// This disables related models
$this->related = false;
function copy() {
return clone $this;
}
return $this->getIterator()->asArray();
$list = $this->limit(1)->all();
/**
* one
*
* Finds and returns a single model instance based on the criteria in
* this QuerySet instance.
*
* Throws:
* DoesNotExist - if no such model exists with the given criteria
* ObjectNotUnique - if more than one model matches the given criteria
*
* Returns:
* (Object<Model>) a single instance of the sought model is guarenteed.
* If no such model or multiple models exist, an exception is thrown.
*/
if (count($list) == 0)
throw new DoesNotExist();
elseif (count($list) > 1)
throw new ObjectNotUnique('One object was expected; however '
.'multiple objects in the database matched the query. '
.sprintf('In fact, there are %d matching objects.', count($list))
);
// Defer to the iterator if fetching already started
if (isset($this->_iterator)) {
return $this->_iterator->count();
}
elseif (isset($this->count)) {
return $this->count;
$class = $this->compiler;
$compiler = new $class();
return $this->count = $compiler->compileCount($this);
/**
* Similar to count, except that the LIMIT and OFFSET parts are not
* considered in the counts. That is, this will return the count of rows
* if the query were not windowed with limit() and offset().
*
* For MySQL, the query will be submitted and fetched and the
* SQL_CALC_FOUND_ROWS hint will be sent in the query. Afterwards, the
* result of FOUND_ROWS() is fetched and is the result of this function.
*
* The result of this function is cached. If further changes are made
* after this is run, the changes should be made in a clone.
*/
function total() {
if (isset($this->total))
return $this->total;
1381
1382
1383
1384
1385
1386
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
// Optimize the query with the CALC_FOUND_ROWS if
// - the compiler supports it
// - the iterator hasn't yet been built, that is, the query for this
// statement has not yet been sent to the database
$compiler = $this->compiler;
if ($compiler::supportsOption(self::OPT_MYSQL_FOUND_ROWS)
&& !isset($this->_iterator)
) {
// This optimization requires caching
$this->options(array(
self::OPT_MYSQL_FOUND_ROWS => 1,
self::OPT_NOCACHE => null,
));
$this->exists(true);
$compiler = new $compiler();
return $this->total = $compiler->getFoundRows();
}
$query = clone $this;
$query->limit(false)->offset(false)->order_by(false);
return $this->total = $query->count();
}
function toSql($compiler, $model, $alias=false) {
// FIXME: Force root model of the compiler to $model
$exec = $this->getQuery(array('compiler' => get_class($compiler),
'parent' => $compiler, 'subquery' => true));
// Rewrite the parameter numbers so they fit the parameter numbers
// of the current parameters of the $compiler
$sql = preg_replace_callback("/:(\d+)/",
function($m) use ($compiler, $exec) {
$compiler->params[] = $exec->params[$m[1]-1];
return ':'.count($compiler->params);
}, $exec->sql);
return "({$sql})".($alias ? " AS {$alias}" : '');
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
/**
* exists
*
* Determines if there are any rows in this QuerySet. This can be
* achieved either by evaluating a SELECT COUNT(*) query or by
* attempting to fetch the first row from the recordset and return
* boolean success.
*
* Parameters:
* $fetch - (bool) TRUE if a compile and fetch should be attempted
* instead of a SELECT COUNT(*). This would be recommended if an
* accurate count is not required and the records would be fetched
* if this method returns TRUE.
*
* Returns:
* (bool) TRUE if there would be at least one record in this QuerySet
*/
function exists($fetch=false) {
if ($fetch) {
return (bool) $this[0];
}
return $this->count() > 0;
}
function annotate($annotations) {
if (!is_array($annotations))
$annotations = func_get_args();
foreach ($annotations as $name=>$A) {
if (is_int($name))
$name = $A->getFieldName();
$A->setAlias($name);
}
function aggregate($annotations) {
// Aggregate works like annotate, except that it sets up values
// fetching which will disable model creation
$this->annotate($annotations);
// Disable other fields from being fetched
$this->aggregated = true;
$this->related = false;
return $this;
}
function options($options) {
// Make an array with $options as the only key
if (!is_array($options))
$options = array($options => 1);
$this->options = array_merge($this->options, $options);
return $this;
}
function hasOption($option) {
return isset($this->options[$option]);
}
function countSelectFields() {
$count = count($this->values) + count($this->annotations);
if (isset($this->extra['select']))
foreach (@$this->extra['select'] as $S)
$count += count($S);
return $count;
}
function union(QuerySet $other, $all=true) {
// Values and values_list _must_ match for this to work
if ($this->countSelectFields() != $other->countSelectFields())
throw new OrmException('Union queries must have matching values counts');
// TODO: Clear OFFSET and LIMIT in the $other query
$this->chain[] = array($other, $all);
return $this;
}
function delete() {
$class = $this->compiler;
$compiler = new $class();
// XXX: Mark all in-memory cached objects as deleted
$ex = $compiler->compileBulkDelete($this);
$ex->execute();
return $ex->affected_rows();
}
function update(array $what) {
$class = $this->compiler;
$compiler = new $class;
$ex = $compiler->compileBulkUpdate($this, $what);
$ex->execute();
return $ex->affected_rows();
}
function __clone() {
unset($this->_iterator);
unset($this->query);
function __call($name, $args) {
if (!is_callable(array($this->getIterator(), $name)))
throw new OrmException('Call to undefined method QuerySet::'.$name);
return $args
? call_user_func_array(array($this->getIterator(), $name), $args)
: call_user_func(array($this->getIterator(), $name));
}
function getIterator($iterator=false) {
if (!isset($this->_iterator)) {
$class = $iterator ?: $this->getIteratorClass();
$it = new $class($this);
if (!$this->hasOption(self::OPT_NOCACHE)) {
if ($this->iter == self::ITER_MODELS)
// Add findFirst() and such
$it = new ModelResultSet($it);
else
$it = new CachedResultSet($it);
}
else {
$it = $it->getIterator();
}
$this->_iterator = $it;
}
function getIteratorClass() {
switch ($this->iter) {
case self::ITER_MODELS:
return 'ModelInstanceManager';
case self::ITER_HASH:
return 'HashArrayIterator';
case self::ITER_ROW:
return 'FlatArrayIterator';
}
}
// ArrayAccess interface
function offsetExists($offset) {
return $this->getIterator()->offsetExists($offset);
}
function offsetGet($offset) {
return $this->getIterator()->offsetGet($offset);
}
function offsetUnset($a) {
throw new Exception(__('QuerySet is read-only'));
throw new Exception(__('QuerySet is read-only'));
return (string) $this->getQuery();
function getQuery($options=array()) {
if (isset($this->query))
return $this->query;
// Load defaults from model
$model = $this->model;
$meta = $model::getMeta();
$options += $this->options;
if ($options['nosort'])
$query->ordering = array();
elseif (!$query->ordering && $meta['ordering'])
$query->ordering = $meta['ordering'];
if (false !== $query->related && !$query->related && !$query->values && $meta['select_related'])
$query->related = $meta['select_related'];
if (!$query->defer && $meta['defer'])
$query->defer = $meta['defer'];
$class = $options['compiler'] ?: $this->compiler;
$compiler = new $class($options);
$this->query = $compiler->compileSelect($query);
/**
* Fetch a model class which can be used to render the QuerySet as a
* subquery to be used as a JOIN.
*/
function asView() {
$unique = spl_object_hash($this);
$classname = "QueryView{$unique}";
if (class_exists($classname))
return $classname;
$class = <<<EOF
class {$classname} extends VerySimpleModel {
static \$meta = array(
'view' => true,
);
static \$queryset;
static function getQuery(\$compiler) {
return ' ('.static::\$queryset->getQuery().') ';
}
static function getSqlAddParams(\$compiler) {
return static::\$queryset->toSql(\$compiler, self::\$queryset->model);
}
EOF;
eval($class); // Ugh
$classname::$queryset = $this;
return $classname;
}
function serialize() {
$info = get_object_vars($this);
unset($info['query']);
unset($info['limit']);
unset($info['offset']);
unset($info['_iterator']);
return serialize($info);
}
function unserialize($data) {
$data = unserialize($data);
foreach ($data as $name => $val) {
$this->{$name} = $val;
}
}
class DoesNotExist extends Exception {}
class ObjectNotUnique extends Exception {}
class CachedResultSet
extends BaseList
implements ArrayAccess {
protected $inner;
protected $eoi = false;
function __construct(IteratorAggregate $iterator) {
$this->inner = $iterator->getIterator();
function fillTo($level) {
while (!$this->eoi && count($this->storage) < $level) {
if (!$this->inner->valid()) {
$this->eoi = true;
break;
}
$this->storage[] = $this->inner->current();
$this->inner->next();
}
function asArray() {
$this->fillTo(PHP_INT_MAX);
function getCache() {
return $this->storage;
}
function reset() {
$this->eoi = false;
$this->storage = array();
// XXX: Should the inner be recreated to refetch?
$this->inner->rewind();
function getIterator() {
$this->asArray();
return new ArrayIterator($this->storage);
}
function offsetExists($offset) {
$this->fillTo($offset+1);
return count($this->storage) > $offset;
$this->fillTo($offset+1);
return $this->storage[$offset];
throw new Exception(__('QuerySet is read-only'));
throw new Exception(__('QuerySet is read-only'));
return count($this->storage);
/**
* Sort the instrumented list in place. This would be useful to change the
* sorting order of the items in the list without fetching the list from
* the database again.
*
* Parameters:
* $key - (callable|int) A callable function to produce the sort keys
* or one of the SORT_ constants used by the array_multisort
* function
* $reverse - (bool) true if the list should be sorted descending
*
* Returns:
* This instrumented list for chaining and inlining.
*/
function sort($key=false, $reverse=false) {
// Fetch all records into the cache
$this->asArray();
parent::sort($key, $reverse);
return $this;
}
/**
* Reverse the list item in place. Returns this object for chaining
*/
function reverse() {
$this->asArray();
return parent::reverse();
class ModelResultSet
extends CachedResultSet {
/**
* Find the first item in the current set which matches the given criteria.
* This would be used in favor of ::filter() which might trigger another
* database query. The criteria is intended to be quite simple and should
* not traverse relationships which have not already been fetched.
* Otherwise, the ::filter() or ::window() methods would provide better
* performance.
*
* Example:
* >>> $a = new User();
* >>> $a->roles->add(Role::lookup(['name' => 'administator']));
* >>> $a->roles->findFirst(['roles__name__startswith' => 'admin']);
* <Role: administrator>
*/
function findFirst($criteria) {
$records = $this->findAll($criteria, 1);
return count($records) > 0 ? $records[0] : null;
}
/**
* Find all the items in the current set which match the given criteria.
* This would be used in favor of ::filter() which might trigger another
* database query. The criteria is intended to be quite simple and should
* not traverse relationships which have not already been fetched.
* Otherwise, the ::filter() or ::window() methods would provide better
* performance, as they can provide results with one more trip to the
* database.
*/
function findAll($criteria, $limit=false) {
$records = new ListObject();
foreach ($this as $record) {
$matches = true;
foreach ($criteria as $field=>$check) {
if (!SqlCompiler::evaluate($record, $check, $field)) {
$matches = false;
break;
}
}
if ($matches)
$records[] = $record;
if ($limit && count($records) == $limit)
break;
}
return $records;
}
}
class ModelInstanceManager
implements IteratorAggregate {
var $queryset;
var $model;
var $map;
static $objectCache = array();
function __construct(QuerySet $queryset) {
$this->queryset = $queryset;
$this->model = $queryset->model;
}
function cache($model) {
$key = sprintf('%s.%s',
$model::$meta->model, implode('.', $model->get('pk')));
self::$objectCache[$key] = $model;
}
/**
* uncache
*
* Drop the cached reference to the model. If the model is deleted
* database-side. Lookups for the same model should not be short
* circuited to retrieve the cached reference.
*/
static function uncache($model) {
$key = sprintf('%s.%s',
$model::$meta->model, implode('.', $model->pk));
unset(self::$objectCache[$key]);
}
static function flushCache() {
self::$objectCache = array();
}
static function checkCache($modelClass, $fields) {
$key = $modelClass::$meta->model;
foreach ($modelClass::getMeta('pk') as $f)
$key .= '.'.$fields[$f];
return @self::$objectCache[$key];
}
/**
* getOrBuild
*
* Builds a new model from the received fields or returns the model
* already stashed in the model cache. Caching helps to ensure that
* multiple lookups for the same model identified by primary key will
* fetch the exact same model. Therefore, changes made to the model
* anywhere in the project will be reflected everywhere.
*
* For annotated models (models build from querysets with annotations),
* the built or cached model is wrapped in an AnnotatedModel instance.
* The annotated fields are in the AnnotatedModel instance and the
* database-backed fields are managed by the Model instance.
*/
function getOrBuild($modelClass, $fields, $cache=true) {
// Check for NULL primary key, used with related model fetching. If
// the PK is NULL, then consider the object to also be NULL
foreach ($modelClass::getMeta('pk') as $pkf) {
if (!isset($fields[$pkf])) {
return null;
}
}
$annotations = $this->queryset->annotations;
$extras = array();
// For annotations, drop them from the $fields list and add them to
// an $extras list. The fields passed to the root model should only
// be the root model's fields. The annotated fields will be wrapped
// using an AnnotatedModel instance.
if ($annotations && $modelClass == $this->model) {
if (array_key_exists($name, $fields)) {
$extras[$name] = $fields[$name];
unset($fields[$name]);
}
}
}
// Check the cache for the model instance first
if (!($m = self::checkCache($modelClass, $fields))) {
// Construct and cache the object
$m = $modelClass::__hydrate($fields);
// XXX: defer may refer to fields not in this model
$m->__deferred__ = $this->queryset->defer;
$m->__onload();
if ($cache)
$this->cache($m);
}
// Wrap annotations in an AnnotatedModel
if ($extras) {
$m = AnnotatedModel::wrap($m, $extras, $modelClass);
// TODO: If the model has deferred fields which are in $fields,
// those can be resolved here
return $m;
}
/**
* buildModel
*
* This method builds the model including related models from the record
* received. For related recordsets, a $map should be setup inside this
* object prior to using this method. The $map is assumed to have this
* configuration:
*
* array(array(<fieldNames>, <modelClass>, <relativePath>))
*
* Where $modelClass is the name of the foreign (with respect to the
* root model ($this->model), $fieldNames is the number and names of
* fields in the row for this model, $relativePath is the path that
* describes the relationship between the root model and this model,
* 'user__account' for instance.
function buildModel($row, $cache=true) {
if ($this->map) {
if ($this->model != $this->map[0][1])
throw new OrmException('Internal select_related error');
$offset = 0;
foreach ($this->map as $info) {
@list($fields, $model_class, $path) = $info;
$values = array_slice($row, $offset, count($fields));
$record = array_combine($fields, $values);
if (!$path) {
// Build the root model
$model = $this->getOrBuild($this->model, $record, $cache);
$i = 0;
// Traverse the declared path and link the related model
$tail = array_pop($path);
$m = $model;
foreach ($path as $field) {
if (!($m = $m->get($field)))
break;
if ($m) {
// Only apply cache setting to the root model.
// Reference models should use caching
$m->set($tail, $this->getOrBuild($model_class, $record, $cache));
}
$model = $this->getOrBuild($this->model, $row, $cache);
function getIterator() {
$this->resource = $this->queryset->getQuery();
$this->map = $this->resource->getMap();
$cache = !$this->queryset->hasOption(QuerySet::OPT_NOCACHE);
$this->resource->setBuffered($cache);
$func = ($this->map) ? 'getRow' : 'getArray';
1961
1962
1963
1964
1965
1966
1967
1968
1969
1970
1971
1972
1973
1974
1975
1976
1977
1978
1979
1980
1981
1982
1983
1984
$func = array($this->resource, $func);
return new CallbackSimpleIterator(function() use ($func, $cache) {
global $StopIteration;
if ($row = $func())
return $this->buildModel($row, $cache);
$this->resource->close();
throw $StopIteration;
});
}
}
class CallbackSimpleIterator
implements Iterator {
var $current;
var $eoi;
var $callback;
var $key = -1;
function __construct($callback) {
assert(is_callable($callback));
$this->callback = $callback;
1987
1988
1989
1990
1991
1992
1993
1994
1995
1996
1997
1998
1999
2000
2001
2002
2003
2004
2005
2006
2007
2008
2009
2010
2011
2012
2013
2014
function rewind() {
$this->eoi = false;
$this->next();
}
function key() {
return $this->key;
}
function valid() {
if (!isset($this->eoi))
$this->rewind();
return !$this->eoi;
}
function current() {
if ($this->eoi) return false;
return $this->current;
}
function next() {
try {
$cbk = $this->callback;
$this->current = $cbk();
$this->key++;
}
catch (StopIteration $x) {
$this->eoi = true;
2019
2020
2021
2022
2023
2024
2025
2026
2027
2028
2029
2030
2031
2032
2033
2034
2035
2036
2037
2038
2039
2040
2041
2042
// Use a global variable, as constructing exceptions is expensive
class StopIteration extends Exception {}
$StopIteration = new StopIteration();
class FlatArrayIterator
implements IteratorAggregate {
var $queryset;
var $resource;
function __construct(QuerySet $queryset) {
$this->queryset = $queryset;
}
function getIterator() {
$this->resource = $this->queryset->getQuery();
return new CallbackSimpleIterator(function() {
global $StopIteration;
if ($row = $this->resource->getRow())
return $row;
$this->resource->close();
throw $StopIteration;
});
class HashArrayIterator
implements IteratorAggregate {
var $queryset;
var $resource;
function __construct(QuerySet $queryset) {
$this->queryset = $queryset;
}
function getIterator() {
$this->resource = $this->queryset->getQuery();
return new CallbackSimpleIterator(function() {
global $StopIteration;
if ($row = $this->resource->getArray())
return $row;
$this->resource->close();
throw $StopIteration;
});
class InstrumentedList
extends ModelResultSet {
function __construct($fkey, $queryset=false,
$iterator='ModelInstanceManager'
) {
list($model, $this->key) = $fkey;
$queryset = $model::objects()->filter($this->key);
if ($related = $model::getMeta('select_related'))
$queryset->select_related($related);
}
parent::__construct(new $iterator($queryset));
$this->queryset = $queryset;
function add($object, $save=true, $at=false) {
// NOTE: Attempting to compare $object to $this->model will likely
// be problematic, and limits creative applications of the ORM
if (!$object) {
'Attempting to add invalid object to list. Expected <%s>, but got <NULL>',
$this->model
foreach ($this->key as $field=>$value)
$object->set($field, $value);
if (!$object->__new__ && $save)
$object->save();
if ($at !== false)
$this->storage[$at] = $object;
$this->storage[] = $object;
function remove($object, $delete=true) {
if ($delete)
$object->delete();
else
foreach ($this->key as $field=>$value)
$object->set($field, null);
* Slight edit to the standard iteration method which will skip deleted
* items.
function getIterator() {
return new CallbackFilterIterator(parent::getIterator(),
function($i) { return !$i->__deleted__; }
);
/**
* Reduce the list to a subset using a simply key/value constraint. New
* items added to the subset will have the constraint automatically
* added to all new items.
*
* Parameters:
* $criteria - (<Traversable>) criteria by which this list will be
* constrained and filtered.
* $evaluate - (<bool>) if set to TRUE, the criteria will be evaluated
* without making any more trips to the database. NOTE this may yield
* unexpected results if this list does not contain all the records
* from the database which would be matched by another query.
function window($constraint, $evaluate=false) {
$fields = $model::getMeta()->getFieldNames();
$key = $this->key;
foreach ($constraint as $field=>$value) {
if (!is_string($field) || false === in_array($field, $fields))
throw new OrmException('InstrumentedList windowing must be performed on local fields only');
$key[$field] = $value;
}
$list = new static(array($this->model, $key), $this->filter($constraint));
if ($evaluate) {
$list->setCache($this->findAll($constraint));
}
return $list;
}
/**
* Disable database fetching on this list by providing a static list of
* objects. ::add() and ::remove() are still supported.
* XXX: Move this to a parent class?
*/
function setCache(array $cache) {
if (count($this->storage) > 0)
throw new Exception('Cache must be set before fetching records');
// Set cache and disable fetching
$this->reset();
$this->storage = $cache;
// Save all changes made to any list items
function saveAll() {
foreach ($this as $I)
if (!$I->save())
return false;
return true;
}
function exists() {
return $this->queryset->exists();
}
function expunge() {
if ($this->queryset->delete())
$this->reset();
function update(array $what) {
return $this->queryset->update($what);
}
// Fetch a new QuerySet
function objects() {
return clone $this->queryset;
function offsetUnset($a) {
$this->fillTo($a);
$this->storage[$a]->delete();
}
function offsetSet($a, $b) {
$this->fillTo($a);
Loading
Loading full blame...