return parent::pkeyGetClass(get_called_class(), $kv);
}
+ static function pkeyCols()
+ {
+ return parent::pkeyColsClass(get_called_class());
+ }
+
/**
* Get multiple items from the database by key
*
}
/**
- * 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
* @access public
* @return array (associative)
*/
- function table()
+ public function table()
{
$table = static::schemaDef();
return array_map(array($this, 'columnBitmap'), $table['fields']);
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.
function keyTypes()
{
- $table = call_user_func(array(get_class($this), 'schemaDef'));
+ $table = static::schemaDef();
$keys = array();
if (!empty($table['unique keys'])) {
{
$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) {
*/
function _allCacheKeys()
{
- $table = call_user_func(array(get_class($this), 'schemaDef'));
+ $table = static::schemaDef();
$ckeys = array();
if (!empty($table['unique keys'])) {
}
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->tableName()}");
+ }
+ $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->tableName()}");
+ }
+
+ // 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->tableName()}");
+ }
+ $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
+ }
}
projects / quix0rs-gnu-social.git / blobdiff