Merge pull request #160 from Shopify/add-wrapper-around-perform

Add a custom wrapper around #perform
Shopify · May 16, 2023 · 3b22366 · 3b22366
2 parents e476f7c + c43bfc4
commit 3b22366
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/lib/graphql/batch/loader.rb b/lib/graphql/batch/loader.rb
@@ -66,12 +66,21 @@ def resolve #:nodoc:
       return if resolved?
       load_keys = queue
       @queue = nil
-      perform(load_keys)
+
+      around_perform do
+        perform(load_keys)
+      end
+
       check_for_broken_promises(load_keys)
     rescue => err
       reject_pending_promises(load_keys, err)
     end
 
+    # Interface to add custom code for purposes such as instrumenting the performance of the loader.
+    def around_perform
+      yield
+    end
+
     # For Promise#sync
     def wait #:nodoc:
       if executor

diff --git a/lib/graphql/batch/version.rb b/lib/graphql/batch/version.rb
@@ -1,5 +1,5 @@
 module GraphQL
   module Batch
-    VERSION = "0.5.2"
+    VERSION = "0.5.3"
   end
 end