drupal/src/EventSubscriber/HttpLifecycleSubscriber.php at main · perfbaseorg/drupal · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
<?php

namespace Perfbase\Drupal\EventSubscriber;

use Perfbase\Drupal\Config\ConfigResolver;
use Perfbase\Drupal\Lifecycle\HttpRequestLifecycle;
use Perfbase\Drupal\Runtime\ActiveLifecycleRegistry;
use Perfbase\Drupal\Runtime\PerfbaseFactoryInterface;
use Perfbase\Drupal\Support\ErrorHandler;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;
use Symfony\Component\HttpKernel\Event\ExceptionEvent;
use Symfony\Component\HttpKernel\Event\RequestEvent;
use Symfony\Component\HttpKernel\Event\ResponseEvent;
use Symfony\Component\HttpKernel\Event\TerminateEvent;
use Symfony\Component\HttpKernel\KernelEvents;

/**
 *
 */
class HttpLifecycleSubscriber implements EventSubscriberInterface {
  private const KEY = 'http.main';

  private PerfbaseFactoryInterface $factory;

  private ConfigResolver $configResolver;

  private ErrorHandler $errorHandler;

  private ActiveLifecycleRegistry $registry;

  public function __construct(
    PerfbaseFactoryInterface $factory,
    ConfigResolver $configResolver,
    ErrorHandler $errorHandler,
    ActiveLifecycleRegistry $registry,
  ) {
    $this->factory = $factory;
    $this->configResolver = $configResolver;
    $this->errorHandler = $errorHandler;
    $this->registry = $registry;
  }

  /**
   * @return array<string, array<int, mixed>|string>
   */
  public static function getSubscribedEvents(): array {
    return [
      KernelEvents::REQUEST => ['onRequest', 1000],
      KernelEvents::RESPONSE => ['onResponse', -1000],
      KernelEvents::EXCEPTION => ['onException', -1000],
      KernelEvents::TERMINATE => ['onTerminate', -1000],
    ];
  }

  /**
   *
   */
  public function onRequest(RequestEvent $event): void {
    if (!$event->isMainRequest()) {
      return;
    }

    $lifecycle = new HttpRequestLifecycle(
          $event->getRequest(),
          $this->factory,
          $this->configResolver,
          $this->errorHandler
      );

    $this->registry->put(self::KEY, $lifecycle);
    $lifecycle->startProfiling();
  }

  /**
   *
   */
  public function onResponse(ResponseEvent $event): void {
    $lifecycle = $this->registry->get(self::KEY);
    if ($lifecycle instanceof HttpRequestLifecycle) {
      $lifecycle->setResponse($event->getResponse());
    }
  }

  /**
   *
   */
  public function onException(ExceptionEvent $event): void {
    $lifecycle = $this->registry->get(self::KEY);
    if ($lifecycle instanceof HttpRequestLifecycle) {
      $lifecycle->setException($event->getThrowable()->getMessage());
    }
  }

  /**
   *
   */
  public function onTerminate(TerminateEvent $event): void {
    $lifecycle = $this->registry->pull(self::KEY);
    if ($lifecycle instanceof HttpRequestLifecycle) {
      $lifecycle->setResponse($event->getResponse());
      $lifecycle->stopProfiling();
    }
  }

}