Cache adapter instances for current thread

flippercloud · Feb 3, 2024 · a3bdff2 · a3bdff2
1 parent 90808d8
commit a3bdff2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lib/flipper/adapters/connection_pool.rb b/lib/flipper/adapters/connection_pool.rb
@@ -18,8 +18,15 @@ def initialize(pool = nil, &adapter_initializer)
  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
+
+ private
+
+ def adapter_cache
+ Thread.current[:flipper_connection_pool_adapter_cache] ||= {}
+ end
 end