X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=classes%2FManaged_DataObject.php;h=a69a957bcc0225bad8b4085312b1b73a53b9ea13;hb=a39f51c0441b22951412b2c00d88c34f39cb39c9;hp=ae37dd02eaa911f5846bc47f068c144fd31244a9;hpb=50e611a1a9934ceae665ceb27a2765b48125b4f4;p=quix0rs-gnu-social.git diff --git a/classes/Managed_DataObject.php b/classes/Managed_DataObject.php index ae37dd02ea..a69a957bcc 100644 --- a/classes/Managed_DataObject.php +++ b/classes/Managed_DataObject.php @@ -64,6 +64,11 @@ abstract class Managed_DataObject extends Memcached_DataObject return parent::pkeyGetClass(get_called_class(), $kv); } + static function pkeyCols() + { + return parent::pkeyColsClass(get_called_class()); + } + /** * Get multiple items from the database by key * @@ -111,7 +116,7 @@ abstract class Managed_DataObject extends Memcached_DataObject } /** - * Get a multi-instance object in an array + * Get a multi-instance object separated into an array * * This is a utility method to get multiple instances with a given set of * values for a specific key column. Usually used for the primary key when @@ -134,7 +139,7 @@ abstract class Managed_DataObject extends Memcached_DataObject * @access public * @return array (associative) */ - function table() + public function table() { $table = static::schemaDef(); return array_map(array($this, 'columnBitmap'), $table['fields']); @@ -164,7 +169,7 @@ abstract class Managed_DataObject extends Memcached_DataObject function sequenceKey() { - $table = call_user_func(array(get_class($this), 'schemaDef')); + $table = static::schemaDef(); foreach ($table['fields'] as $name => $column) { if ($column['type'] == 'serial') { // We have a serial/autoincrement column. @@ -188,7 +193,7 @@ abstract class Managed_DataObject extends Memcached_DataObject function keyTypes() { - $table = call_user_func(array(get_class($this), 'schemaDef')); + $table = static::schemaDef(); $keys = array(); if (!empty($table['unique keys'])) { @@ -253,7 +258,7 @@ abstract class Managed_DataObject extends Memcached_DataObject { $links = array(); - $table = call_user_func(array(get_class($this), 'schemaDef')); + $table = static::schemaDef(); foreach ($table['foreign keys'] as $keyname => $keydef) { if (count($keydef) == 2 && is_string($keydef[0]) && is_array($keydef[1]) && count($keydef[1]) == 1) { @@ -274,7 +279,7 @@ abstract class Managed_DataObject extends Memcached_DataObject */ function _allCacheKeys() { - $table = call_user_func(array(get_class($this), 'schemaDef')); + $table = static::schemaDef(); $ckeys = array(); if (!empty($table['unique keys'])) { @@ -298,4 +303,154 @@ abstract class Managed_DataObject extends Memcached_DataObject } return $ckeys; } + + public function escapedTableName() + { + return common_database_tablename($this->tableName()); + } + + /** + * Returns an object by looking at the primary key column(s). + * + * Will require all primary key columns to be defined in an associative array + * and ignore any keys which are not part of the primary key. + * + * Will NOT accept NULL values as part of primary key. + * + * @param array $vals Must match all primary key columns for the dataobject. + * + * @return Managed_DataObject of the get_called_class() type + * @throws NoResultException if no object with that primary key + */ + static function getByPK(array $vals) + { + $classname = get_called_class(); + + $pkey = static::pkeyCols(); + if (is_null($pkey)) { + throw new ServerException("Failed to get primary key columns for class '{$classname}'"); + } + + $object = new $classname(); + foreach ($pkey as $col) { + if (!array_key_exists($col, $vals)) { + throw new ServerException("Missing primary key column '{$col}'"); + } elseif (is_null($vals[$col])) { + throw new ServerException("NULL values not allowed in getByPK for column '{$col}'"); + } + $object->$col = $vals[$col]; + } + if (!$object->find(true)) { + throw new NoResultException($object); + } + return $object; + } + + static function getByID($id) + { + if (empty($id)) { + throw new ServerException('Empty ID on lookup'); + } + // getByPK throws exception if id is null + // or if the class does not have a single 'id' column as primary key + return static::getByPK(array('id' => $id)); + } + + /** + * Returns an ID, checked that it is set and reasonably valid + * + * If this dataobject uses a special id field (not 'id'), just + * implement your ID getting method in the child class. + * + * @return int ID of dataobject + * @throws Exception (when ID is not available or not set yet) + */ + public function getID() + { + // FIXME: Make these exceptions more specific (their own classes) + if (!isset($this->id)) { + throw new Exception('No ID set.'); + } elseif (empty($this->id)) { + throw new Exception('Empty ID for object! (not inserted yet?).'); + } + + // FIXME: How about forcing to return an int? Or will that overflow eventually? + return $this->id; + } + + // 'update' won't write key columns, so we have to do it ourselves. + // This also automatically calls "update" _before_ it sets the keys. + // FIXME: This only works with single-column primary keys so far! Beware! + /** + * @param DB_DataObject &$orig Must be "instanceof" $this + * @param string $pid Primary ID column (no escaping is done on column name!) + */ + public function updateWithKeys(&$orig, $pid='id') + { + if (!$orig instanceof $this) { + throw new ServerException('Tried updating a DataObject with a different class than itself.'); + } + + // do it in a transaction + $this->query('BEGIN'); + + $parts = array(); + foreach ($this->keys() as $k) { + if (strcmp($this->$k, $orig->$k) != 0) { + $parts[] = $k . ' = ' . $this->_quote($this->$k); + } + } + if (count($parts) == 0) { + // No changes to keys, it's safe to run ->update(...) + if ($this->update($orig) === false) { + common_log_db_error($this, 'UPDATE', __FILE__); + // rollback as something bad occurred + $this->query('ROLLBACK'); + throw new ServerException("Could not UPDATE non-keys for {$this->__table}"); + } + $orig->decache(); + $this->encache(); + + // commit our db transaction since we won't reach the COMMIT below + $this->query('COMMIT'); + // @FIXME return true only if something changed (otherwise 0) + return true; + } + + $qry = sprintf('UPDATE %1$s SET %2$s WHERE %3$s = %4$s', + common_database_tablename($this->tableName()), + implode(', ', $parts), + $pid, + $this->_quote($this->$pid)); + + $result = $this->query($qry); + if ($result === false) { + common_log_db_error($this, 'UPDATE', __FILE__); + // rollback as something bad occurred + $this->query('ROLLBACK'); + throw new ServerException("Could not UPDATE key fields for {$this->__table}"); + } + + // Update non-keys too, if the previous endeavour worked. + // The ->update call uses "$this" values for keys, that's why we can't do this until + // the keys are updated (because they might differ from $orig and update the wrong entries). + if ($this->update($orig) === false) { + common_log_db_error($this, 'UPDATE', __FILE__); + // rollback as something bad occurred + $this->query('ROLLBACK'); + throw new ServerException("Could not UPDATE non-keys for {$this->__table}"); + } + $orig->decache(); + $this->encache(); + + // commit our db transaction + $this->query('COMMIT'); + // @FIXME return true only if something changed (otherwise 0) + return $result; + } + + static public function beforeSchemaUpdate() + { + // NOOP + } }