fix(current-usage): Manage connection pools with parallel threads

getlago · May 28, 2024 · 49dbef5 · 49dbef5
1 parent f957323
commit 49dbef5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app/services/invoices/customer_usage_service.rb b/app/services/invoices/customer_usage_service.rb
@@ -67,7 +67,9 @@ def add_charge_fees
         .order(Arel.sql('lower(unaccent(billable_metrics.name)) ASC'))
 
       invoice.fees = Parallel.flat_map(query.all, in_threads: ENV['LAGO_PARALLEL_THREADS_COUNT']&.to_i || 1) do |charge|
-        charge_usage(charge)
+        ActiveRecord::Base.connection_pool.with_connection do
+          charge_usage(charge)
+        end
       end
     end