AC-1271: Add rate limiting for payment information endpoint and mutation

Author: arhiopterecs

app/code/Magento/GraphQl/Controller/GraphQl.php

@@ -24,8 +24,10 @@
 use Magento\Framework\GraphQl\Query\Resolver\ContextInterface;
 use Magento\Framework\GraphQl\Schema\SchemaGeneratorInterface;
 use Magento\Framework\Serialize\SerializerInterface;
+use Magento\Framework\Webapi\Exception;
 use Magento\Framework\Webapi\Response;
 use Magento\GraphQl\Helper\Query\Logger\LogData;
+use Magento\GraphQl\Model\Backpressure\GraphQlTooManyRequestsException;
 use Magento\GraphQl\Model\Query\ContextFactoryInterface;
 use Magento\GraphQl\Model\Query\Logger\LoggerPool;
 
@@ -194,6 +196,19 @@ public function dispatch(RequestInterface $request): ResponseInterface
                 $this->contextFactory->create(),
                 $data['variables'] ?? []
             );
+
+            if (isset($result['errors']) && is_array($result['errors'])) {
+                foreach ($result['errors'] as $error) {
+                    if (isset($error['extensions']) && is_array($error['extensions'])) {
+                        foreach ($error['extensions'] as $key => $value) {
+                            if (GraphQlTooManyRequestsException::EXCEPTION_CATEGORY === $value
+                                && 'category' === $key) {
+                                $statusCode = Exception::HTTP_TOO_MANY_REQUESTS;
+                            }
+                        }
+                    }
+                }
+            }
         } catch (\Exception $error) {
             $result['errors'] = isset($result['errors']) ? $result['errors'] : [];
             $result['errors'][] = $this->graphQlError->create($error);