graphql: Add complexity value to query logs

Author: fordN

graphql/src/execution/query.rs

@@ -42,6 +42,7 @@ pub struct Query {
     /// have dummy values
     pub(crate) query_text: Arc<String>,
     pub(crate) variables_text: Arc<String>,
+    pub(crate) complexity: u64,
 }
 
 impl Query {
@@ -88,7 +89,7 @@ impl Query {
             q::OperationDefinition::Query(q::Query { selection_set, .. }) => {
                 (Kind::Query, selection_set)
             }
-            // Queries can be run by just sending a selction set
+            // Queries can be run by just sending a selection set
             q::OperationDefinition::SelectionSet(selection_set) => (Kind::Query, selection_set),
             q::OperationDefinition::Subscription(q::Subscription { selection_set, .. }) => {
                 (Kind::Subscription, selection_set)
@@ -100,20 +101,21 @@ impl Query {
             }
         };
 
-        let query = Arc::new(Self {
+        let mut query = Self {
             schema: query.schema,
             variables,
             fragments,
             selection_set,
             kind,
             query_text,
             variables_text,
-        });
+            complexity: 0,
+        };
 
         query.validate_fields()?;
         query.check_complexity(max_complexity, max_depth)?;
 
-        Ok(query)
+        Ok(Arc::new(query))
     }
 
     /// Return the block constraint for the toplevel query field(s) Since,
@@ -161,6 +163,7 @@ impl Query {
             kind: self.kind,
             query_text: self.query_text.clone(),
             variables_text: self.variables_text.clone(),
+            complexity: self.complexity,
         })
     }
 
@@ -186,7 +189,7 @@ impl Query {
     }
 
     fn check_complexity(
-        &self,
+        &mut self,
         max_complexity: Option<u64>,
         max_depth: u8,
     ) -> Result<(), Vec<QueryExecutionError>> {
@@ -198,6 +201,7 @@ impl Query {
                     max_complexity,
                 )]);
             }
+            self.complexity = complexity;
         }
         Ok(())
     }

graphql/src/query/mod.rs

@@ -86,6 +86,7 @@ where
             "query_time_ms" => start.elapsed().as_millis(),
             "cached" => ctx.cached.load(std::sync::atomic::Ordering::SeqCst),
             "block" => block_ptr.map(|b| b.number).unwrap_or(0),
+            "complexity" => &query.complexity
         );
     }
     result