ActiveRecord.php 24 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
<?php
/**
 * @link http://www.yiiframework.com/
 * @copyright Copyright (c) 2008 Yii Software LLC
 * @license http://www.yiiframework.com/license/
 */

namespace yii\sphinx;

use yii\base\InvalidConfigException;
11
use yii\base\NotSupportedException;
12
use yii\db\BaseActiveRecord;
13 14
use yii\db\StaleObjectException;
use yii\helpers\Inflector;
15
use Yii;
16
use yii\helpers\StringHelper;
17 18

/**
19 20
 * ActiveRecord is the base class for classes representing relational data in terms of objects.
 *
21 22 23
 * Warning: optimistic lock will NOT work in case of updating fields (not attributes) for the
 * runtime indexes!
 *
24 25
 * @property string $snippet Snippet value.
 * @property string $snippetSource Snippet source string. This property is read-only.
26 27 28 29
 *
 * @author Paul Klimov <klimov.paul@gmail.com>
 * @since 2.0
 */
30
abstract class ActiveRecord extends BaseActiveRecord
31
{
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
    /**
     * The insert operation. This is mainly used when overriding [[transactions()]] to specify which operations are transactional.
     */
    const OP_INSERT = 0x01;
    /**
     * The update operation. This is mainly used when overriding [[transactions()]] to specify which operations are transactional.
     */
    const OP_UPDATE = 0x02;
    /**
     * The delete operation. This is mainly used when overriding [[transactions()]] to specify which operations are transactional.
     */
    const OP_DELETE = 0x04;
    /**
     * All three operations: insert, update, delete.
     * This is a shortcut of the expression: OP_INSERT | OP_UPDATE | OP_DELETE.
     */
    const OP_ALL = 0x07;

    /**
     * @var string current snippet value for this Active Record instance.
     * It will be filled up automatically when instance found using [[Query::snippetCallback]]
     * or [[ActiveQuery::snippetByModel()]].
     */
    private $_snippet;

    /**
     * Returns the Sphinx connection used by this AR class.
     * By default, the "sphinx" application component is used as the Sphinx connection.
     * You may override this method if you want to use a different Sphinx connection.
     * @return Connection the Sphinx connection used by this AR class.
     */
    public static function getDb()
    {
65
        return \Yii::$app->get('sphinx');
66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
    }

    /**
     * Creates an [[ActiveQuery]] instance with a given SQL statement.
     *
     * Note that because the SQL statement is already specified, calling additional
     * query modification methods (such as `where()`, `order()`) on the created [[ActiveQuery]]
     * instance will have no effect. However, calling `with()`, `asArray()` or `indexBy()` is
     * still fine.
     *
     * Below is an example:
     *
     * ~~~
     * $customers = Article::findBySql("SELECT * FROM `idx_article` WHERE MATCH('development')")->all();
     * ~~~
     *
82 83
     * @param string $sql the SQL statement to be executed
     * @param array $params parameters to be bound to the SQL statement during execution.
84 85 86 87
     * @return ActiveQuery the newly created [[ActiveQuery]] instance
     */
    public static function findBySql($sql, $params = [])
    {
Alexander Makarov committed
88
        $query = static::find();
89 90 91 92 93 94 95 96 97 98 99 100 101
        $query->sql = $sql;

        return $query->params($params);
    }

    /**
     * Updates the whole table using the provided attribute values and conditions.
     * For example, to change the status to be 1 for all articles which status is 2:
     *
     * ~~~
     * Article::updateAll(['status' => 1], 'status = 2');
     * ~~~
     *
102 103 104 105 106
     * @param array $attributes attribute values (name-value pairs) to be saved into the table
     * @param string|array $condition the conditions that will be put in the WHERE part of the UPDATE SQL.
     * Please refer to [[Query::where()]] on how to specify this parameter.
     * @param array $params the parameters (name => value) to be bound to the query.
     * @return integer the number of rows updated
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
     */
    public static function updateAll($attributes, $condition = '', $params = [])
    {
        $command = static::getDb()->createCommand();
        $command->update(static::indexName(), $attributes, $condition, $params);

        return $command->execute();
    }

    /**
     * Deletes rows in the index using the provided conditions.
     *
     * For example, to delete all articles whose status is 3:
     *
     * ~~~
     * Article::deleteAll('status = 3');
     * ~~~
     *
125 126 127 128
     * @param string|array $condition the conditions that will be put in the WHERE part of the DELETE SQL.
     * Please refer to [[Query::where()]] on how to specify this parameter.
     * @param array $params the parameters (name => value) to be bound to the query.
     * @return integer the number of rows deleted
129 130 131 132 133 134 135 136 137 138
     */
    public static function deleteAll($condition = '', $params = [])
    {
        $command = static::getDb()->createCommand();
        $command->delete(static::indexName(), $condition, $params);

        return $command->execute();
    }

    /**
Alexander Makarov committed
139
     * @inheritdoc
140
     */
Alexander Makarov committed
141
    public static function find()
142
    {
143
        return Yii::createObject(ActiveQuery::className(), [get_called_class()]);
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
    }

    /**
     * Declares the name of the Sphinx index associated with this AR class.
     * By default this method returns the class name as the index name by calling [[Inflector::camel2id()]].
     * For example, 'Article' becomes 'article', and 'StockItem' becomes
     * 'stock_item'. You may override this method if the index is not named after this convention.
     * @return string the index name
     */
    public static function indexName()
    {
        return Inflector::camel2id(StringHelper::basename(get_called_class()), '_');
    }

    /**
     * Returns the schema information of the Sphinx index associated with this AR class.
160
     * @return IndexSchema the schema information of the Sphinx index associated with this AR class.
161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
     * @throws InvalidConfigException if the index for the AR class does not exist.
     */
    public static function getIndexSchema()
    {
        $schema = static::getDb()->getIndexSchema(static::indexName());
        if ($schema !== null) {
            return $schema;
        } else {
            throw new InvalidConfigException("The index does not exist: " . static::indexName());
        }
    }

    /**
     * Returns the primary key name for this AR class.
     * The default implementation will return the primary key as declared
     * in the Sphinx index, which is associated with this AR class.
     *
     * Note that an array should be returned even for a table with single primary key.
     *
     * @return string[] the primary keys of the associated Sphinx index.
     */
    public static function primaryKey()
    {
        return [static::getIndexSchema()->primaryKey];
    }

    /**
     * Builds a snippet from provided data and query, using specified index settings.
189 190 191 192
     * @param string|array $source is the source data to extract a snippet from.
     * It could be either a single string or array of strings.
     * @param string $match the full-text query to build snippets for.
     * @param array $options list of options in format: optionName => optionValue
193
     * @return string|array built snippet in case "source" is a string, list of built snippets
194
     * in case "source" is an array.
195 196 197 198 199 200 201 202 203 204 205 206 207 208
     */
    public static function callSnippets($source, $match, $options = [])
    {
        $command = static::getDb()->createCommand();
        $command->callSnippets(static::indexName(), $source, $match, $options);
        if (is_array($source)) {
            return $command->queryColumn();
        } else {
            return $command->queryScalar();
        }
    }

    /**
     * Returns tokenized and normalized forms of the keywords, and, optionally, keyword statistics.
209 210 211
     * @param string $text the text to break down to keywords.
     * @param boolean $fetchStatistic whether to return document and hit occurrence statistics
     * @return array keywords and statistics
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230
     */
    public static function callKeywords($text, $fetchStatistic = false)
    {
        $command = static::getDb()->createCommand();
        $command->callKeywords(static::indexName(), $text, $fetchStatistic);

        return $command->queryAll();
    }

    /**
     * @param string $snippet
     */
    public function setSnippet($snippet)
    {
        $this->_snippet = $snippet;
    }

    /**
     * Returns current snippet value or generates new one from given match.
231 232
     * @param string $match snippet source query
     * @param array $options list of options in format: optionName => optionValue
233 234 235 236 237 238 239 240 241 242 243 244 245
     * @return string snippet value
     */
    public function getSnippet($match = null, $options = [])
    {
        if ($match !== null) {
            $this->_snippet = $this->fetchSnippet($match, $options);
        }

        return $this->_snippet;
    }

    /**
     * Builds up the snippet value from the given query.
246 247
     * @param string $match the full-text query to build snippets for.
     * @param array $options list of options in format: optionName => optionValue
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265
     * @return string snippet value.
     */
    protected function fetchSnippet($match, $options = [])
    {
        return static::callSnippets($this->getSnippetSource(), $match, $options);
    }

    /**
     * Returns the string, which should be used as a source to create snippet for this
     * Active Record instance.
     * Child classes must implement this method to return the actual snippet source text.
     * For example:
     * ~~~
     * public function getSnippetSource()
     * {
     *     return $this->snippetSourceRelation->content;
     * }
     * ~~~
266
     * @return string snippet source string.
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298
     * @throws \yii\base\NotSupportedException if this is not supported by the Active Record class
     */
    public function getSnippetSource()
    {
        throw new NotSupportedException($this->className() . ' does not provide snippet source.');
    }

    /**
     * Declares which operations should be performed within a transaction in different scenarios.
     * The supported DB operations are: [[OP_INSERT]], [[OP_UPDATE]] and [[OP_DELETE]],
     * which correspond to the [[insert()]], [[update()]] and [[delete()]] methods, respectively.
     * By default, these methods are NOT enclosed in a transaction.
     *
     * In some scenarios, to ensure data consistency, you may want to enclose some or all of them
     * in transactions. You can do so by overriding this method and returning the operations
     * that need to be transactional. For example,
     *
     * ~~~
     * return [
     *     'admin' => self::OP_INSERT,
     *     'api' => self::OP_INSERT | self::OP_UPDATE | self::OP_DELETE,
     *     // the above is equivalent to the following:
     *     // 'api' => self::OP_ALL,
     *
     * ];
     * ~~~
     *
     * The above declaration specifies that in the "admin" scenario, the insert operation ([[insert()]])
     * should be done in a transaction; and in the "api" scenario, all the operations should be done
     * in a transaction.
     *
     * @return array the declarations of transactional operations. The array keys are scenarios names,
299
     * and the array values are the corresponding transaction operations.
300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
     */
    public function transactions()
    {
        return [];
    }

    /**
     * Returns the list of all attribute names of the model.
     * The default implementation will return all column names of the table associated with this AR class.
     * @return array list of attribute names.
     */
    public function attributes()
    {
        return array_keys(static::getIndexSchema()->columns);
    }

    /**
     * Inserts a row into the associated Sphinx index using the attribute values of this record.
     *
     * This method performs the following steps in order:
     *
     * 1. call [[beforeValidate()]] when `$runValidation` is true. If validation
     *    fails, it will skip the rest of the steps;
     * 2. call [[afterValidate()]] when `$runValidation` is true.
     * 3. call [[beforeSave()]]. If the method returns false, it will skip the
     *    rest of the steps;
     * 4. insert the record into index. If this fails, it will skip the rest of the steps;
     * 5. call [[afterSave()]];
     *
     * In the above step 1, 2, 3 and 5, events [[EVENT_BEFORE_VALIDATE]],
     * [[EVENT_BEFORE_INSERT]], [[EVENT_AFTER_INSERT]] and [[EVENT_AFTER_VALIDATE]]
     * will be raised by the corresponding methods.
     *
     * Only the [[changedAttributes|changed attribute values]] will be inserted.
     *
     * For example, to insert an article record:
     *
     * ~~~
     * $article = new Article;
     * $article->id = $id;
     * $article->genre_id = $genreId;
     * $article->content = $content;
     * $article->insert();
     * ~~~
     *
345 346 347 348 349
     * @param boolean $runValidation whether to perform validation before saving the record.
     * If the validation fails, the record will not be inserted.
     * @param array $attributes list of attributes that need to be saved. Defaults to null,
     * meaning all attributes that are loaded from index will be saved.
     * @return boolean whether the attributes are valid and the record is inserted successfully.
350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
     * @throws \Exception in case insert failed.
     */
    public function insert($runValidation = true, $attributes = null)
    {
        if ($runValidation && !$this->validate($attributes)) {
            return false;
        }
        $db = static::getDb();
        if ($this->isTransactional(self::OP_INSERT) && $db->getTransaction() === null) {
            $transaction = $db->beginTransaction();
            try {
                $result = $this->insertInternal($attributes);
                if ($result === false) {
                    $transaction->rollBack();
                } else {
                    $transaction->commit();
                }
            } catch (\Exception $e) {
                $transaction->rollBack();
                throw $e;
            }
        } else {
            $result = $this->insertInternal($attributes);
        }

        return $result;
    }

    /**
     * @see ActiveRecord::insert()
     */
    private function insertInternal($attributes = null)
    {
        if (!$this->beforeSave(true)) {
            return false;
        }
        $values = $this->getDirtyAttributes($attributes);
        if (empty($values)) {
            foreach ($this->getPrimaryKey(true) as $key => $value) {
                $values[$key] = $value;
            }
        }
        $db = static::getDb();
        $command = $db->createCommand()->insert($this->indexName(), $values);
        if (!$command->execute()) {
            return false;
        }
397

Carsten Brandt committed
398 399 400
        $changedAttributes = array_fill_keys(array_keys($values), null);
        $this->setOldAttributes($values);
        $this->afterSave(true, $changedAttributes);
401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426

        return true;
    }

    /**
     * Saves the changes to this active record into the associated Sphinx index.
     *
     * This method performs the following steps in order:
     *
     * 1. call [[beforeValidate()]] when `$runValidation` is true. If validation
     *    fails, it will skip the rest of the steps;
     * 2. call [[afterValidate()]] when `$runValidation` is true.
     * 3. call [[beforeSave()]]. If the method returns false, it will skip the
     *    rest of the steps;
     * 4. save the record into index. If this fails, it will skip the rest of the steps;
     * 5. call [[afterSave()]];
     *
     * In the above step 1, 2, 3 and 5, events [[EVENT_BEFORE_VALIDATE]],
     * [[EVENT_BEFORE_UPDATE]], [[EVENT_AFTER_UPDATE]] and [[EVENT_AFTER_VALIDATE]]
     * will be raised by the corresponding methods.
     *
     * Only the [[changedAttributes|changed attribute values]] will be saved into database.
     *
     * For example, to update an article record:
     *
     * ~~~
427
     * $article = Article::findOne($id);
428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
     * $article->genre_id = $genreId;
     * $article->group_id = $groupId;
     * $article->update();
     * ~~~
     *
     * Note that it is possible the update does not affect any row in the table.
     * In this case, this method will return 0. For this reason, you should use the following
     * code to check if update() is successful or not:
     *
     * ~~~
     * if ($this->update() !== false) {
     *     // update successful
     * } else {
     *     // update failed
     * }
     * ~~~
     *
445 446 447 448 449 450
     * @param boolean $runValidation whether to perform validation before saving the record.
     * If the validation fails, the record will not be inserted into the database.
     * @param array $attributeNames list of attributes that need to be saved. Defaults to null,
     * meaning all attributes that are loaded from DB will be saved.
     * @return integer|boolean the number of rows affected, or false if validation fails
     * or [[beforeSave()]] stops the updating process.
451
     * @throws StaleObjectException if [[optimisticLock|optimistic locking]] is enabled and the data
452 453
     * being updated is outdated.
     * @throws \Exception in case update failed.
454
     */
455
    public function update($runValidation = true, $attributeNames = null)
456
    {
457
        if ($runValidation && !$this->validate($attributeNames)) {
458 459 460 461 462 463
            return false;
        }
        $db = static::getDb();
        if ($this->isTransactional(self::OP_UPDATE) && $db->getTransaction() === null) {
            $transaction = $db->beginTransaction();
            try {
464
                $result = $this->updateInternal($attributeNames);
465 466 467 468 469 470 471 472 473 474
                if ($result === false) {
                    $transaction->rollBack();
                } else {
                    $transaction->commit();
                }
            } catch (\Exception $e) {
                $transaction->rollBack();
                throw $e;
            }
        } else {
475
            $result = $this->updateInternal($attributeNames);
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
        }

        return $result;
    }

    /**
     * @see CActiveRecord::update()
     * @throws StaleObjectException
     */
    protected function updateInternal($attributes = null)
    {
        if (!$this->beforeSave(false)) {
            return false;
        }
        $values = $this->getDirtyAttributes($attributes);
        if (empty($values)) {
492
            $this->afterSave(false, $values);
493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533
            return 0;
        }

        // Replace is supported only by runtime indexes and necessary only for field update
        $useReplace = false;
        $indexSchema = $this->getIndexSchema();
        if ($this->getIndexSchema()->isRuntime) {
            foreach ($values as $name => $value) {
                $columnSchema = $indexSchema->getColumn($name);
                if ($columnSchema->isField) {
                    $useReplace = true;
                    break;
                }
            }
        }

        if ($useReplace) {
            $values = array_merge($values, $this->getOldPrimaryKey(true));
            $command = static::getDb()->createCommand();
            $command->replace(static::indexName(), $values);
            // We do not check the return value of replace because it's possible
            // that the REPLACE statement doesn't change anything and thus returns 0.
            $rows = $command->execute();
        } else {
            $condition = $this->getOldPrimaryKey(true);
            $lock = $this->optimisticLock();
            if ($lock !== null) {
                if (!isset($values[$lock])) {
                    $values[$lock] = $this->$lock + 1;
                }
                $condition[$lock] = $this->$lock;
            }
            // We do not check the return value of updateAll() because it's possible
            // that the UPDATE statement doesn't change anything and thus returns 0.
            $rows = $this->updateAll($values, $condition);

            if ($lock !== null && !$rows) {
                throw new StaleObjectException('The object being updated is outdated.');
            }
        }

534
        $changedAttributes = [];
535
        foreach ($values as $name => $value) {
536
            $changedAttributes[$name] = $this->getOldAttribute($name);
Carsten Brandt committed
537
            $this->setOldAttribute($name, $value);
538
        }
539
        $this->afterSave(false, $changedAttributes);
540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556

        return $rows;
    }

    /**
     * Deletes the index entry corresponding to this active record.
     *
     * This method performs the following steps in order:
     *
     * 1. call [[beforeDelete()]]. If the method returns false, it will skip the
     *    rest of the steps;
     * 2. delete the record from the index;
     * 3. call [[afterDelete()]].
     *
     * In the above step 1 and 3, events named [[EVENT_BEFORE_DELETE]] and [[EVENT_AFTER_DELETE]]
     * will be raised by the corresponding methods.
     *
557 558
     * @return integer|boolean the number of rows deleted, or false if the deletion is unsuccessful for some reason.
     * Note that it is possible the number of rows deleted is 0, even though the deletion execution is successful.
559
     * @throws StaleObjectException if [[optimisticLock|optimistic locking]] is enabled and the data
560 561
     * being deleted is outdated.
     * @throws \Exception in case delete failed.
562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604
     */
    public function delete()
    {
        $db = static::getDb();
        $transaction = $this->isTransactional(self::OP_DELETE) && $db->getTransaction() === null ? $db->beginTransaction() : null;
        try {
            $result = false;
            if ($this->beforeDelete()) {
                // we do not check the return value of deleteAll() because it's possible
                // the record is already deleted in the database and thus the method will return 0
                $condition = $this->getOldPrimaryKey(true);
                $lock = $this->optimisticLock();
                if ($lock !== null) {
                    $condition[$lock] = $this->$lock;
                }
                $result = $this->deleteAll($condition);
                if ($lock !== null && !$result) {
                    throw new StaleObjectException('The object being deleted is outdated.');
                }
                $this->setOldAttributes(null);
                $this->afterDelete();
            }
            if ($transaction !== null) {
                if ($result === false) {
                    $transaction->rollBack();
                } else {
                    $transaction->commit();
                }
            }
        } catch (\Exception $e) {
            if ($transaction !== null) {
                $transaction->rollBack();
            }
            throw $e;
        }

        return $result;
    }

    /**
     * Returns a value indicating whether the given active record is the same as the current one.
     * The comparison is made by comparing the index names and the primary key values of the two active records.
     * If one of the records [[isNewRecord|is new]] they are also considered not equal.
605 606
     * @param ActiveRecord $record record to compare to
     * @return boolean whether the two active records refer to the same row in the same index.
607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
     */
    public function equals($record)
    {
        if ($this->isNewRecord || $record->isNewRecord) {
            return false;
        }

        return $this->indexName() === $record->indexName() && $this->getPrimaryKey() === $record->getPrimaryKey();
    }

    /**
     * @inheritdoc
     */
    public static function populateRecord($record, $row)
    {
        $columns = static::getIndexSchema()->columns;
        foreach ($row as $name => $value) {
624 625 626
            if (isset($columns[$name])) {
                if ($columns[$name]->isMva) {
                    $mvaValue = explode(',', $value);
627
                    $row[$name] = array_map(array($columns[$name], 'phpTypecast'), $mvaValue);
628
                } else {
629
                    $row[$name] = $columns[$name]->phpTypecast($value);
630
                }
631 632 633 634 635 636 637
            }
        }
        parent::populateRecord($record, $row);
    }

    /**
     * Returns a value indicating whether the specified operation is transactional in the current [[scenario]].
638
     * @param integer $operation the operation to check. Possible values are [[OP_INSERT]], [[OP_UPDATE]] and [[OP_DELETE]].
639 640 641 642 643 644 645 646 647
     * @return boolean whether the specified operation is transactional in the current [[scenario]].
     */
    public function isTransactional($operation)
    {
        $scenario = $this->getScenario();
        $transactions = $this->transactions();

        return isset($transactions[$scenario]) && ($transactions[$scenario] & $operation);
    }
648
}