mirror of
https://github.com/Vateron-Media/XC_VM.git
synced 2026-10-23 20:03:11 +02:00
Divarion-D
48b3fbf0ca
- ConnectionTracker: null checks for RedisManager::instance() - UsersCronJob: safe array access for getConnections() - table.php: null coalesce for rFailsPS + Redis fallback
592 lines
28 KiB
PHP
592 lines
28 KiB
PHP
<?php
|
|
|
|
/**
|
|
* UsersCronJob — users cron job
|
|
*
|
|
* @package XC_VM_CLI_CronJobs
|
|
* @author Divarion_D <https://github.com/Divarion-D>
|
|
* @copyright 2025-2026 Vateron Media
|
|
* @link https://github.com/Vateron-Media/XC_VM
|
|
* @license AGPL-3.0 https://www.gnu.org/licenses/agpl-3.0.html
|
|
*/
|
|
|
|
require_once __DIR__ . '/../CronTrait.php';
|
|
|
|
class UsersCronJob implements CommandInterface {
|
|
use CronTrait;
|
|
|
|
private $rPHPPIDs = array();
|
|
private $rServers = array();
|
|
|
|
public function getName(): string {
|
|
return 'cron:users';
|
|
}
|
|
|
|
public function getDescription(): string {
|
|
return 'Cron: manage user connections, Redis sync, divergence';
|
|
}
|
|
|
|
public function execute(array $rArgs): int {
|
|
if (!$this->assertRunAsXcVm()) {
|
|
return 1;
|
|
}
|
|
|
|
set_time_limit(0);
|
|
ini_set('memory_limit', -1);
|
|
|
|
$this->setProcessTitle('XC_VM[Users]');
|
|
$this->acquireCronLock();
|
|
|
|
global $db;
|
|
|
|
$rSync = null;
|
|
$this->rServers = ServerRepository::getAll();
|
|
|
|
if (!empty($rArgs[0]) && $this->rServers[SERVER_ID]['is_main']) {
|
|
RedisManager::ensureConnected();
|
|
|
|
if (RedisManager::isConnected()) {
|
|
$rSync = intval($rArgs[0]);
|
|
|
|
if ($rSync == 1) {
|
|
$rDeSync = $rRedisUsers = $rRedisUpdate = $rRedisSet = array();
|
|
$db->query('SELECT * FROM `lines_live` WHERE `hls_end` = 0;');
|
|
$rRows = $db->get_rows();
|
|
|
|
if (count($rRows) > 0) {
|
|
$rStreamIDs = array();
|
|
|
|
foreach ($rRows as $rRow) {
|
|
$streamId = (int)$rRow['stream_id'];
|
|
if ($streamId > 0 && !in_array($streamId, $rStreamIDs)) {
|
|
$rStreamIDs[] = $streamId;
|
|
}
|
|
}
|
|
|
|
$rOnDemand = array();
|
|
if (count($rStreamIDs) > 0) {
|
|
$db->query('SELECT `stream_id`, `server_id`, `on_demand` FROM `streams_servers` WHERE `stream_id` IN (' . implode(',', $rStreamIDs) . ');');
|
|
foreach ($db->get_rows() as $rRow) {
|
|
$rOnDemand[$rRow['stream_id']][$rRow['server_id']] = intval($rRow['on_demand']);
|
|
}
|
|
}
|
|
|
|
$rRedis = RedisManager::instance()->multi();
|
|
|
|
foreach ($rRows as $rRow) {
|
|
echo 'Resynchronising UUID: ' . $rRow['uuid'] . "\n";
|
|
|
|
if (empty($rRow['hmac_id'])) {
|
|
$rRow['identity'] = $rRow['user_id'];
|
|
} else {
|
|
$rRow['identity'] = $rRow['hmac_id'] . '_' . $rRow['hmac_identifier'];
|
|
}
|
|
|
|
$rRow['on_demand'] = ($rOnDemand[$rRow['stream_id']][$rRow['server_id']] ?: 0);
|
|
$rRedis->zAdd('LINE#' . $rRow['identity'], $rRow['date_start'], $rRow['uuid']);
|
|
$rRedis->zAdd('LINE_ALL#' . $rRow['identity'], $rRow['date_start'], $rRow['uuid']);
|
|
$rRedis->zAdd('STREAM#' . $rRow['stream_id'], $rRow['date_start'], $rRow['uuid']);
|
|
$rRedis->zAdd('SERVER#' . $rRow['server_id'], $rRow['date_start'], $rRow['uuid']);
|
|
|
|
if ($rRow['user_id']) {
|
|
$rRedis->zAdd('SERVER_LINES#' . $rRow['server_id'], $rRow['user_id'], $rRow['uuid']);
|
|
}
|
|
|
|
if ($rRow['proxy_id']) {
|
|
$rRedis->zAdd('PROXY#' . $rRow['proxy_id'], $rRow['date_start'], $rRow['uuid']);
|
|
}
|
|
|
|
$rRedis->zAdd('CONNECTIONS', $rRow['date_start'], $rRow['uuid']);
|
|
$rRedis->zAdd('LIVE', $rRow['date_start'], $rRow['uuid']);
|
|
$rRedis->set($rRow['uuid'], igbinary_serialize($rRow));
|
|
$rDeSync[] = $rRow['uuid'];
|
|
}
|
|
$rRedis->exec();
|
|
|
|
if (count($rDeSync) > 0) {
|
|
$db->query("DELETE FROM `lines_live` WHERE `uuid` IN ('" . implode("','", $rDeSync) . "');");
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
echo "Couldn't connect to Redis.\n";
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
if (SettingsManager::getAll()['redis_handler'] && $this->rServers[SERVER_ID]['is_main']) {
|
|
$this->rServers = ServerRepository::getAll(true);
|
|
|
|
foreach ($this->rServers as $rServer) {
|
|
$rDecodedPids = json_decode($rServer['php_pids'] ?? '', true);
|
|
$this->rPHPPIDs[$rServer['id']] = is_array($rDecodedPids) ? array_map('intval', $rDecodedPids) : [];
|
|
}
|
|
}
|
|
|
|
$this->loadCron();
|
|
|
|
return 0;
|
|
}
|
|
|
|
private function processDeletions($rDelete, $rDelStream = array()) {
|
|
global $db;
|
|
$rTime = time();
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
if ($rDelete['count'] > 0) {
|
|
$rRedis = RedisManager::instance()->multi();
|
|
|
|
foreach ($rDelete['line'] as $rUserID => $rUUIDs) {
|
|
$rRedis->zRem('LINE#' . $rUserID, ...$rUUIDs);
|
|
$rRedis->zRem('LINE_ALL#' . $rUserID, ...$rUUIDs);
|
|
}
|
|
|
|
foreach ($rDelete['stream'] as $rStreamID => $rUUIDs) {
|
|
$rRedis->zRem('STREAM#' . $rStreamID, ...$rUUIDs);
|
|
}
|
|
|
|
foreach ($rDelete['server'] as $rServerID => $rUUIDs) {
|
|
$rRedis->zRem('SERVER#' . $rServerID, ...$rUUIDs);
|
|
$rRedis->zRem('SERVER_LINES#' . $rServerID, ...$rUUIDs);
|
|
}
|
|
|
|
foreach ($rDelete['proxy'] as $rProxyID => $rUUIDs) {
|
|
$rRedis->zRem('PROXY#' . $rProxyID, ...$rUUIDs);
|
|
}
|
|
|
|
if (count($rDelete['uuid']) > 0) {
|
|
$rRedis->zRem('CONNECTIONS', ...$rDelete['uuid']);
|
|
$rRedis->zRem('LIVE', ...$rDelete['uuid']);
|
|
$rRedis->sRem('ENDED', ...$rDelete['uuid']);
|
|
$rRedis->del(...$rDelete['uuid']);
|
|
}
|
|
|
|
$rRedis->exec();
|
|
}
|
|
} else {
|
|
foreach ($rDelete as $rServerID => $rConnections) {
|
|
if (count($rConnections) > 0) {
|
|
$db->query("DELETE FROM `lines_live` WHERE `uuid` IN ('" . implode("','", $rConnections) . "')");
|
|
}
|
|
}
|
|
}
|
|
|
|
foreach ((SettingsManager::getAll()['redis_handler'] ? $rDelete['server'] : $rDelete) as $rServerID => $rConnections) {
|
|
if ($rServerID != SERVER_ID) {
|
|
$rQuery = '';
|
|
|
|
foreach ($rConnections as $rConnection) {
|
|
$rQuery .= '(' . $rServerID . ',1,' . $rTime . ',' . $db->escape(json_encode(array('type' => 'delete_con', 'uuid' => $rConnection))) . '),';
|
|
}
|
|
$rQuery = rtrim($rQuery, ',');
|
|
|
|
if (!empty($rQuery)) {
|
|
$db->query('INSERT INTO `signals`(`server_id`, `cache`, `time`, `custom_data`) VALUES ' . $rQuery . ';');
|
|
}
|
|
}
|
|
}
|
|
|
|
foreach ($rDelStream as $rStreamID => $rConnections) {
|
|
foreach ($rConnections as $rConnection) {
|
|
@unlink(CONS_TMP_PATH . $rStreamID . '/' . $rConnection);
|
|
}
|
|
}
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
return array('line' => array(), 'server' => array(), 'server_lines' => array(), 'proxy' => array(), 'stream' => array(), 'uuid' => array(), 'count' => 0);
|
|
}
|
|
|
|
return array();
|
|
}
|
|
|
|
private function loadCron(): void {
|
|
global $db;
|
|
|
|
$rServers = $this->rServers;
|
|
$rPHPPIDs = $this->rPHPPIDs;
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
RedisManager::ensureConnected();
|
|
}
|
|
|
|
$rStartTime = time();
|
|
|
|
if (!SettingsManager::getAll()['redis_handler'] || $rServers[SERVER_ID]['is_main']) {
|
|
$rAutoKick = SettingsManager::getAll()['user_auto_kick_hours'] * 3600;
|
|
$rLiveKeys = $rDelete = $rDeleteStream = array();
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rRedisDelete = array('line' => array(), 'server' => array(), 'server_lines' => array(), 'proxy' => array(), 'stream' => array(), 'uuid' => array(), 'count' => 0);
|
|
$rUsers = array();
|
|
$rResult = ConnectionTracker::getConnections();
|
|
$rKeys = $rResult[0] ?? [];
|
|
$rConnections = $rResult[1] ?? [];
|
|
$i = 0;
|
|
|
|
for ($rSize = count($rConnections); $i < $rSize; $i++) {
|
|
$rConnection = $rConnections[$i];
|
|
|
|
if (is_array($rConnection)) {
|
|
$rUsers[$rConnection['identity']][] = $rConnection;
|
|
$rLiveKeys[] = $rConnection['uuid'];
|
|
} else {
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['uuid'][] = $rKeys[$i];
|
|
}
|
|
}
|
|
unset($rConnections);
|
|
} else {
|
|
$rUsers = ConnectionTracker::getConnections(($rServers[SERVER_ID]['is_main'] ? null : SERVER_ID));
|
|
}
|
|
|
|
$rRestreamerArray = $rMaxConnectionsArray = array();
|
|
$rUserIDs = InputValidator::confirmIDs(array_keys($rUsers));
|
|
|
|
if (count($rUserIDs) > 0) {
|
|
$db->query('SELECT `id`, `max_connections`, `is_restreamer` FROM `lines` WHERE `id` IN (' . implode(',', $rUserIDs) . ');');
|
|
|
|
foreach ($db->get_rows() as $rRow) {
|
|
$rMaxConnectionsArray[$rRow['id']] = $rRow['max_connections'];
|
|
$rRestreamerArray[$rRow['id']] = $rRow['is_restreamer'];
|
|
}
|
|
}
|
|
|
|
if (SettingsManager::getAll()['redis_handler'] && $rServers[SERVER_ID]['is_main']) {
|
|
foreach (ConnectionTracker::getEnded() as $rConnection) {
|
|
if (is_array($rConnection)) {
|
|
if (!in_array($rConnection['container'], array('ts', 'hls', 'rtmp')) && time() - $rConnection['hls_last_read'] < 300) {
|
|
$rClose = false;
|
|
} else {
|
|
$rClose = true;
|
|
}
|
|
|
|
if ($rClose) {
|
|
echo 'Close connection: ' . $rConnection['uuid'] . "\n";
|
|
ConnectionTracker::closeConnection($rConnection, false, false);
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['line'][$rConnection['identity']][] = $rConnection['uuid'];
|
|
$rRedisDelete['stream'][$rConnection['stream_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['server'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['uuid'][] = $rConnection['uuid'];
|
|
|
|
if ($rConnection['proxy_id']) {
|
|
$rRedisDelete['proxy'][$rConnection['proxy_id']][] = $rConnection['uuid'];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if ($rRedisDelete['count'] >= 1000) {
|
|
$rRedisDelete = $this->processDeletions($rRedisDelete, $rRedisDelete['stream']);
|
|
}
|
|
}
|
|
|
|
foreach ($rUsers as $rUserID => $rConnections) {
|
|
$rActiveCount = 0;
|
|
$rMaxConnections = $rMaxConnectionsArray[$rUserID];
|
|
$rIsRestreamer = ($rRestreamerArray[$rUserID] ?: false);
|
|
|
|
foreach ($rConnections as $rKey => $rConnection) {
|
|
if ($rConnection['server_id'] == SERVER_ID || SettingsManager::getAll()['redis_handler']) {
|
|
if (!isset($rConnection['exp_date']) || is_null($rConnection['exp_date']) || $rConnection['exp_date'] >= $rStartTime) {
|
|
$rTotalTime = $rStartTime - $rConnection['date_start'];
|
|
|
|
if (!($rAutoKick != 0 && $rAutoKick <= $rTotalTime) || $rIsRestreamer) {
|
|
if ($rConnection['container'] == 'hls') {
|
|
if (30 <= $rStartTime - $rConnection['hls_last_read'] || $rConnection['hls_end'] == 1) {
|
|
echo 'Close connection: ' . $rConnection['uuid'] . "\n";
|
|
ConnectionTracker::closeConnection($rConnection, false, false);
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['line'][$rConnection['identity']][] = $rConnection['uuid'];
|
|
$rRedisDelete['stream'][$rConnection['stream_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['server'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['uuid'][] = $rConnection['uuid'];
|
|
|
|
if ($rConnection['user_id']) {
|
|
$rRedisDelete['server_lines'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
}
|
|
|
|
if ($rConnection['proxy_id']) {
|
|
$rRedisDelete['proxy'][$rConnection['proxy_id']][] = $rConnection['uuid'];
|
|
}
|
|
} else {
|
|
$rDelete[$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rDeleteStream[$rConnection['stream_id']] = $rDelete[$rConnection['server_id']];
|
|
}
|
|
}
|
|
} else {
|
|
if ($rConnection['container'] != 'rtmp') {
|
|
if ($rConnection['server_id'] == SERVER_ID) {
|
|
$rIsRunning = ProcessManager::isRunning($rConnection['pid'], 'php-fpm');
|
|
} else {
|
|
if ($rConnection['date_start'] <= $rServers[$rConnection['server_id']]['last_check_ago'] - 1 && 0 < count($rPHPPIDs[$rConnection['server_id']])) {
|
|
$rIsRunning = in_array(intval($rConnection['pid']), $rPHPPIDs[$rConnection['server_id']]);
|
|
} else {
|
|
$rIsRunning = true;
|
|
}
|
|
}
|
|
|
|
if (($rConnection['hls_end'] == 1 && ($rStartTime - $rConnection['hls_last_read']) >= 300) || !$rIsRunning) {
|
|
echo 'Close connection: ' . $rConnection['uuid'] . "\n";
|
|
ConnectionTracker::closeConnection($rConnection, false, false);
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['line'][$rConnection['identity']][] = $rConnection['uuid'];
|
|
$rRedisDelete['stream'][$rConnection['stream_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['server'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['uuid'][] = $rConnection['uuid'];
|
|
|
|
if ($rConnection['user_id']) {
|
|
$rRedisDelete['server_lines'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
}
|
|
|
|
if ($rConnection['proxy_id']) {
|
|
$rRedisDelete['proxy'][$rConnection['proxy_id']][] = $rConnection['uuid'];
|
|
}
|
|
} else {
|
|
$rDelete[$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rDeleteStream[$rConnection['stream_id']] = $rDelete[$rConnection['server_id']];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
echo 'Close connection: ' . $rConnection['uuid'] . "\n";
|
|
ConnectionTracker::closeConnection($rConnection, false, false);
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['line'][$rConnection['identity']][] = $rConnection['uuid'];
|
|
$rRedisDelete['stream'][$rConnection['stream_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['server'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['uuid'][] = $rConnection['uuid'];
|
|
|
|
if ($rConnection['user_id']) {
|
|
$rRedisDelete['server_lines'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
}
|
|
|
|
if ($rConnection['proxy_id']) {
|
|
$rRedisDelete['proxy'][$rConnection['proxy_id']][] = $rConnection['uuid'];
|
|
}
|
|
} else {
|
|
$rDelete[$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rDeleteStream[$rConnection['stream_id']] = $rDelete[$rConnection['server_id']];
|
|
}
|
|
}
|
|
} else {
|
|
echo 'Close connection: ' . $rConnection['uuid'] . "\n";
|
|
ConnectionTracker::closeConnection($rConnection, false, false);
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['line'][$rConnection['identity']][] = $rConnection['uuid'];
|
|
$rRedisDelete['stream'][$rConnection['stream_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['server'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['uuid'][] = $rConnection['uuid'];
|
|
|
|
if ($rConnection['user_id']) {
|
|
$rRedisDelete['server_lines'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
}
|
|
|
|
if ($rConnection['proxy_id']) {
|
|
$rRedisDelete['proxy'][$rConnection['proxy_id']][] = $rConnection['uuid'];
|
|
}
|
|
} else {
|
|
$rDelete[$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rDeleteStream[$rConnection['stream_id']] = $rDelete[$rConnection['server_id']];
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!$rConnection['hls_end']) {
|
|
$rActiveCount++;
|
|
}
|
|
}
|
|
|
|
if ($rServers[SERVER_ID]['is_main'] && 0 < $rMaxConnections && $rMaxConnections < $rActiveCount) {
|
|
foreach ($rConnections as $rConnection) {
|
|
if (!$rConnection['hls_end']) {
|
|
echo 'Close connection: ' . $rConnection['uuid'] . "\n";
|
|
ConnectionTracker::closeConnection($rConnection, false, false);
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rRedisDelete['count']++;
|
|
$rRedisDelete['line'][$rConnection['identity']][] = $rConnection['uuid'];
|
|
$rRedisDelete['stream'][$rConnection['stream_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['server'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rRedisDelete['uuid'][] = $rConnection['uuid'];
|
|
|
|
if ($rConnection['user_id']) {
|
|
$rRedisDelete['server_lines'][$rConnection['server_id']][] = $rConnection['uuid'];
|
|
}
|
|
|
|
if ($rConnection['proxy_id']) {
|
|
$rRedisDelete['proxy'][$rConnection['proxy_id']][] = $rConnection['uuid'];
|
|
}
|
|
} else {
|
|
$rDelete[$rConnection['server_id']][] = $rConnection['uuid'];
|
|
$rDeleteStream[$rConnection['stream_id']] = $rDelete[$rConnection['server_id']];
|
|
}
|
|
|
|
$rActiveCount--;
|
|
}
|
|
|
|
if ($rActiveCount >= $rMaxConnections) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (SettingsManager::getAll()['redis_handler'] && 1000 <= $rRedisDelete['count']) {
|
|
$rRedisDelete = $this->processDeletions($rRedisDelete, $rRedisDelete['stream']);
|
|
} else {
|
|
if (!SettingsManager::getAll()['redis_handler'] && count($rDelete) >= 1000) {
|
|
$rDelete = $this->processDeletions($rDelete, $rDeleteStream);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (SettingsManager::getAll()['redis_handler'] && 0 < $rRedisDelete['count']) {
|
|
$this->processDeletions($rRedisDelete, $rRedisDelete['stream']);
|
|
} else {
|
|
if (!SettingsManager::getAll()['redis_handler'] && count($rDelete) > 0) {
|
|
$this->processDeletions($rDelete, $rDeleteStream);
|
|
}
|
|
}
|
|
}
|
|
|
|
$rConnectionSpeeds = glob(DIVERGENCE_TMP_PATH . '*');
|
|
|
|
if (count($rConnectionSpeeds) > 0) {
|
|
$rBitrates = [];
|
|
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rStreamMap = [];
|
|
|
|
$db->query('SELECT `stream_id`, `bitrate` FROM `streams_servers` WHERE `server_id` = ? AND `bitrate` IS NOT NULL;', SERVER_ID);
|
|
foreach ($db->get_rows() as $rRow) {
|
|
$bitrate = intval($rRow['bitrate']);
|
|
if ($bitrate > 0) {
|
|
$rStreamMap[intval($rRow['stream_id'])] = intval($bitrate / 8 * 0.92);
|
|
}
|
|
}
|
|
|
|
$rUUIDs = [];
|
|
foreach ($rConnectionSpeeds as $rConnectionSpeed) {
|
|
if (!empty($rConnectionSpeed)) {
|
|
$rUUIDs[] = basename($rConnectionSpeed);
|
|
}
|
|
}
|
|
|
|
if (count($rUUIDs) > 0) {
|
|
$rConnections = array_map(
|
|
static fn($v) => ($v !== false) ? igbinary_unserialize($v) : null,
|
|
RedisManager::instance()->mGet($rUUIDs)
|
|
);
|
|
|
|
foreach ($rConnections as $rConnection) {
|
|
if (!is_array($rConnection)) {
|
|
continue;
|
|
}
|
|
|
|
$uuid = $rConnection['uuid'];
|
|
$streamId = intval($rConnection['stream_id']);
|
|
|
|
if (!isset($rStreamMap[$streamId])) {
|
|
continue;
|
|
}
|
|
|
|
$rBitrates[$uuid] = $rStreamMap[$streamId];
|
|
}
|
|
}
|
|
|
|
unset($rStreamMap);
|
|
} else {
|
|
$db->query('SELECT `lines_live`.`uuid`, `streams_servers`.`bitrate` FROM `lines_live` LEFT JOIN `streams_servers` ON `lines_live`.`stream_id` = `streams_servers`.`stream_id` AND `lines_live`.`server_id` = `streams_servers`.`server_id` WHERE `lines_live`.`server_id` = ?;', SERVER_ID);
|
|
|
|
foreach ($db->get_rows() as $rRow) {
|
|
$bitrate = intval($rRow['bitrate']);
|
|
if ($bitrate > 0) {
|
|
$rBitrates[$rRow['uuid']] = intval($bitrate / 8 * 0.92);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!SettingsManager::getAll()['redis_handler']) {
|
|
$rUUIDMap = array();
|
|
$db->query('SELECT `uuid`, `activity_id` FROM `lines_live`;');
|
|
foreach ($db->get_rows() as $rRow) {
|
|
$rUUIDMap[$rRow['uuid']] = $rRow['activity_id'];
|
|
}
|
|
}
|
|
|
|
$rLiveQuery = $rDivergenceUpdate = [];
|
|
|
|
foreach ($rConnectionSpeeds as $rConnectionSpeed) {
|
|
if (empty($rConnectionSpeed)) {
|
|
continue;
|
|
}
|
|
|
|
$rUUID = basename($rConnectionSpeed);
|
|
$rAverageSpeed = intval(file_get_contents($rConnectionSpeed));
|
|
|
|
if (!isset($rBitrates[$rUUID]) || $rBitrates[$rUUID] <= 0) {
|
|
$rDivergenceUpdate[] = "('" . $rUUID . "', 0)";
|
|
|
|
if (!SettingsManager::getAll()['redis_handler'] && isset($rUUIDMap[$rUUID])) {
|
|
$rLiveQuery[] = '(' . $rUUIDMap[$rUUID] . ', 0)';
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
$realBitrate = $rBitrates[$rUUID];
|
|
$rDivergence = intval(($rAverageSpeed - $realBitrate) / $realBitrate * 100);
|
|
|
|
if ($rDivergence > 0) {
|
|
$rDivergence = 0;
|
|
}
|
|
|
|
$rDivergenceUpdate[] = "('" . $rUUID . "', " . abs($rDivergence) . ')';
|
|
|
|
if (!SettingsManager::getAll()['redis_handler'] && isset($rUUIDMap[$rUUID])) {
|
|
$rLiveQuery[] = '(' . $rUUIDMap[$rUUID] . ', ' . abs($rDivergence) . ')';
|
|
}
|
|
}
|
|
|
|
if (count($rDivergenceUpdate) > 0) {
|
|
$rUpdateQuery = implode(',', $rDivergenceUpdate);
|
|
$db->query('INSERT INTO `lines_divergence`(`uuid`,`divergence`) VALUES ' . $rUpdateQuery . ' ON DUPLICATE KEY UPDATE `divergence`=VALUES(`divergence`);');
|
|
}
|
|
|
|
if (!SettingsManager::getAll()['redis_handler'] && count($rLiveQuery) > 0) {
|
|
$rLiveQueryStr = implode(',', $rLiveQuery);
|
|
$db->query('INSERT INTO `lines_live`(`activity_id`,`divergence`) VALUES ' . $rLiveQueryStr . ' ON DUPLICATE KEY UPDATE `divergence`=VALUES(`divergence`);');
|
|
}
|
|
|
|
shell_exec('rm -f ' . DIVERGENCE_TMP_PATH . '*');
|
|
}
|
|
|
|
if ($rServers[SERVER_ID]['is_main']) {
|
|
if (SettingsManager::getAll()['redis_handler']) {
|
|
$rDeleteQuery = "DELETE FROM `lines_divergence` WHERE `uuid` NOT IN ('" . implode("','", $rLiveKeys) . "');";
|
|
for ($rRetry = 0; $rRetry < 3; $rRetry++) {
|
|
if ($db->query($rDeleteQuery)) {
|
|
break;
|
|
}
|
|
usleep(200000);
|
|
}
|
|
} else {
|
|
$db->query('DELETE FROM `lines_divergence` WHERE `uuid` NOT IN (SELECT `uuid` FROM `lines_live`);');
|
|
}
|
|
}
|
|
|
|
if ($rServers[SERVER_ID]['is_main']) {
|
|
$db->query('DELETE FROM `lines_live` WHERE `uuid` IS NULL;');
|
|
}
|
|
}
|
|
}
|