includes/db/ORMRow.php

   1 <?php
   2 /**
   3  * Abstract base class for representing objects that are stored in some DB table.
   4  * This is basically an ORM-like wrapper around rows in database tables that
   5  * aims to be both simple and very flexible. It is centered around an associative
   6  * array of fields and various methods to do common interaction with the database.
   7  *
   8  * Documentation inline and at https://www.mediawiki.org/wiki/Manual:ORMTable
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License along
  21  * with this program; if not, write to the Free Software Foundation, Inc.,
  22  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  23  * http://www.gnu.org/copyleft/gpl.html
  24  *
  25  * @since 1.20
  26  *
  27  * @file ORMRow.php
  28  * @ingroup ORM
  29  *
  30  * @license GNU GPL v2 or later
  31  * @author Jeroen De Dauw < jeroendedauw@gmail.com >
  32  */
  33
  34 class ORMRow implements IORMRow {
  35
  36         /**
  37          * The fields of the object.
  38          * field name (w/o prefix) => value
  39          *
  40          * @since 1.20
  41          * @var array
  42          */
  43         protected $fields = array( 'id' => null );
  44
  45         /**
  46          * If the object should update summaries of linked items when changed.
  47          * For example, update the course_count field in universities when a course in courses is deleted.
  48          * Settings this to false can prevent needless updating work in situations
  49          * such as deleting a university, which will then delete all it's courses.
  50          *
  51          * @deprecated since 1.22
  52          * @since 1.20
  53          * @var bool
  54          */
  55         protected $updateSummaries = true;
  56
  57         /**
  58          * Indicates if the object is in summary mode.
  59          * This mode indicates that only summary fields got updated,
  60          * which allows for optimizations.
  61          *
  62          * @deprecated since 1.22
  63          * @since 1.20
  64          * @var bool
  65          */
  66         protected $inSummaryMode = false;
  67
  68         /**
  69          * @deprecated since 1.22
  70          * @since 1.20
  71          * @var ORMTable|null
  72          */
  73         protected $table;
  74
  75         /**
  76          * Constructor.
  77          *
  78          * @since 1.20
  79          *
  80          * @param IORMTable|null $table Deprecated since 1.22
  81          * @param array|null $fields
  82          * @param boolean $loadDefaults Deprecated since 1.22
  83          */
  84         public function __construct( IORMTable $table = null, $fields = null, $loadDefaults = false ) {
  85                 $this->table = $table;
  86
  87                 if ( !is_array( $fields ) ) {
  88                         $fields = array();
  89                 }
  90
  91                 if ( $loadDefaults ) {
  92                         $fields = array_merge( $this->table->getDefaults(), $fields );
  93                 }
  94
  95                 $this->setFields( $fields );
  96         }
  97
  98         /**
  99          * Load the specified fields from the database.
 100          *
 101          * @since 1.20
 102          * @deprecated since 1.22
 103          *
 104          * @param array|null $fields
 105          * @param boolean $override
 106          * @param boolean $skipLoaded
 107          *
 108          * @return bool Success indicator
 109          */
 110         public function loadFields( $fields = null, $override = true, $skipLoaded = false ) {
 111                 if ( is_null( $this->getId() ) ) {
 112                         return false;
 113                 }
 114
 115                 if ( is_null( $fields ) ) {
 116                         $fields = array_keys( $this->table->getFields() );
 117                 }
 118
 119                 if ( $skipLoaded ) {
 120                         $fields = array_diff( $fields, array_keys( $this->fields ) );
 121                 }
 122
 123                 if ( !empty( $fields ) ) {
 124                         $result = $this->table->rawSelectRow(
 125                                 $this->table->getPrefixedFields( $fields ),
 126                                 array( $this->table->getPrefixedField( 'id' ) => $this->getId() ),
 127                                 array( 'LIMIT' => 1 ),
 128                                 __METHOD__
 129                         );
 130
 131                         if ( $result !== false ) {
 132                                 $this->setFields( $this->table->getFieldsFromDBResult( $result ), $override );
 133                                 return true;
 134                         }
 135                         return false;
 136                 }
 137
 138                 return true;
 139         }
 140
 141         /**
 142          * Gets the value of a field.
 143          *
 144          * @since 1.20
 145          *
 146          * @param string $name Field name
 147          * @param $default mixed: Default value to return when none is found
 148          * (default: null)
 149          *
 150          * @throws MWException
 151          * @return mixed
 152          */
 153         public function getField( $name, $default = null ) {
 154                 if ( $this->hasField( $name ) ) {
 155                         return $this->fields[$name];
 156                 } elseif ( !is_null( $default ) ) {
 157                         return $default;
 158                 } else {
 159                         throw new MWException( 'Attempted to get not-set field ' . $name );
 160                 }
 161         }
 162
 163         /**
 164          * Gets the value of a field but first loads it if not done so already.
 165          *
 166          * @since 1.20
 167          * @deprecated since 1.22
 168          *
 169          * @param $name string
 170          *
 171          * @return mixed
 172          */
 173         public function loadAndGetField( $name ) {
 174                 if ( !$this->hasField( $name ) ) {
 175                         $this->loadFields( array( $name ) );
 176                 }
 177
 178                 return $this->getField( $name );
 179         }
 180
 181         /**
 182          * Remove a field.
 183          *
 184          * @since 1.20
 185          *
 186          * @param string $name
 187          */
 188         public function removeField( $name ) {
 189                 unset( $this->fields[$name] );
 190         }
 191
 192         /**
 193          * Returns the objects database id.
 194          *
 195          * @since 1.20
 196          *
 197          * @return integer|null
 198          */
 199         public function getId() {
 200                 return $this->getField( 'id' );
 201         }
 202
 203         /**
 204          * Sets the objects database id.
 205          *
 206          * @since 1.20
 207          *
 208          * @param integer|null $id
 209          */
 210         public function setId( $id ) {
 211                 $this->setField( 'id', $id );
 212         }
 213
 214         /**
 215          * Gets if a certain field is set.
 216          *
 217          * @since 1.20
 218          *
 219          * @param string $name
 220          *
 221          * @return boolean
 222          */
 223         public function hasField( $name ) {
 224                 return array_key_exists( $name, $this->fields );
 225         }
 226
 227         /**
 228          * Gets if the id field is set.
 229          *
 230          * @since 1.20
 231          *
 232          * @return boolean
 233          */
 234         public function hasIdField() {
 235                 return $this->hasField( 'id' )
 236                         && !is_null( $this->getField( 'id' ) );
 237         }
 238
 239         /**
 240          * Gets the fields => values to write to the table.
 241          *
 242          * @since 1.20
 243          * @deprecated since 1.22
 244          *
 245          * @return array
 246          */
 247         protected function getWriteValues() {
 248                 $values = array();
 249
 250                 foreach ( $this->table->getFields() as $name => $type ) {
 251                         if ( array_key_exists( $name, $this->fields ) ) {
 252                                 $value = $this->fields[$name];
 253
 254                                 // Skip null id fields so that the DBMS can set the default.
 255                                 if ( $name === 'id' && is_null ( $value ) ) {
 256                                         continue;
 257                                 }
 258
 259                                 switch ( $type ) {
 260                                         case 'array':
 261                                                 $value = (array)$value;
 262                                         // fall-through!
 263                                         case 'blob':
 264                                                 $value = serialize( $value );
 265                                         // fall-through!
 266                                 }
 267
 268                                 $values[$this->table->getPrefixedField( $name )] = $value;
 269                         }
 270                 }
 271
 272                 return $values;
 273         }
 274
 275         /**
 276          * Sets multiple fields.
 277          *
 278          * @since 1.20
 279          *
 280          * @param array $fields The fields to set
 281          * @param boolean $override Override already set fields with the provided values?
 282          */
 283         public function setFields( array $fields, $override = true ) {
 284                 foreach ( $fields as $name => $value ) {
 285                         if ( $override || !$this->hasField( $name ) ) {
 286                                 $this->setField( $name, $value );
 287                         }
 288                 }
 289         }
 290
 291         /**
 292          * Serializes the object to an associative array which
 293          * can then easily be converted into JSON or similar.
 294          *
 295          * @since 1.20
 296          *
 297          * @param null|array $fields
 298          * @param boolean $incNullId
 299          *
 300          * @return array
 301          */
 302         public function toArray( $fields = null, $incNullId = false ) {
 303                 $data = array();
 304                 $setFields = array();
 305
 306                 if ( !is_array( $fields ) ) {
 307                         $setFields = $this->getSetFieldNames();
 308                 } else {
 309                         foreach ( $fields as $field ) {
 310                                 if ( $this->hasField( $field ) ) {
 311                                         $setFields[] = $field;
 312                                 }
 313                         }
 314                 }
 315
 316                 foreach ( $setFields as $field ) {
 317                         if ( $incNullId || $field != 'id' || $this->hasIdField() ) {
 318                                 $data[$field] = $this->getField( $field );
 319                         }
 320                 }
 321
 322                 return $data;
 323         }
 324
 325         /**
 326          * Load the default values, via getDefaults.
 327          *
 328          * @since 1.20
 329          * @deprecated since 1.22
 330          *
 331          * @param boolean $override
 332          */
 333         public function loadDefaults( $override = true ) {
 334                 $this->setFields( $this->table->getDefaults(), $override );
 335         }
 336
 337         /**
 338          * Writes the answer to the database, either updating it
 339          * when it already exists, or inserting it when it doesn't.
 340          *
 341          * @since 1.20
 342          * @deprecated since 1.22 Use IORMTable->updateRow or ->insertRow
 343          *
 344          * @param string|null $functionName
 345          *
 346          * @return boolean Success indicator
 347          */
 348         public function save( $functionName = null ) {
 349                 if ( $this->hasIdField() ) {
 350                         return $this->table->updateRow( $this, $functionName );
 351                 } else {
 352                         return $this->table->insertRow( $this, $functionName );
 353                 }
 354         }
 355
 356         /**
 357          * Updates the object in the database.
 358          *
 359          * @since 1.20
 360          * @deprecated since 1.22
 361          *
 362          * @param string|null $functionName
 363          *
 364          * @return boolean Success indicator
 365          */
 366         protected function saveExisting( $functionName = null ) {
 367                 $dbw = $this->table->getWriteDbConnection();
 368
 369                 $success = $dbw->update(
 370                         $this->table->getName(),
 371                         $this->getWriteValues(),
 372                         $this->table->getPrefixedValues( $this->getUpdateConditions() ),
 373                         is_null( $functionName ) ? __METHOD__ : $functionName
 374                 );
 375
 376                 $this->table->releaseConnection( $dbw );
 377
 378                 // DatabaseBase::update does not always return true for success as documented...
 379                 return $success !== false;
 380         }
 381
 382         /**
 383          * Returns the WHERE considtions needed to identify this object so
 384          * it can be updated.
 385          *
 386          * @since 1.20
 387          *
 388          * @return array
 389          */
 390         protected function getUpdateConditions() {
 391                 return array( 'id' => $this->getId() );
 392         }
 393
 394         /**
 395          * Inserts the object into the database.
 396          *
 397          * @since 1.20
 398          * @deprecated since 1.22
 399          *
 400          * @param string|null $functionName
 401          * @param array|null $options
 402          *
 403          * @return boolean Success indicator
 404          */
 405         protected function insert( $functionName = null, array $options = null ) {
 406                 $dbw = $this->table->getWriteDbConnection();
 407
 408                 $success = $dbw->insert(
 409                         $this->table->getName(),
 410                         $this->getWriteValues(),
 411                         is_null( $functionName ) ? __METHOD__ : $functionName,
 412                         $options
 413                 );
 414
 415                 // DatabaseBase::insert does not always return true for success as documented...
 416                 $success = $success !== false;
 417
 418                 if ( $success ) {
 419                         $this->setField( 'id', $dbw->insertId() );
 420                 }
 421
 422                 $this->table->releaseConnection( $dbw );
 423
 424                 return $success;
 425         }
 426
 427         /**
 428          * Removes the object from the database.
 429          *
 430          * @since 1.20
 431          * @deprecated since 1.22, use IORMTable->removeRow
 432          *
 433          * @return boolean Success indicator
 434          */
 435         public function remove() {
 436                 $this->beforeRemove();
 437
 438                 $success = $this->table->removeRow( $this, __METHOD__ );
 439
 440                 if ( $success ) {
 441                         $this->onRemoved();
 442                 }
 443
 444                 return $success;
 445         }
 446
 447         /**
 448          * Gets called before an object is removed from the database.
 449          *
 450          * @since 1.20
 451          * @deprecated since 1.22
 452          */
 453         protected function beforeRemove() {
 454                 $this->loadFields( $this->getBeforeRemoveFields(), false, true );
 455         }
 456
 457         /**
 458          * Before removal of an object happens, @see beforeRemove gets called.
 459          * This method loads the fields of which the names have been returned by this one (or all fields if null is returned).
 460          * This allows for loading info needed after removal to get rid of linked data and the like.
 461          *
 462          * @since 1.20
 463          *
 464          * @return array|null
 465          */
 466         protected function getBeforeRemoveFields() {
 467                 return array();
 468         }
 469
 470         /**
 471          * Gets called after successful removal.
 472          * Can be overridden to get rid of linked data.
 473          *
 474          * @since 1.20
 475          * @deprecated since 1.22
 476          */
 477         protected function onRemoved() {
 478                 $this->setField( 'id', null );
 479         }
 480
 481         /**
 482          * Return the names and values of the fields.
 483          *
 484          * @since 1.20
 485          *
 486          * @return array
 487          */
 488         public function getFields() {
 489                 return $this->fields;
 490         }
 491
 492         /**
 493          * Return the names of the fields.
 494          *
 495          * @since 1.20
 496          *
 497          * @return array
 498          */
 499         public function getSetFieldNames() {
 500                 return array_keys( $this->fields );
 501         }
 502
 503         /**
 504          * Sets the value of a field.
 505          * Strings can be provided for other types,
 506          * so this method can be called from unserialization handlers.
 507          *
 508          * @since 1.20
 509          *
 510          * @param string $name
 511          * @param mixed $value
 512          *
 513          * @throws MWException
 514          */
 515         public function setField( $name, $value ) {
 516                 $this->fields[$name] = $value;
 517         }
 518
 519         /**
 520          * Add an amount (can be negative) to the specified field (needs to be numeric).
 521          *
 522          * @since 1.20
 523          * @deprecated since 1.22, use IORMTable->addToField
 524          *
 525          * @param string $field
 526          * @param integer $amount
 527          *
 528          * @return boolean Success indicator
 529          */
 530         public function addToField( $field, $amount ) {
 531                 return $this->table->addToField( $this->getUpdateConditions(), $field, $amount );
 532         }
 533
 534         /**
 535          * Return the names of the fields.
 536          *
 537          * @since 1.20
 538          * @deprecated since 1.22
 539          *
 540          * @return array
 541          */
 542         public function getFieldNames() {
 543                 return array_keys( $this->table->getFields() );
 544         }
 545
 546         /**
 547          * Computes and updates the values of the summary fields.
 548          *
 549          * @since 1.20
 550          * @deprecated since 1.22
 551          *
 552          * @param array|string|null $summaryFields
 553          */
 554         public function loadSummaryFields( $summaryFields = null ) {
 555
 556         }
 557
 558         /**
 559          * Sets the value for the @see $updateSummaries field.
 560          *
 561          * @since 1.20
 562          * @deprecated since 1.22
 563          *
 564          * @param boolean $update
 565          */
 566         public function setUpdateSummaries( $update ) {
 567                 $this->updateSummaries = $update;
 568         }
 569
 570         /**
 571          * Sets the value for the @see $inSummaryMode field.
 572          *
 573          * @since 1.20
 574          * @deprecated since 1.22
 575          *
 576          * @param boolean $summaryMode
 577          */
 578         public function setSummaryMode( $summaryMode ) {
 579                 $this->inSummaryMode = $summaryMode;
 580         }
 581
 582         /**
 583          * Returns the table this IORMRow is a row in.
 584          *
 585          * @since 1.20
 586          * @deprecated since 1.22
 587          *
 588          * @return IORMTable
 589          */
 590         public function getTable() {
 591                 return $this->table;
 592         }
 593
 594 }