lib/arel/middleware/database_executor.rb in arel_toolkit-0.4.2 vs lib/arel/middleware/database_executor.rb in arel_toolkit-0.4.3

@@ -9,16 +9,16 @@
 
       def initialize(middleware)
         @middleware = middleware
       end
 
-      def run(enhanced_arel, context, final_block)
+      def run(arel, context, final_block)
         @index = 0
         @context = context
         @final_block = final_block
 
-        result = call(enhanced_arel)
+        result = call(arel)
         check_return_type result
         result
       ensure
         @index = 0
         @context = nil
@@ -44,10 +44,19 @@
 
       private
 
       def execute_sql(next_arel)
         sql, binds = next_arel.to_sql_and_binds
+
+        context[:cache_accessor].write(
+          transformed_sql: sql,
+          transformed_binds: binds,
+          original_sql: context[:original_sql],
+          original_binds: context[:original_binds],
+        )
+
         sql_result = final_block.call(sql, binds)
+
         check_return_type sql_result
         sql_result
       end
 
       def check_argument_type(next_arel)