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

Add a custom wrapper around #perform

Author: toneymathews

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

lib/graphql/batch/version.rb

@@ -1,5 +1,5 @@
 module GraphQL
   module Batch
-    VERSION = "0.5.2"
+    VERSION = "0.5.3"
   end
 end