Add redis queue status widget

This commit is contained in:
Ian Morland
2021-05-27 19:17:16 +01:00
parent d0d911303f
commit 49a6a88934
3 changed files with 96 additions and 0 deletions

View File

@@ -0,0 +1,73 @@
<?php
namespace GlowingBlue\RedisSetup\Provider;
use Flarum\Extend\Frontend;
use Flarum\Extension\ExtensionManager;
use Flarum\Foundation\AbstractServiceProvider;
use Flarum\Frontend\Document;
use Flarum\Settings\SettingsRepositoryInterface;
use Illuminate\Contracts\Cache\Store;
use Illuminate\Contracts\Events\Dispatcher;
use Illuminate\Queue\Events\Looping;
use Illuminate\Queue\RedisQueue;
use Illuminate\Contracts\Queue\Queue as QueueContract;
class QueueProvider extends AbstractServiceProvider
{
private $connection = 'default';
public function boot()
{
/** @var SettingsRepositoryInterface */
$settings = resolve(SettingsRepositoryInterface::class);
if (!(bool) $settings->get('glowingblue-redis.enableQueue', false)) {
return;
}
/** @var ExtensionManager $extensions */
$extensions = resolve(ExtensionManager::class);
(new Frontend('admin'))
->content([$this, 'adminWidgetAttributes'])
->extend($this->container, $extensions->getExtension('glowingblue-redis-setup'));
/** @var Dispatcher $dispatcher */
$dispatcher = $this->container->make(Dispatcher::class);
$dispatcher->listen(Looping::class, [$this, 'trackQueues']);
}
public function adminWidgetAttributes(Document $document)
{
/** @var Store $cache */
$cache = resolve('cache.store');
/** @var QueueContract $queue */
$queue = resolve(QueueContract::class);
$queues = $cache->get('blomstra.queue.queues-seen') ?? [];
if ($queue instanceof RedisQueue) {
$load = [];
foreach ($queues as $name) {
$load[$name] = $queue->getRedis()
->connection($this->connection)
->llen('queues:' . $name);
}
}
$document->payload['blomstraQueuesSeen'] = $queues;
$document->payload['blomstraQueuesLoad'] = $load ?? null;
}
public function trackQueues(Looping $event)
{
/** @var Store $cache */
$cache = resolve('cache.store');
$queues = $cache->get('blomstra.queue.queues-seen') ?? [];
$queues = array_merge($queues, (array) explode(',', $event->queue));
$cache->put('blomstra.queue.queues-seen', array_unique($queues), 60);
}
}