bpolaszek / mercure-php-hub

src/Command/StressSubscribersCommand.php

<?php

namespace BenTools\MercurePHP\Command;

use Lcobucci\JWT\Builder;
use Lcobucci\JWT\Signer\Hmac\Sha256;
use Lcobucci\JWT\Signer\Key;
use RingCentral\Psr7\Uri;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Style\SymfonyStyle;
use Symfony\Component\HttpClient\Exception\TimeoutException;
use Symfony\Component\HttpClient\Exception\TransportException;
use Symfony\Component\HttpClient\HttpClient;
use Symfony\Contracts\HttpClient\HttpClientInterface;

final class StressSubscribersCommand extends Command
{
    protected static $defaultName = 'mercure:stress:subscribers';

    protected function execute(InputInterface $input, OutputInterface $output): int
    {
        $output = new SymfonyStyle($input, $output);
        $start = time();

        if (null === $input->getOption('jwt-key')) {
            $output->error('No jwt-key provided.');

            return 1;
        }

        if (null === $input->getOption('topics')) {
            $output->error('No topics provided.');

            return 1;
        }

        $jwt = (new Builder())
            ->withClaim('mercure', ['subscribe' => ['*']])
            ->getToken(new Sha256(), new Key($input->getOption('jwt-key')));

It seems like $input->getOption('jwt-key') can also be of type string[]; however, parameter $content of Lcobucci\JWT\Signer\Key::__construct() does only seem to accept string, maybe add an additional type check?

        $client = HttpClient::create(
            [
                'http_version' => '2.0',
                'verify_peer' => false,
                'verify_host' => false,
                'headers' => [
                    'Authorization' => 'Bearer ' . $jwt,
                ],
            ]
        );

        $url = new Uri($input->getArgument('url'));

It seems like $input->getArgument('url') can also be of type string[]; however, parameter $uri of RingCentral\Psr7\Uri::__construct() does only seem to accept string, maybe add an additional type check?

        $qs = '';
        $topics = \explode(',', $input->getOption('topics'));
        foreach ($topics as $topic) {
            $qs .= '&topic=' . $topic;
        }

        $url = $url->withQuery($qs);
        $nbSubscribers = (int) $input->getOption('subscribers');

        $requests = function (HttpClientInterface $client, string $url) use ($nbSubscribers) {
            for ($i = 0; $i < $nbSubscribers; $i++) {
                yield $client->request('GET', $url);
            }
        };

        $duration = (int) $input->getOption('duration');
        $times = [];
        try {
            foreach ($client->stream($requests($client, $url), $duration) as $response => $chunk) {
                $now = \time();
                $times[$now] ??= 0;
                $times[$now]++;

                if ($now >= ($start + $duration)) {
                    $response->cancel();
                }
            }
        } catch (TimeoutException | TransportException $e) {

Consider adding a comment why this CATCH block is empty.

        }

        $average = (int) \round(\array_sum($times) / \count($times));
        $output->success(\sprintf('Average messages /sec: %d', $average));

        return 0;
    }

    protected function configure(): void
    {
        $this->setDescription('(Testing purposes) Simulates n subscribers on a Mercure Hub.');
        $this->addArgument('url', InputArgument::REQUIRED, 'Mercure Hub url.');
        $this->addOption('jwt-key', null, InputOption::VALUE_REQUIRED, 'Subscriber JWT KEY.');
        $this->addOption('topics', null, InputOption::VALUE_REQUIRED, 'Topics to subscribe, comma-separated.');
        $this->addOption('duration', null, InputOption::VALUE_OPTIONAL, 'Duration of the test, in seconds.', 10);
        $this->addOption('subscribers', null, InputOption::VALUE_OPTIONAL, 'Number of subscribers', 100);
    }
}