mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-29 01:08:56 +00:00
157 lines
5.4 KiB
PHP
157 lines
5.4 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace App\Services\Runners;
|
|
|
|
use App\Models\Settings;
|
|
use Illuminate\Support\Carbon;
|
|
use Illuminate\Support\Facades\DB;
|
|
|
|
class BackfillRunner extends BaseRunner
|
|
{
|
|
/**
|
|
* @param array<string, mixed> $options
|
|
*/
|
|
public function backfill(array $options = []): void
|
|
{
|
|
$select = 'SELECT name';
|
|
if (($options[0] ?? false) !== false) {
|
|
$select .= ', '.$options[0].' AS max';
|
|
}
|
|
$select .= ' FROM usenet_groups WHERE backfill = 1';
|
|
$work = DB::select($select);
|
|
|
|
$maxProcesses = (int) Settings::settingValue('backfillthreads');
|
|
|
|
$count = count($work);
|
|
if ($count === 0) {
|
|
$this->headerNone();
|
|
|
|
return;
|
|
}
|
|
|
|
// Streaming mode
|
|
if ((bool) config('nntmux.stream_fork_output', false) === true) {
|
|
$commands = [];
|
|
foreach ($work as $group) {
|
|
$commands[] = PHP_BINARY.' artisan update:backfill '.$group->name.(isset($group->max) ? (' '.$group->max) : '');
|
|
}
|
|
$this->runStreamingCommands($commands, $maxProcesses, 'backfill'); // @phpstan-ignore argument.type
|
|
|
|
return;
|
|
}
|
|
|
|
$this->headerStart('backfill', $count, $maxProcesses);
|
|
|
|
// Build commands array for parallel execution
|
|
$commands = [];
|
|
foreach ($work as $group) {
|
|
$commands[$group->name] = PHP_BINARY.' artisan update:backfill '.$group->name.(isset($group->max) ? (' '.$group->max) : '');
|
|
}
|
|
|
|
// Process using parallel commands with configurable timeout
|
|
$results = $this->runParallelCommands($commands, $maxProcesses);
|
|
|
|
foreach ($results as $groupName => $output) {
|
|
echo $output;
|
|
cli()->primary('Backfilled group '.$groupName);
|
|
}
|
|
}
|
|
|
|
public function safeBackfill(): void
|
|
{
|
|
// make sure short_groups is up-to-date - Updated to use new script location (modernized)
|
|
$this->executeCommand(PHP_BINARY.' app/Services/Tmux/Scripts/update_groups.php');
|
|
|
|
$backfill_qty = (int) Settings::settingValue('backfill_qty');
|
|
$backfill_order = (int) Settings::settingValue('backfill_order');
|
|
$backfill_days = (int) Settings::settingValue('backfill_days');
|
|
$maxMessages = (int) Settings::settingValue('maxmssgs');
|
|
$threads = (int) Settings::settingValue('backfillthreads');
|
|
|
|
$orderby = match ($backfill_order) {
|
|
1 => 'ORDER BY first_record_postdate DESC',
|
|
2 => 'ORDER BY first_record_postdate ASC',
|
|
3 => 'ORDER BY name ASC',
|
|
4 => 'ORDER BY name DESC',
|
|
5 => 'ORDER BY a.last_record DESC',
|
|
default => 'ORDER BY a.last_record ASC',
|
|
};
|
|
|
|
$backfilldays = '0';
|
|
if ($backfill_days === 1) {
|
|
$backfilldays = 'g.backfill_target';
|
|
} elseif ($backfill_days === 2) {
|
|
$backfilldays = (string) now()->diffInDays(Carbon::createFromFormat('Y-m-d', Settings::settingValue('safebackfilldate')), true);
|
|
}
|
|
|
|
$sql = 'SELECT g.name,
|
|
g.first_record AS our_first,
|
|
MAX(a.first_record) AS their_first,
|
|
MAX(a.last_record) AS their_last
|
|
FROM usenet_groups g
|
|
INNER JOIN short_groups a ON g.name = a.name
|
|
WHERE g.first_record IS NOT NULL
|
|
AND g.first_record_postdate IS NOT NULL
|
|
AND g.backfill = 1
|
|
AND (NOW() - INTERVAL '.$backfilldays.' DAY ) < g.first_record_postdate
|
|
GROUP BY a.name, a.last_record, g.name, g.first_record
|
|
'.$orderby.' LIMIT 1';
|
|
|
|
$data = DB::select($sql);
|
|
|
|
$groupName = '';
|
|
$count = 0;
|
|
if (! empty($data) && isset($data[0]->name)) {
|
|
$groupName = $data[0]->name;
|
|
$count = ($data[0]->our_first - $data[0]->their_first);
|
|
}
|
|
|
|
if ($count <= 0) {
|
|
$this->headerNone();
|
|
if (config('nntmux.echocli') && $groupName !== '') {
|
|
cli()->primary('No backfill needed for group '.$groupName);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
$getEach = ($count > ($backfill_qty * $threads))
|
|
? (int) ceil(($backfill_qty * $threads) / $maxMessages)
|
|
: (int) ceil($count / $maxMessages);
|
|
|
|
$queues = [];
|
|
for ($i = 0; $i <= $getEach - 1; $i++) {
|
|
$queues[$i] = sprintf('get_range backfill %s %s %s %s', $groupName, $data[0]->our_first - $i * $maxMessages - $maxMessages, $data[0]->our_first - $i * $maxMessages - 1, $i + 1);
|
|
}
|
|
|
|
// Streaming mode
|
|
if ((bool) config('nntmux.stream_fork_output', false) === true) {
|
|
$commands = [];
|
|
foreach ($queues as $queue) {
|
|
$commands[] = $this->buildDnrCommand($queue);
|
|
}
|
|
$this->runStreamingCommands($commands, $threads, 'safe_backfill'); // @phpstan-ignore argument.type
|
|
|
|
return;
|
|
}
|
|
|
|
$this->headerStart('safe_backfill', count($queues), $threads);
|
|
|
|
// Build commands array for parallel execution
|
|
$commands = [];
|
|
foreach ($queues as $idx => $queue) {
|
|
$commands[$idx] = $this->buildDnrCommand($queue);
|
|
}
|
|
|
|
// Process using parallel commands with configurable timeout
|
|
$results = $this->runParallelCommands($commands, $threads);
|
|
|
|
foreach ($results as $idx => $output) {
|
|
echo $output;
|
|
cli()->primary('Backfilled group '.$groupName);
|
|
}
|
|
}
|
|
}
|