PerfbaseQueueWorkerManager.php

<?php

namespace Perfbase\Drupal\Queue;

use Drupal\Core\Cache\CacheBackendInterface;
use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\Core\Queue\QueueWorkerInterface;
use Drupal\Core\Queue\QueueWorkerManager;
use Perfbase\Drupal\Config\ConfigResolver;
use Perfbase\Drupal\Runtime\PerfbaseFactoryInterface;
use Perfbase\Drupal\Support\ErrorHandler;

/**
 * Queue worker manager with Perfbase profiling wrappers.
 */
class PerfbaseQueueWorkerManager extends QueueWorkerManager {
  private PerfbaseFactoryInterface $perfbaseFactory;

  private ConfigResolver $configResolver;

  private ErrorHandler $errorHandler;

  /**
   * @param \Traversable<string, string> $namespaces
   */
  public function __construct(
    \Traversable $namespaces,
    CacheBackendInterface $cache_backend,
    ModuleHandlerInterface $module_handler,
    PerfbaseFactoryInterface $factory,
    ConfigResolver $configResolver,
    ErrorHandler $errorHandler,
  ) {
    parent::__construct($namespaces, $cache_backend, $module_handler);
    $this->setCacheBackend($cache_backend, 'queue_plugins');
    $this->perfbaseFactory = $factory;
    $this->configResolver = $configResolver;
    $this->errorHandler = $errorHandler;
  }

  /**
   * {@inheritdoc}
   *
   * @param array<string, mixed> $configuration
   */
  public function createInstance($plugin_id, array $configuration = []): QueueWorkerInterface {
    /** @var QueueWorkerInterface $instance */
    $instance = parent::createInstance($plugin_id, $configuration);

    return $this->wrapInstance($instance, (string) $plugin_id);
  }

  /**
   * @return QueueWorkerInterface
   */
  protected function wrapInstance(QueueWorkerInterface $instance, string $pluginId): QueueWorkerInterface {
    return new ProfiledQueueWorker(
      $instance,
      $pluginId,
      $this->perfbaseFactory,
      $this->configResolver,
      $this->errorHandler
    );
  }

}