You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
3.1 KiB
104 lines
3.1 KiB
<?php |
|
/** |
|
* @link http://www.yiiframework.com/ |
|
* @copyright Copyright (c) 2008 Yii Software LLC |
|
* @license http://www.yiiframework.com/license/ |
|
*/ |
|
|
|
namespace yii\behaviors; |
|
|
|
use yii\base\Behavior; |
|
use yii\base\Event; |
|
use yii\db\Expression; |
|
use yii\db\ActiveRecord; |
|
|
|
/** |
|
* AutoTimestamp will automatically fill the attributes about creation time and updating time. |
|
* |
|
* AutoTimestamp fills the attributes when the associated AR model is being inserted or updated. |
|
* You may specify an AR to use this behavior like the following: |
|
* |
|
* ~~~ |
|
* public function behaviors() |
|
* { |
|
* return array( |
|
* 'timestamp' => array( |
|
* 'class' => 'yii\behaviors\AutoTimestamp', |
|
* ), |
|
* ); |
|
* } |
|
* ~~~ |
|
* |
|
* By default, AutoTimestamp will fill the `create_time` attribute with the current timestamp |
|
* when the associated AR object is being inserted; it will fill the `update_time` attribute |
|
* with the timestamp when the AR object is being updated. |
|
* |
|
* @author Qiang Xue <qiang.xue@gmail.com> |
|
* @since 2.0 |
|
*/ |
|
class AutoTimestamp extends Behavior |
|
{ |
|
/** |
|
* @var array list of attributes that are to be automatically filled with timestamps. |
|
* The array keys are the ActiveRecord events upon which the attributes are to be filled with timestamps, |
|
* and the array values are the corresponding attribute to be updated. You can use a string to represent |
|
* a single attribute, or an array to represent a list of attributes. |
|
* The default setting is to update the `create_time` attribute upon AR insertion, |
|
* and update the `update_time` attribute upon AR updating. |
|
*/ |
|
public $attributes = array( |
|
ActiveRecord::EVENT_BEFORE_INSERT => 'create_time', |
|
ActiveRecord::EVENT_BEFORE_UPDATE => 'update_time', |
|
); |
|
/** |
|
* @var \Closure|Expression The expression that will be used for generating the timestamp. |
|
* This can be either an anonymous function that returns the timestamp value, |
|
* or an [[Expression]] object representing a DB expression (e.g. `new Expression('NOW()')`). |
|
* If not set, it will use the value of `time()` to fill the attributes. |
|
*/ |
|
public $timestamp; |
|
|
|
|
|
/** |
|
* Declares event handlers for the [[owner]]'s events. |
|
* @return array events (array keys) and the corresponding event handler methods (array values). |
|
*/ |
|
public function events() |
|
{ |
|
$events = $this->attributes; |
|
foreach ($events as $i => $event) { |
|
$events[$i] = 'updateTimestamp'; |
|
} |
|
return $events; |
|
} |
|
|
|
/** |
|
* Updates the attributes with the current timestamp. |
|
* @param Event $event |
|
*/ |
|
public function updateTimestamp($event) |
|
{ |
|
$attributes = isset($this->attributes[$event->name]) ? (array)$this->attributes[$event->name] : array(); |
|
if (!empty($attributes)) { |
|
$timestamp = $this->evaluateTimestamp(); |
|
foreach ($attributes as $attribute) { |
|
$this->owner->$attribute = $timestamp; |
|
} |
|
} |
|
} |
|
|
|
/** |
|
* Gets the current timestamp. |
|
* @return mixed the timestamp value |
|
*/ |
|
protected function evaluateTimestamp() |
|
{ |
|
if ($this->timestamp instanceof Expression) { |
|
return $this->timestamp; |
|
} elseif ($this->timestamp !== null) { |
|
return call_user_func($this->timestamp); |
|
} else { |
|
return time(); |
|
} |
|
} |
|
}
|
|
|