luyadev/luya-module-admin

<?php

namespace luya\admin\behaviors;

use luya\admin\models\QueueLog;
use luya\admin\models\QueueLogError;
use Yii;
use yii\base\Behavior;
use yii\queue\ExecEvent;
use yii\queue\JobEvent;
use yii\queue\JobInterface;
use yii\queue\PushEvent;
use yii\queue\Queue;

/**
 * Log Behavior.
 *
 * @author Roman Zhuravlev <zhuravljov@gmail.com>
 * @since 2.0.0
 */
class QueueLogBehavior extends Behavior
{
    /**
     * @var Queue
     * @inheritdoc
     */
    public $owner;

    /**
     * @inheritdoc
     */
    public function events()
    {
        return [
            Queue::EVENT_AFTER_PUSH => 'afterPush',
            Queue::EVENT_BEFORE_EXEC => 'beforeExec',
            Queue::EVENT_AFTER_EXEC => 'afterExec',
            Queue::EVENT_AFTER_ERROR => 'afterError',
        ];
    }

    /**
     * @param PushEvent $event
     */
    public function afterPush(PushEvent $event)
    {
        $log = new QueueLog();
        $log->queue_id = $event->id;
        $log->push_timestamp = time();
        $log->title = $this->getJobTitle($event);
        $log->save();
    }

    /**
     * @param ExecEvent $event
     */
    public function beforeExec(ExecEvent $event)
    {
        $log = QueueLog::findOne(['queue_id' => $event->id]);

        if ($log) {
            $log->updateAttributes(['run_timestamp' => time(), 'title' => $this->getExecTitle($event)]);
        }
    }

    /**
     * @param ExecEvent $event
     */
    public function afterExec(ExecEvent $event)
    {
        $log = QueueLog::findOne(['queue_id' => $event->id]);

        if ($log) {
            $log->updateAttributes(['end_timestamp' => time(), 'title' => $this->getExecTitle($event), 'is_error' => false]);
        }
    }

    /**
     * @param ExecEvent $event
     */
    public function afterError(ExecEvent $event)
    {
        $log = QueueLog::findOne(['queue_id' => $event->id]);

        /** @var \Exception $err */
        $err = $event->error;
        if ($log) {
            $log->updateAttributes(['end_timestamp' => time(), 'is_error' => true, 'title' => $this->getExecTitle($event)]);

            $errorLog = new QueueLogError();
            $errorLog->queue_log_id = $log->id;
            $errorLog->message = $err->getMessage();
            $errorLog->code = (string) $err->getCode();
            $errorLog->trace = $err->getTraceAsString();
            $errorLog->file = $err->getFile();
            $errorLog->line = (string) $err->getLine();
            $errorLog->save();
        }

        // send error
        Yii::$app->errorHandler->transferMessage($event->error->getMessage(), $event->error->getFile(), $event->error->getLine());
    }
    /**
     * @param JobEvent $event
     * @return string
     */
    protected function getJobTitle(JobEvent $event)
    {
        return $event->job instanceof JobInterface ? $event->job !== null ? $event->job::class : self::class : 'Unknown job Title';
    }

    /**
     * @param ExecEvent $event
     * @return string
     */
    protected function getExecTitle(ExecEvent $event)
    {
        $title = $this->getJobTitle($event);
        $extra = "attempt: $event->attempt";
        if ($pid = $event->sender->getWorkerPid()) {
            $extra .= ", PID: $pid";
        }
        return "$title ($extra)";
    }
}