Merge pull request #3968 from rmosolgo/stack-trace-benchmark

Add a benchmark for stack trace depth

Author: rmosolgo

Rakefile

@@ -122,6 +122,12 @@ namespace :bench do
     prepare_benchmark
     GraphQLBenchmark.profile_schema_memory_footprint
   end
+
+  desc "Check the depth of the stacktrace during execution"
+  task :profile_stack_depth do
+    prepare_benchmark
+    GraphQLBenchmark.profile_stack_depth
+  end
 end
 
 namespace :test do

benchmark/run.rb

@@ -262,4 +262,79 @@ def self.profile_schema_memory_footprint
 
     report.pretty_print
   end
+
+  class StackDepthSchema < GraphQL::Schema
+    class Thing < GraphQL::Schema::Object
+      field :thing, self do
+        argument :lazy, Boolean, default_value: false
+      end
+
+      def thing(lazy:)
+        if lazy
+          -> { :something }
+        else
+          :something
+        end
+      end
+
+      field :stack_trace_depth, Integer do
+        argument :lazy, Boolean, default_value: false
+      end
+
+      def stack_trace_depth(lazy:)
+        get_depth = -> {
+          graphql_caller = caller.select { |c| c.include?("graphql") }
+          graphql_caller.size
+        }
+
+        if lazy
+          get_depth
+        else
+          get_depth.call
+        end
+      end
+    end
+
+    class Query < GraphQL::Schema::Object
+      field :thing, Thing
+
+      def thing
+        :something
+      end
+    end
+
+    query(Query)
+    lazy_resolve(Proc, :call)
+  end
+
+  def self.profile_stack_depth
+    query_str = <<-GRAPHQL
+    query($lazyThing: Boolean!, $lazyStackTrace: Boolean!) {
+      thing {
+        thing(lazy: $lazyThing) {
+          thing(lazy: $lazyThing) {
+            thing(lazy: $lazyThing) {
+              thing(lazy: $lazyThing) {
+                stackTraceDepth(lazy: $lazyStackTrace)
+              }
+            }
+          }
+        }
+      }
+    }
+    GRAPHQL
+
+    very_lazy_res = StackDepthSchema.execute(query_str, variables: { lazyThing: true, lazyStackTrace: true })
+    lazy_res = StackDepthSchema.execute(query_str, variables: { lazyThing: true, lazyStackTrace: false })
+    eager_res = StackDepthSchema.execute(query_str, variables: { lazyThing: false, lazyStackTrace: false })
+    get_depth = ->(result) { result["data"]["thing"]["thing"]["thing"]["thing"]["thing"]["stackTraceDepth"] }
+
+    puts <<~RESULT
+    Result         Depth
+    ---------------------
+    Eager          #{get_depth.call(eager_res)}
+    Lazy           #{get_depth.call(lazy_res)}
+    Very Lazy      #{get_depth.call(very_lazy_res)}
+    RESULT
+  end
 end