feat: add APQ plugin

Author: chrfritsch

graphql.services.yml

@@ -39,6 +39,14 @@ services:
     tags:
       - { name: cache.context }
 
+  # Cache bin for the persisted queries.
+  cache.graphql.apq:
+    class: Drupal\Core\Cache\CacheBackendInterface
+    tags:
+      - { name: cache.bin }
+    factory: cache_factory:get
+    arguments: [graphql_apq]
+
   # Cache bin for the parsed sdl ast.
   cache.graphql.ast:
     class: Drupal\Core\Cache\CacheBackendInterface
@@ -108,6 +116,13 @@ services:
     tags:
       - { name: event_subscriber }
 
+  # Cache the queries to be persistent.
+  graphql.apq_subscriber:
+    class: Drupal\graphql\EventSubscriber\ApqSubscriber
+    arguments: ['@cache.graphql.apq', '@request_stack']
+    tags:
+      - { name: event_subscriber }
+
   # Plugin manager for schemas
   plugin.manager.graphql.schema:
     class: Drupal\graphql\Plugin\SchemaPluginManager

src/EventSubscriber/ApqSubscriber.php

@@ -0,0 +1,70 @@
+<?php
+
+namespace Drupal\graphql\EventSubscriber;
+
+use Drupal\Component\Serialization\Json;
+use Drupal\Core\Cache\CacheBackendInterface;
+use Drupal\graphql\Event\OperationEvent;
+use Symfony\Component\EventDispatcher\EventSubscriberInterface;
+use Symfony\Component\HttpFoundation\RequestStack;
+
+/**
+ * Save persisted queries to cache.
+ */
+class ApqSubscriber implements EventSubscriberInterface {
+
+  /**
+   * The cache to store persisted queries.
+   *
+   * @var \Drupal\Core\Cache\CacheBackendInterface
+   */
+  protected $cache;
+
+  /**
+   * The current request.
+   *
+   * @var \Symfony\Component\HttpFoundation\Request|null
+   */
+  protected $request;
+
+  /**
+   * Constructs a ApqSubscriber object.
+   *
+   * @param \Drupal\Core\Cache\CacheBackendInterface $cache
+   *   The cache to store persisted queries.
+   * @param \Symfony\Component\HttpFoundation\RequestStack $requestStack
+   *   The request stack.
+   */
+  public function __construct(CacheBackendInterface $cache, RequestStack $requestStack) {
+    $this->cache = $cache;
+    $this->request = $requestStack->getCurrentRequest();
+  }
+
+  /**
+   * Handle operation start events.
+   *
+   * @param \Drupal\graphql\Event\OperationEvent $event
+   *   The kernel event object.
+   */
+  public function onBeforeOperation(OperationEvent $event): void {
+    try {
+      $json = Json::decode($this->request->getContent());
+      if (!empty($json['extensions']['persistedQuery']['sha256Hash']) && !empty($json['query'])) {
+        $this->cache->set($json['extensions']['persistedQuery']['sha256Hash'], $json['query']);
+      }
+
+    }
+    catch (\Exception $exception) {
+    }
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public static function getSubscribedEvents() {
+    return [
+      OperationEvent::GRAPHQL_OPERATION_BEFORE => 'onBeforeOperation',
+    ];
+  }
+
+}

src/Plugin/GraphQL/PersistedQuery/AutomaticPersistedQuery.php

@@ -0,0 +1,56 @@
+<?php
+
+namespace Drupal\graphql\Plugin\GraphQL\PersistedQuery;
+
+use Drupal\Core\Cache\CacheBackendInterface;
+use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
+use Drupal\graphql\PersistedQuery\PersistedQueryPluginBase;
+use GraphQL\Server\OperationParams;
+use GraphQL\Server\RequestError;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * Load persisted queries from the cache.
+ *
+ * @PersistedQuery(
+ *   id = "automatic_persisted_query",
+ *   label = "Automatic Persisted Query",
+ *   description = "Load persisted queries from the cache."
+ * )
+ */
+class AutomaticPersistedQuery extends PersistedQueryPluginBase implements ContainerFactoryPluginInterface {
+
+  /**
+   * The cache to store persisted queries.
+   *
+   * @var \Drupal\Core\Cache\CacheBackendInterface
+   */
+  protected $cache;
+
+  /**
+   * {@inheritdoc}
+   */
+  public function __construct(array $configuration, $plugin_id, $plugin_definition, CacheBackendInterface $cache) {
+    parent::__construct($configuration, $plugin_id, $plugin_definition);
+
+    $this->cache = $cache;
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
+    return new static($configuration, $plugin_id, $plugin_definition, $container->get('cache.graphql.apq'));
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public function getQuery($id, OperationParams $operation) {
+    if ($query = $this->cache->get($id)) {
+      return $query->data;
+    }
+    throw new RequestError('PersistedQueryNotFound');
+  }
+
+}