graphql: cache graphql validation

Author: kamilkisiela

graphql/src/execution/query.rs

@@ -2,12 +2,13 @@ use graph::data::graphql::DocumentExt as _;
 use graph::data::value::Object;
 use graphql_parser::Pos;
 use graphql_tools::validation::rules::*;
+use graphql_tools::validation::utils::ValidationError;
 use graphql_tools::validation::validate::{validate, ValidationPlan};
 use lazy_static::lazy_static;
 use std::collections::{BTreeMap, HashMap, HashSet};
 use std::hash::{Hash, Hasher};
 use std::iter::FromIterator;
-use std::sync::Arc;
+use std::sync::{Arc, Mutex, PoisonError};
 use std::time::Instant;
 use std::{collections::hash_map::DefaultHasher, convert::TryFrom};
 
@@ -25,6 +26,11 @@ use crate::schema::ast::{self as sast};
 use crate::values::coercion;
 use crate::{execution::get_field, schema::api::ErrorPolicy};
 
+lazy_static! {
+    static ref GRAPHQL_VALIDATION_CACHE: Mutex<HashMap<u64, Vec<ValidationError>>> =
+        Mutex::new(HashMap::<u64, Vec<ValidationError>>::new());
+}
+
 lazy_static! {
     static ref GRAPHQL_VALIDATION_PLAN: ValidationPlan =
         ValidationPlan::from(if !ENV_VARS.graphql.enable_validations {
@@ -150,8 +156,17 @@ impl Query {
         max_complexity: Option<u64>,
         max_depth: u8,
     ) -> Result<Arc<Self>, Vec<QueryExecutionError>> {
-        let validation_errors =
-            validate(schema.document(), &query.document, &GRAPHQL_VALIDATION_PLAN);
+        GRAPHQL_VALIDATION_CACHE
+            .lock()
+            .unwrap_or_else(PoisonError::into_inner)
+            .entry(query.shape_hash)
+            .or_insert_with(|| {
+                validate(
+                    &schema.document(),
+                    &query.document,
+                    &GRAPHQL_VALIDATION_PLAN,
+                )
+            });
 
         if !validation_errors.is_empty() {
             if !ENV_VARS.graphql.silent_graphql_validations {
@@ -160,7 +175,7 @@ impl Query {
                     .map(|e| {
                         QueryExecutionError::ValidationError(
                             e.locations.first().cloned(),
-                            e.message,
+                            e.message.clone(),
                         )
                     })
                     .collect());