93 lines
4.7 KiB
PHP
93 lines
4.7 KiB
PHP
<?php
|
|
|
|
namespace plugin\model\process\video;
|
|
|
|
use plugin\control\expose\helper\Uploads;
|
|
use plugin\model\app\model\PluginModelTask;
|
|
use plugin\model\app\model\PluginModelTaskResult;
|
|
use plugin\model\utils\enum\ModelScene;
|
|
use plugin\model\utils\enum\ModelTaskStatus;
|
|
use plugin\model\utils\enum\ModelType;
|
|
use plugin\notification\expose\helper\Push;
|
|
use plugin\shortplay\app\model\PluginShortplayDramaStoryboard;
|
|
use support\Log;
|
|
use think\facade\Db;
|
|
use Workerman\Coroutine;
|
|
use Workerman\Crontab\Crontab;
|
|
use Workerman\Timer;
|
|
|
|
class VideoTransfer
|
|
{
|
|
public function onWorkerStart($worker)
|
|
{
|
|
$id = $worker->id;
|
|
new Crontab('*/5 * * * * *', function () use ($id) {
|
|
try {
|
|
if ($id) {
|
|
Timer::sleep(0.3 * $id);
|
|
}
|
|
Coroutine::create(function () {
|
|
$PluginModelTask = PluginModelTask::where(['status' => ModelTaskStatus::WAIT_DOWNLOAD['value'], 'model_type' => ModelType::TOVIDEO['value']])->order('last_heartbeat asc,id asc')->lock(true)->find();
|
|
if (!$PluginModelTask) {
|
|
return;
|
|
}
|
|
$PluginModelTask->status = ModelTaskStatus::DOWNLOADING['value'];
|
|
$PluginModelTask->last_heartbeat = date('Y-m-d H:i:s', strtotime('+5 seconds'));
|
|
$PluginModelTask->save();
|
|
$PluginModelTaskResult = PluginModelTaskResult::where('task_id', $PluginModelTask->id)->find();
|
|
if (!$PluginModelTaskResult) {
|
|
return;
|
|
}
|
|
try {
|
|
$ModelScene = ModelScene::get($PluginModelTask->scene);
|
|
$classify = Uploads::getClassify($PluginModelTask->channels_uid, 'uploads/' . $PluginModelTask->scene, $ModelScene['label']);
|
|
$result = Uploads::download($PluginModelTask->channels_uid, $PluginModelTaskResult->video, $classify);
|
|
} catch (\Throwable $th) {
|
|
$PluginModelTask = PluginModelTask::where('id', $PluginModelTask->id)->find();
|
|
$PluginModelTask->status = ModelTaskStatus::WAIT_DOWNLOAD['value'];
|
|
$PluginModelTask->save();
|
|
Log::error("视频转存处理失败：" . $th->getMessage(), $th->getTrace());
|
|
return;
|
|
}
|
|
$pushData = [
|
|
'task_id' => $PluginModelTask->id,
|
|
'id' => $PluginModelTask->alias_id,
|
|
'video' => $result->file_url
|
|
];
|
|
$event = 'generate' . strtolower(str_replace('_', '', $PluginModelTask->scene));
|
|
Db::startTrans();
|
|
try {
|
|
$PluginModelTask = PluginModelTask::where('id', $PluginModelTask->id)->find();
|
|
$PluginModelTask->status = ModelTaskStatus::SUCCESS['value'];
|
|
$PluginModelTask->save();
|
|
$PluginModelTaskResult = PluginModelTaskResult::where('task_id', $PluginModelTask->id)->find();
|
|
$PluginModelTaskResult->video_path = $result->file_name;
|
|
$PluginModelTaskResult->save();
|
|
switch ($PluginModelTask->scene) {
|
|
case ModelScene::STORYBOARD_VIDEO['value']:
|
|
$PluginShortplayDramaStoryboard = PluginShortplayDramaStoryboard::where('id', $PluginModelTask->alias_id)->find();
|
|
$PluginShortplayDramaStoryboard->video = $result->file_name;
|
|
$PluginShortplayDramaStoryboard->use_material_type='video';
|
|
$PluginShortplayDramaStoryboard->save();
|
|
$event = 'generatestoryboard';
|
|
$pushData['event'] = ModelScene::STORYBOARD_VIDEO['value'];
|
|
break;
|
|
}
|
|
Db::commit();
|
|
} catch (\Throwable $th) {
|
|
Db::rollback();
|
|
Log::error("视频转存保存失败：" . $th->getMessage(), $th->getTrace());
|
|
}
|
|
Push::send([
|
|
'uid' => $PluginModelTask->uid,
|
|
'channels_uid' => $PluginModelTask->channels_uid,
|
|
'event' => $event
|
|
], $pushData);
|
|
});
|
|
} catch (\Throwable $th) {
|
|
Log::error("视频转存定时任务失败：" . $th->getMessage(), $th->getTrace());
|
|
}
|
|
});
|
|
}
|
|
}
|