| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949 |
- <?php
- /**
- * @link http://www.yiiframework.com/
- * @copyright Copyright (c) 2008 Yii Software LLC
- * @license http://www.yiiframework.com/license/
- */
- namespace yii\elasticsearch;
- use Yii;
- use yii\base\InvalidArgumentException;
- use yii\base\InvalidCallException;
- use yii\base\InvalidConfigException;
- use yii\base\NotSupportedException;
- use yii\db\ActiveQueryInterface;
- use yii\db\ActiveRecordInterface;
- use yii\db\BaseActiveRecord;
- use yii\db\StaleObjectException;
- use yii\helpers\ArrayHelper;
- use yii\helpers\Inflector;
- use yii\helpers\StringHelper;
- /**
- * ActiveRecord is the base class for classes representing relational data in terms of objects.
- *
- * This class implements the ActiveRecord pattern for the fulltext search and data storage
- * [Elasticsearch](https://www.elastic.co/products/elasticsearch).
- *
- * For defining a record a subclass should at least implement the [[attributes()]] method
- * to define attributes.
- * IMPORTANT: The primary key (the `_id` attribute) MUST NOT be included in [[attributes()]].
- *
- * The following is an example model called `Customer`:
- *
- * ```php
- * class Customer extends \yii\elasticsearch\ActiveRecord
- * {
- * public function attributes()
- * {
- * return ['name', 'address', 'registration_date'];
- * }
- * }
- * ```
- *
- * You may override [[index()]] and [[type()]] to define the index and type this record represents.
- * Types are being deprecated, and it is recommended to have a single type per index. For more information
- * read about [removal of mapping types](https://www.elastic.co/guide/en/elasticsearch/reference/current/removal-of-types.html).
- * For Elasticsearch 7 and later (as configured in [[Connection]], [[type()]] is ignored.
- *
- * @property mixed $_id The primary key of the record. Can only be written to for new records, otherwise read-only.
- * @property array|null $highlight A list of arrays with highlighted excerpts indexed by field names.
- * This property is read-only.
- * @property float $score Returns the score of this record when it was retrieved via a [[find()]] query.
- * This property is read-only.
- * @property array|null $explanation An explanation for each hit on how its score was computed.
- * This property is read-only.
- *
- * @author Carsten Brandt <mail@cebe.cc>
- * @since 2.0
- */
- class ActiveRecord extends BaseActiveRecord
- {
- private $_id;
- private $_score;
- private $_version;
- private $_highlight;
- private $_explanation;
- /**
- * Returns the database connection used by this AR class.
- * By default, the "elasticsearch" application component is used as the database connection.
- * You may override this method if you want to use a different database connection.
- * @return Connection the database connection used by this AR class.
- */
- public static function getDb()
- {
- return \Yii::$app->get('elasticsearch');
- }
- /**
- * @inheritdoc
- * @return ActiveQuery the newly created [[ActiveQuery]] instance.
- */
- public static function find()
- {
- return Yii::createObject(ActiveQuery::className(), [get_called_class()]);
- }
- /**
- * @inheritdoc
- */
- public static function findOne($condition)
- {
- if (!is_array($condition)) {
- return static::get($condition);
- }
- if (!ArrayHelper::isAssociative($condition)) {
- $records = static::mget(array_values($condition));
- return empty($records) ? null : reset($records);
- }
- $condition = static::filterCondition($condition);
- return static::find()->andWhere($condition)->one();
- }
- /**
- * @inheritdoc
- */
- public static function findAll($condition)
- {
- if (!ArrayHelper::isAssociative($condition)) {
- return static::mget(is_array($condition) ? array_values($condition) : [$condition]);
- }
- $condition = static::filterCondition($condition);
- return static::find()->andWhere($condition)->all();
- }
- /**
- * Filter out condition parts that are array valued, to prevent building arbitrary conditions.
- * @param array $condition
- */
- private static function filterCondition($condition)
- {
- foreach ($condition as $k => $v) {
- if (is_array($v)) {
- $condition[$k] = array_values($v);
- foreach ($v as $vv) {
- if (is_array($vv)) {
- throw new InvalidArgumentException('Nested arrays are not allowed in condition for findAll() and findOne().');
- }
- }
- }
- }
- return $condition;
- }
- /**
- * Gets a record by its primary key.
- *
- * @param mixed $primaryKey the primaryKey value
- * @param array $options options given in this parameter are passed to Elasticsearch
- * as request URI parameters.
- * Please refer to the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-get.html)
- * for more details on these options.
- * @return static|null The record instance or null if it was not found.
- */
- public static function get($primaryKey, $options = [])
- {
- if ($primaryKey === null) {
- return null;
- }
- $command = static::getDb()->createCommand();
- $result = $command->get(static::index(), static::type(), $primaryKey, $options);
- if ($result && $result['found']) {
- $model = static::instantiate($result);
- static::populateRecord($model, $result);
- $model->afterFind();
- return $model;
- }
- return null;
- }
- /**
- * Gets a list of records by its primary keys.
- *
- * @param array $primaryKeys an array of primaryKey values
- * @param array $options options given in this parameter are passed to Elasticsearch
- * as request URI parameters.
- *
- * Please refer to the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-get.html)
- * for more details on these options.
- * @return array The record instances, or empty array if nothing was found
- */
- public static function mget(array $primaryKeys, $options = [])
- {
- if (empty($primaryKeys)) {
- return [];
- }
- if (count($primaryKeys) === 1) {
- $model = static::get(reset($primaryKeys));
- return $model === null ? [] : [$model];
- }
- $command = static::getDb()->createCommand();
- $result = $command->mget(static::index(), static::type(), $primaryKeys, $options);
- $models = [];
- foreach ($result['docs'] as $doc) {
- if ($doc['found']) {
- $model = static::instantiate($doc);
- static::populateRecord($model, $doc);
- $model->afterFind();
- $models[] = $model;
- }
- }
- return $models;
- }
- // TODO add more like this feature https://www.elastic.co/guide/en/elasticsearch/reference/current/search-more-like-this.html
- // TODO add percolate functionality https://www.elastic.co/guide/en/elasticsearch/reference/current/search-percolate.html
- // TODO implement copy and move as pk change is not possible
- /**
- * @return float returns the score of this record when it was retrieved via a [[find()]] query.
- */
- public function getScore()
- {
- return $this->_score;
- }
- /**
- * @return array|null A list of arrays with highlighted excerpts indexed by field names.
- */
- public function getHighlight()
- {
- return $this->_highlight;
- }
- /**
- * @return array|null An explanation for each hit on how its score was computed.
- * @since 2.0.5
- */
- public function getExplanation()
- {
- return $this->_explanation;
- }
- /**
- * Alias to [[get_id()]]. Returns the primary key value.
- * @param bool $asArray
- * @return mixed
- * @deprecated since 2.1.0
- */
- public function getPrimaryKey($asArray = false)
- {
- $pk = static::primaryKey()[0];
- if ($asArray) {
- return [$pk => $this->$pk];
- } else {
- return $this->$pk;
- }
- }
- /**
- * Alias to [[set_id()]]. Sets the primary key value.
- * @param mixed $value
- * @throws \yii\base\InvalidCallException when record is not new
- * @deprecated since 2.1.0
- */
- public function setPrimaryKey($value)
- {
- $pk = static::primaryKey()[0];
- $this->$pk = $value;
- }
- /**
- * Sets the `_id` attribute that holds the primary key (for compatibility with relations)
- * @param mixed $value
- * @throws \yii\base\InvalidCallException when record is not new
- */
- public function set_id($value)
- {
- $pk = static::primaryKey()[0];
- if ($this->getIsNewRecord()) {
- $this->$pk = $value;
- } else {
- throw new InvalidCallException('Changing the primaryKey of an already saved record is not allowed.');
- }
- }
- /**
- * Returns the `_id` attribute that holds the primary key (for compatibility with relations)
- * @return mixed
- */
- public function get_id()
- {
- $pk = static::primaryKey()[0];
- return $this->$pk;
- }
- /**
- * @inheritdoc
- */
- public function getOldPrimaryKey($asArray = false)
- {
- $pk = static::primaryKey()[0];
- if ($this->getIsNewRecord()) {
- $id = null;
- } else {
- $id = $this->_id;
- }
- if ($asArray) {
- return [$pk => $id];
- } else {
- return $id;
- }
- }
- /**
- * This method defines the attribute that uniquely identifies a record.
- * The name of the primary key attribute is `_id`, and can not be changed.
- *
- * Elasticsearch does not support composite primary keys in the traditional sense. However to match the signature
- * of the [[\yii\db\ActiveRecordInterface|ActiveRecordInterface]] this methods returns an array instead of a
- * single string.
- *
- * @return string[] array of primary key attributes. Only the first element of the array will be used.
- */
- final public static function primaryKey()
- {
- return ['_id'];
- }
- /**
- * Returns the list of all attribute names of the model.
- *
- * This method must be overridden by child classes to define available attributes.
- * IMPORTANT: The primary key (the `_id` attribute) MUST NOT be included in [[attributes()]].
- *
- * Attributes are names of fields of the corresponding Elasticsearch document.
- *
- * @return string[] list of attribute names.
- * @throws \yii\base\InvalidConfigException if not overridden in a child class.
- */
- public function attributes()
- {
- throw new InvalidConfigException('The attributes() method of Elasticsearch ActiveRecord has to be implemented by child classes.');
- }
- /**
- * A list of attributes that should be treated as array valued when retrieved through [[ActiveQuery::fields]].
- *
- * If not listed by this method, attributes retrieved through [[ActiveQuery::fields]] will converted to a scalar value
- * when the result array contains only one value.
- *
- * @return string[] list of attribute names. Must be a subset of [[attributes()]].
- */
- public function arrayAttributes()
- {
- return [];
- }
- /**
- * @return string the name of the index this record is stored in.
- */
- public static function index()
- {
- return Inflector::pluralize(Inflector::camel2id(StringHelper::basename(get_called_class()), '-'));
- }
- /**
- * Returns the name of the type of this record.
- * IMPORTANT: For Elasticsearch 7 and later, [[type()]] is ignored.
- * @return string the name of the type of this record.
- */
- public static function type()
- {
- return Inflector::camel2id(StringHelper::basename(get_called_class()), '-');
- }
- /**
- * @inheritdoc
- *
- * @param ActiveRecord $record the record to be populated. In most cases this will be an instance
- * created by [[instantiate()]] beforehand.
- * @param array $row attribute values (name => value)
- */
- public static function populateRecord($record, $row)
- {
- $attributes = [];
- if (isset($row['_source'])) {
- $attributes = $row['_source'];
- }
- if (isset($row['fields'])) {
- // reset fields in case it is scalar value
- $arrayAttributes = $record->arrayAttributes();
- foreach ($row['fields'] as $key => $value) {
- if (!isset($arrayAttributes[$key]) && count($value) === 1) {
- $row['fields'][$key] = reset($value);
- }
- }
- $attributes = array_merge($attributes, $row['fields']);
- }
- parent::populateRecord($record, $attributes);
- $pk = static::primaryKey()[0];
- $record->_id = $row[$pk];
- $record->_highlight = isset($row['highlight']) ? $row['highlight'] : null;
- $record->_score = isset($row['_score']) ? $row['_score'] : null;
- $record->_version = isset($row['_version']) ? $row['_version'] : null; // TODO version should always be available...
- $record->_explanation = isset($row['_explanation']) ? $row['_explanation'] : null;
- }
- /**
- * Creates an active record instance.
- *
- * This method is called together with [[populateRecord()]] by [[ActiveQuery]].
- * It is not meant to be used for creating new records directly.
- *
- * You may override this method if the instance being created
- * depends on the row data to be populated into the record.
- * For example, by creating a record based on the value of a column,
- * you may implement the so-called single-table inheritance mapping.
- * @param array $row row data to be populated into the record.
- * This array consists of the following keys:
- * - `_source`: refers to the attributes of the record.
- * - `_type`: the type this record is stored in.
- * - `_index`: the index this record is stored in.
- * @return static the newly created active record
- */
- public static function instantiate($row)
- {
- return new static;
- }
- /**
- * Inserts a document into the associated 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 database. 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 [[dirtyAttributes|changed attribute values]] will be inserted into database.
- *
- * If the [[primaryKey|primary key]] is not set (null) during insertion,
- * it will be populated with a randomly generated value after insertion.
- *
- * For example, to insert a customer record:
- *
- * ~~~
- * $customer = new Customer;
- * $customer->name = $name;
- * $customer->email = $email;
- * $customer->insert();
- * ~~~
- *
- * @param bool $runValidation whether to perform validation before saving the record.
- * If the validation fails, the record will not be inserted into the database.
- * @param array $attributes list of attributes that need to be saved. Defaults to null,
- * meaning all attributes will be saved.
- * @param array $options options given in this parameter are passed to Elasticsearch
- * as request URI parameters. These are among others:
- *
- * - `routing` define shard placement of this record.
- * - `parent` by giving the primaryKey of another record this defines a parent-child relation
- *
- * Please refer to the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-index_.html)
- * for more details on these options.
- *
- * By default the `op_type` is set to `create` if model primary key is present.
- * @return bool whether the attributes are valid and the record is inserted successfully.
- */
- public function insert($runValidation = true, $attributes = null, $options = [ ])
- {
- if ($runValidation && !$this->validate($attributes)) {
- return false;
- }
- if (!$this->beforeSave(true)) {
- return false;
- }
- $values = $this->getDirtyAttributes($attributes);
- if ($this->getPrimaryKey() !== null) {
- $options['op_type'] = isset($options['op_type']) ? $options['op_type'] : 'create';
- }
- $response = static::getDb()->createCommand()->insert(
- static::index(),
- static::type(),
- $values,
- $this->getPrimaryKey(),
- $options
- );
- if ($response === false) {
- return false;
- }
- $pk = static::primaryKey()[0];
- $this->$pk = $response['_id'];
- if ($pk != '_id') {
- $values[$pk] = $response['_id'];
- }
- $this->_version = $response['_version'];
- $this->_score = null;
- $changedAttributes = array_fill_keys(array_keys($values), null);
- $this->setOldAttributes($values);
- $this->afterSave(true, $changedAttributes);
- return true;
- }
- /**
- * @inheritdoc
- *
- * @param bool $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 attribute names that need to be saved. Defaults to null,
- * meaning all attributes that are loaded from DB will be saved.
- * @param array $options options given in this parameter are passed to Elasticsearch
- * as request URI parameters. These are among others:
- *
- * - `routing` define shard placement of this record.
- * - `parent` by giving the primaryKey of another record this defines a parent-child relation
- * - `timeout` timeout waiting for a shard to become available.
- * - `replication` the replication type for the delete/index operation (sync or async).
- * - `consistency` the write consistency of the index/delete operation.
- * - `refresh` refresh the relevant primary and replica shards (not the whole index) immediately after the operation occurs, so that the updated document appears in search results immediately.
- * - `detect_noop` this parameter will become part of the request body and will prevent the index from getting updated when nothing has changed.
- *
- * Please refer to the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-update.html#docs-update-api-query-params)
- * for more details on these options.
- *
- * The following parameters are Yii specific:
- *
- * - `optimistic_locking` set this to `true` to enable optimistic locking, avoid updating when the record has changed since it
- * has been loaded from the database. Yii will set the `version` parameter to the value stored in [[version]].
- * See the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/guide/current/optimistic-concurrency-control.html) for details.
- *
- * Make sure the record has been fetched with a [[version]] before. This is only the case
- * for records fetched via [[get()]] and [[mget()]] by default. For normal queries, the `_version` field has to be fetched explicitly.
- *
- * @return int|bool the number of rows affected, or false if validation fails
- * or [[beforeSave()]] stops the updating process.
- * @throws StaleObjectException if optimistic locking is enabled and the data being updated is outdated.
- * @throws InvalidParamException if no [[version]] is available and optimistic locking is enabled.
- * @throws Exception in case update failed.
- */
- public function update($runValidation = true, $attributeNames = null, $options = [])
- {
- if ($runValidation && !$this->validate($attributeNames)) {
- return false;
- }
- return $this->updateInternal($attributeNames, $options);
- }
- /**
- * @param array $attributes attributes to update
- * @param array $options options given in this parameter are passed to Elasticsearch
- * as request URI parameters. See [[update()]] for details.
- * @return int|false the number of rows affected, or false if [[beforeSave()]] stops the updating process.
- * @throws StaleObjectException if optimistic locking is enabled and the data being updated is outdated.
- * @throws InvalidParamException if no [[version]] is available and optimistic locking is enabled.
- * @throws Exception in case update failed.
- * @see update()
- */
- protected function updateInternal($attributes = null, $options = [])
- {
- if (!$this->beforeSave(false)) {
- return false;
- }
- $values = $this->getDirtyAttributes($attributes);
- if (empty($values)) {
- $this->afterSave(false, $values);
- return 0;
- }
- if (isset($options['optimistic_locking']) && $options['optimistic_locking']) {
- if ($this->_version === null) {
- throw new InvalidArgumentException('Unable to use optimistic locking on a record that has no version set. Refer to the docs of ActiveRecord::update() for details.');
- }
- $options['version'] = $this->_version;
- unset($options['optimistic_locking']);
- }
- try {
- $result = static::getDb()->createCommand()->update(
- static::index(),
- static::type(),
- $this->getOldPrimaryKey(false),
- $values,
- $options
- );
- } catch (Exception $e) {
- // HTTP 409 is the response in case of failed optimistic locking
- // https://www.elastic.co/guide/en/elasticsearch/guide/current/optimistic-concurrency-control.html
- if (isset($e->errorInfo['responseCode']) && $e->errorInfo['responseCode'] == 409) {
- throw new StaleObjectException('The object being updated is outdated.', $e->errorInfo, $e->getCode(), $e);
- }
- throw $e;
- }
- if (is_array($result) && isset($result['_version'])) {
- $this->_version = $result['_version'];
- }
- $changedAttributes = [];
- foreach ($values as $name => $value) {
- $changedAttributes[$name] = $this->getOldAttribute($name);
- $this->setOldAttribute($name, $value);
- }
- $this->afterSave(false, $changedAttributes);
- if ($result === false) {
- return 0;
- } else {
- return 1;
- }
- }
- /**
- * Performs a quick and highly efficient scroll/scan query to get the list of primary keys that
- * satisfy the given condition. If condition is a list of primary keys
- * (e.g.: `['_id' => ['1', '2', '3']]`), the query is not performed for performance considerations.
- * @param array $condition please refer to [[ActiveQuery::where()]] on how to specify this parameter
- * @return array primary keys that correspond to given conditions
- * @see updateAll()
- * @see updateAllCounters()
- * @see deleteAll()
- * @since 2.0.4
- */
- protected static function primaryKeysByCondition($condition)
- {
- $pkName = static::primaryKey()[0];
- if (count($condition) == 1 && isset($condition[$pkName])) {
- $primaryKeys = (array)$condition[$pkName];
- } else {
- //fetch only document metadata (no fields), 1000 documents per shard
- $query = static::find()->where($condition)->asArray()->source(false)->limit(1000);
- $primaryKeys = [];
- foreach ($query->each('1m') as $document) {
- $primaryKeys[] = $document['_id'];
- }
- }
- return $primaryKeys;
- }
- /**
- * Updates all records that match a certain condition.
- * For example, to change the status to be 1 for all customers whose status is 2:
- *
- * ~~~
- * Customer::updateAll(['status' => 1], ['status' => 2]);
- * ~~~
- *
- * @param array $attributes attribute values (name-value pairs) to be saved into the table
- * @param array $condition the conditions that will be passed to the `where()` method when building the query.
- * Please refer to [[ActiveQuery::where()]] on how to specify this parameter.
- * @return int the number of rows updated
- * @throws Exception on error.
- * @see [[ActiveRecord::primaryKeysByCondition()]]
- */
- public static function updateAll($attributes, $condition = [])
- {
- $primaryKeys = static::primaryKeysByCondition($condition);
- if (empty($primaryKeys)) {
- return 0;
- }
- $bulkCommand = static::getDb()->createBulkCommand([
- "index" => static::index(),
- "type" => static::type(),
- ]);
- foreach ($primaryKeys as $pk) {
- $bulkCommand->addAction(["update" => ["_id" => $pk]], ["doc" => $attributes]);
- }
- $response = $bulkCommand->execute();
- $n = 0;
- $errors = [];
- foreach ($response['items'] as $item) {
- if (isset($item['update']['status']) && $item['update']['status'] == 200) {
- $n++;
- } else {
- $errors[] = $item['update'];
- }
- }
- if (!empty($errors) || isset($response['errors']) && $response['errors']) {
- throw new Exception(__METHOD__ . ' failed updating records.', $errors);
- }
- return $n;
- }
- /**
- * Updates all matching records using the provided counter changes and conditions.
- * For example, to add 1 to age of all customers whose status is 2,
- *
- * ~~~
- * Customer::updateAllCounters(['age' => 1], ['status' => 2]);
- * ~~~
- *
- * @param array $counters the counters to be updated (attribute name => increment value).
- * Use negative values if you want to decrement the counters.
- * @param array $condition the conditions that will be passed to the `where()` method when building the query.
- * Please refer to [[ActiveQuery::where()]] on how to specify this parameter.
- * @return int the number of rows updated
- * @throws Exception on error.
- * @see [[ActiveRecord::primaryKeysByCondition()]]
- */
- public static function updateAllCounters($counters, $condition = [])
- {
- $primaryKeys = static::primaryKeysByCondition($condition);
- if (empty($primaryKeys) || empty($counters)) {
- return 0;
- }
- $bulkCommand = static::getDb()->createBulkCommand([
- "index" => static::index(),
- "type" => static::type(),
- ]);
- foreach ($primaryKeys as $pk) {
- $script = '';
- foreach ($counters as $counter => $value) {
- $script .= "ctx._source.{$counter} += params.{$counter};\n";
- }
- $bulkCommand->addAction(["update" => ["_id" => $pk]], [
- 'script' => [
- 'inline' => $script,
- 'params' => $counters,
- 'lang' => 'painless',
- ],
- ]);
- }
- $response = $bulkCommand->execute();
- $n = 0;
- $errors = [];
- foreach ($response['items'] as $item) {
- if (isset($item['update']['status']) && $item['update']['status'] == 200) {
- $n++;
- } else {
- $errors[] = $item['update'];
- }
- }
- if (!empty($errors) || isset($response['errors']) && $response['errors']) {
- throw new Exception(__METHOD__ . ' failed updating records counters.', $errors);
- }
- return $n;
- }
- /**
- * @inheritdoc
- *
- * @param array $options options given in this parameter are passed to Elasticsearch
- * as request URI parameters. These are among others:
- *
- * - `routing` define shard placement of this record.
- * - `parent` by giving the primaryKey of another record this defines a parent-child relation
- * - `timeout` timeout waiting for a shard to become available.
- * - `replication` the replication type for the delete/index operation (sync or async).
- * - `consistency` the write consistency of the index/delete operation.
- * - `refresh` refresh the relevant primary and replica shards (not the whole index) immediately after the operation occurs, so that the updated document appears in search results immediately.
- *
- * Please refer to the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-delete.html)
- * for more details on these options.
- *
- * The following parameters are Yii specific:
- *
- * - `optimistic_locking` set this to `true` to enable optimistic locking, avoid updating when the record has changed since it
- * has been loaded from the database. Yii will set the `version` parameter to the value stored in [[version]].
- * See the [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-delete.html#delete-versioning) for details.
- *
- * Make sure the record has been fetched with a [[version]] before. This is only the case
- * for records fetched via [[get()]] and [[mget()]] by default. For normal queries, the `_version` field has to be fetched explicitly.
- *
- * @return int|bool 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.
- * @throws StaleObjectException if optimistic locking is enabled and the data being deleted is outdated.
- * @throws Exception in case delete failed.
- */
- public function delete($options = [])
- {
- if (!$this->beforeDelete()) {
- return false;
- }
- if (isset($options['optimistic_locking']) && $options['optimistic_locking']) {
- if ($this->_version === null) {
- throw new InvalidArgumentException('Unable to use optimistic locking on a record that has no version set. Refer to the docs of ActiveRecord::delete() for details.');
- }
- $options['version'] = $this->_version;
- unset($options['optimistic_locking']);
- }
- try {
- $result = static::getDb()->createCommand()->delete(
- static::index(),
- static::type(),
- $this->getOldPrimaryKey(false),
- $options
- );
- } catch (Exception $e) {
- // HTTP 409 is the response in case of failed optimistic locking
- // https://www.elastic.co/guide/en/elasticsearch/guide/current/optimistic-concurrency-control.html
- if (isset($e->errorInfo['responseCode']) && $e->errorInfo['responseCode'] == 409) {
- throw new StaleObjectException('The object being deleted is outdated.', $e->errorInfo, $e->getCode(), $e);
- }
- throw $e;
- }
- $this->setOldAttributes(null);
- $this->afterDelete();
- if ($result === false) {
- return 0;
- } else {
- return 1;
- }
- }
- /**
- * Deletes rows in the table using the provided conditions.
- * WARNING: If you do not specify any condition, this method will delete ALL rows in the table.
- *
- * For example, to delete all customers whose status is 3:
- *
- * ~~~
- * Customer::deleteAll(['status' => 3]);
- * ~~~
- *
- * @param array $condition the conditions that will be passed to the `where()` method when building the query.
- * Please refer to [[ActiveQuery::where()]] on how to specify this parameter.
- * @return int the number of rows deleted
- * @throws Exception on error.
- * @see [[ActiveRecord::primaryKeysByCondition()]]
- */
- public static function deleteAll($condition = [])
- {
- $primaryKeys = static::primaryKeysByCondition($condition);
- if (empty($primaryKeys)) {
- return 0;
- }
- $bulkCommand = static::getDb()->createBulkCommand([
- "index" => static::index(),
- "type" => static::type(),
- ]);
- foreach ($primaryKeys as $pk) {
- $bulkCommand->addDeleteAction($pk);
- }
- $response = $bulkCommand->execute();
- $n = 0;
- $errors = [];
- foreach ($response['items'] as $item) {
- if (isset($item['delete']['status']) && $item['delete']['status'] == 200) {
- if (isset($item['delete']['found']) && $item['delete']['found']) {
- # ES5 uses "found"
- $n++;
- } elseif (isset($item['delete']['result']) && $item['delete']['result'] == "deleted") {
- # ES6 uses "result"
- $n++;
- }
- } else {
- $errors[] = $item['delete'];
- }
- }
- if (!empty($errors) || isset($response['errors']) && $response['errors']) {
- throw new Exception(__METHOD__ . ' failed deleting records.', $errors);
- }
- return $n;
- }
- /**
- * This method has no effect in Elasticsearch ActiveRecord.
- *
- * Elasticsearch ActiveRecord uses [native Optimistic locking](https://www.elastic.co/guide/en/elasticsearch/guide/current/optimistic-concurrency-control.html).
- * See [[update()]] for more details.
- */
- public function optimisticLock()
- {
- return null;
- }
- /**
- * Destroys the relationship in current model.
- *
- * This method is not supported by Elasticsearch.
- */
- public function unlinkAll($name, $delete = false)
- {
- throw new NotSupportedException('unlinkAll() is not supported by Elasticsearch, use unlink() instead.');
- }
- public function link($name, $model, $extraColumns = [])
- {
- $relation = $this->getRelation($name);
- if ($relation->via === null) {
- $this->validateViaRelationLink($model, $relation);
- }
- parent::link($name, $model, $extraColumns);
- }
- /**
- * Validates model so that it does not contain array as its keys while linking.
- *
- * @param ActiveRecordInterface $model the model to be linked with the current one.
- * @param ActiveQueryInterface|ActiveQuery the relational query object.
- */
- protected function validateViaRelationLink($model, $relation)
- {
- $p1 = $model->isPrimaryKey(array_keys($relation->link));
- $p2 = static::isPrimaryKey(array_values($relation->link));
- $atLeastOneExists = !$this->getIsNewRecord() || !$model->getIsNewRecord();
- $foreign = null;
- $link = null;
- if ($p1 && $p2 && $atLeastOneExists) {
- if ($this->getIsNewRecord()) {
- $foreign = $this;
- $link = array_flip($relation->link);
- } else {
- $foreign = $model;
- $link = $relation->link;
- }
- } elseif ($p1) {
- $foreign = $this;
- $link = array_flip($relation->link);
- } elseif ($p2) {
- $foreign = $model;
- $link = $relation->link;
- }
- if ($foreign && $link) {
- foreach ($link as $fk => $pk) {
- if (is_array($foreign->{$fk})) {
- throw new InvalidCallException('Unable to link models: foreign model cannot be linked if its property is an array.');
- }
- }
- }
- }
- }
|