Cache adapter instances based on client

flippercloud · Feb 3, 2024 · 595c15f · 595c15f
1 parent 90808d8
commit 595c15f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/flipper/adapters/connection_pool.rb b/lib/flipper/adapters/connection_pool.rb
@@ -13,12 +13,14 @@ class Flipper::Adapters::ConnectionPool
   def initialize(pool = nil, &adapter_initializer)
     @pool = pool
     @adapter_initializer = adapter_initializer
+    @adapter_cache = {}
   end
 
   OPERATIONS.each do |method|
     define_method(method) do |*args|
       @pool.with do |connection|
-        @adapter_initializer.call(connection).public_send(method, *args)
+        @adapter_cache[connection] ||= @adapter_initializer.call(connection)
+        @adapter_cache[connection].public_send(method, *args)
       end
     end
   end