Add Timeout Logic

app/controllers/graphql_controller.rb

@@ -1,14 +1,20 @@
 class GraphqlController < ApplicationController
+  GRAPHQL_TIMEOUT = 10
+
   before_action :authenticate
 
   def execute
     query_string = params[:query].to_s
     variables = ensure_hash(params[:variables])
-    context = {
-      user: @user
-    }
+    context = { user: @user }
+
+    result = Graph.query(
+      query_string,
+      variables: variables,
+      context: context,
+      timeout: GRAPHQL_TIMEOUT
+    )
 
-    result = Graph::Schema.execute(query_string, variables: variables, context: context)
     render json: result
   end
 

app/models/graph.rb

@@ -1,5 +1,13 @@
+require 'timeout'
+
 module Graph
   class << self
+    def query(query_string, variables: {}, context: {}, timeout: nil)
+      Timeout.timeout(timeout) do
+        Graph::Schema.execute(query_string, variables: variables, context: context)
+      end
+    end
+
     def find_by_id_field(type, model)
       GraphQL::Field.define do
         type type