Use ActiveModel::Callbacks on the SerializableResource

lib/active_model/serializable_resource.rb

@@ -2,6 +2,15 @@
 module ActiveModel
   class SerializableResource
     ADAPTER_OPTION_KEYS = Set.new([:include, :fields, :adapter, :meta, :meta_key, :links])
+    extend ActiveModel::Callbacks
+
+    define_model_callbacks :render
+
+    around_render do |_, block, _|
+      notify_active_support do
+        block.call
+      end
+    end
 
     # Primary interface to composing a resource with a serializer and adapter.
     # @return the serializable_resource, ready for #as_json/#to_json/#serializable_hash.
@@ -12,19 +21,19 @@ def initialize(resource, options = {})
     end
 
     def serializable_hash(*args)
-      notify_active_support do
+      run_callbacks :render do
         adapter.serializable_hash(*args)
       end
     end
 
     def as_json(*args)
-      notify_active_support do
+      run_callbacks :render do
         adapter.as_json(*args)
       end
     end
 
     def to_json(*args)
-      notify_active_support do
+      run_callbacks :render do
         adapter.to_json(*args)
       end
     end