flipper-active_record 0.26.0 → 0.26.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 96994023a61be27a5bb30418c7bc367c82689ee4f964bb838996b84ff93e4725
-  data.tar.gz: 3fbd092338493aaa9da024517e35a2daef1c176526858b94bb93ccafa3c876a9
+  metadata.gz: 5244fa15983cbfcba71bcf8d555d71d3e9142314d035dc7dad084ea413d0abfd
+  data.tar.gz: 810a08e5c66d66765f8a09e5b4554804792200b0fa8084fd38bb12a2db44cb08
 SHA512:
-  metadata.gz: 2d1d74510fbba3f6b933c277c7747d2e0057e26aefb472889ef7dc463f80316cd7e8d0a9dd2f4d7916888b2bf40266ecb68dd84889f38efab59b058885d54c2c
-  data.tar.gz: f885cc037bfd91646b895056b54802ec512d56c1a670c98fa93d67d7c8de3b39bd1e5512ce630fc95e71c1db604ca69430a5aca1cf962d6de322d100c00a4ceb
+  metadata.gz: 20d45ea8dc75cbab5be320ea9ca22487f7a15483088c2370d48bb0e1b01e1fbf6cb657f9cd3f17ffd02c4fbeef1e20a80d3f328eb5a29d03961ddbfc87bd1439
+  data.tar.gz: 5d3871e886483ee2c9cefa90ef1fb918b4fd795c46121251b62c4c7543868578ba519f6f9547a58a749c87f4840e09496ef9c403f1765b91851be8bfd44c5d6b

data/lib/flipper/adapters/active_record.rb

@@ -53,18 +53,20 @@ module Flipper
 
       # Public: The set of known features.
       def features
-        @feature_class.all.map(&:key).to_set
+        with_connection(@feature_class) { @feature_class.all.map(&:key).to_set }
       end
 
       # Public: Adds a feature to the set of known features.
       def add(feature)
-        # race condition, but add is only used by enable/disable which happen
-        # super rarely, so it shouldn't matter in practice
-        @feature_class.transaction do
-          unless @feature_class.where(key: feature.key).first
-            begin
-              @feature_class.create! { |f| f.key = feature.key }
-            rescue ::ActiveRecord::RecordNotUnique
+        with_connection(@feature_class) do
+          # race condition, but add is only used by enable/disable which happen
+          # super rarely, so it shouldn't matter in practice
+          @feature_class.transaction do
+            unless @feature_class.where(key: feature.key).first
+              begin
+                @feature_class.create! { |f| f.key = feature.key }
+              rescue ::ActiveRecord::RecordNotUnique
+              end
             end
           end
         end
@@ -74,16 +76,18 @@ module Flipper
 
       # Public: Removes a feature from the set of known features.
       def remove(feature)
-        @feature_class.transaction do
-          @feature_class.where(key: feature.key).destroy_all
-          clear(feature)
+        with_connection(@feature_class) do
+          @feature_class.transaction do
+            @feature_class.where(key: feature.key).destroy_all
+            clear(feature)
+          end
         end
         true
       end
 
       # Public: Clears the gate values for a feature.
       def clear(feature)
-        @gate_class.where(feature_key: feature.key).destroy_all
+        with_connection(@gate_class) { @gate_class.where(feature_key: feature.key).destroy_all }
         true
       end
 
@@ -91,35 +95,39 @@ module Flipper
       #
       # Returns a Hash of Flipper::Gate#key => value.
       def get(feature)
-        db_gates = @gate_class.where(feature_key: feature.key)
+        db_gates = with_connection(@gate_class) { @gate_class.where(feature_key: feature.key) }
         result_for_feature(feature, db_gates)
       end
 
       def get_multi(features)
-        db_gates = @gate_class.where(feature_key: features.map(&:key))
-        grouped_db_gates = db_gates.group_by(&:feature_key)
-        result = {}
-        features.each do |feature|
-          result[feature.key] = result_for_feature(feature, grouped_db_gates[feature.key])
+        with_connection(@gate_class) do
+          db_gates = @gate_class.where(feature_key: features.map(&:key))
+          grouped_db_gates = db_gates.group_by(&:feature_key)
+          result = {}
+          features.each do |feature|
+            result[feature.key] = result_for_feature(feature, grouped_db_gates[feature.key])
+          end
+          result
         end
-        result
       end
 
       def get_all
-        features = ::Arel::Table.new(@feature_class.table_name.to_sym)
-        gates = ::Arel::Table.new(@gate_class.table_name.to_sym)
-        rows_query = features.join(gates, Arel::Nodes::OuterJoin)
-          .on(features[:key].eq(gates[:feature_key]))
-          .project(features[:key].as('feature_key'), gates[:key], gates[:value])
-        rows = @feature_class.connection.select_all rows_query
-        db_gates = rows.map { |row| @gate_class.new(row) }
-        grouped_db_gates = db_gates.group_by(&:feature_key)
-        result = Hash.new { |hash, key| hash[key] = default_config }
-        features = grouped_db_gates.keys.map { |key| Flipper::Feature.new(key, self) }
-        features.each do |feature|
-          result[feature.key] = result_for_feature(feature, grouped_db_gates[feature.key])
+        with_connection(@feature_class) do
+          features = ::Arel::Table.new(@feature_class.table_name.to_sym)
+          gates = ::Arel::Table.new(@gate_class.table_name.to_sym)
+          rows_query = features.join(gates, Arel::Nodes::OuterJoin)
+            .on(features[:key].eq(gates[:feature_key]))
+            .project(features[:key].as('feature_key'), gates[:key], gates[:value])
+          rows = @feature_class.connection.select_all rows_query
+          db_gates = rows.map { |row| @gate_class.new(row) }
+          grouped_db_gates = db_gates.group_by(&:feature_key)
+          result = Hash.new { |hash, key| hash[key] = default_config }
+          features = grouped_db_gates.keys.map { |key| Flipper::Feature.new(key, self) }
+          features.each do |feature|
+            result[feature.key] = result_for_feature(feature, grouped_db_gates[feature.key])
+          end
+          result
         end
-        result
       end
 
       # Public: Enables a gate for a given thing.
@@ -158,7 +166,9 @@ module Flipper
         when :integer
           set(feature, gate, thing)
         when :set
-          @gate_class.where(feature_key: feature.key, key: gate.key, value: thing.value).destroy_all
+          with_connection(@gate_class) do
+            @gate_class.where(feature_key: feature.key, key: gate.key, value: thing.value).destroy_all
+          end
         else
           unsupported_data_type gate.data_type
         end
@@ -175,18 +185,20 @@ module Flipper
 
       def set(feature, gate, thing, options = {})
         clear_feature = options.fetch(:clear, false)
-        @gate_class.transaction do
-          clear(feature) if clear_feature
-          @gate_class.where(feature_key: feature.key, key: gate.key).destroy_all
-          begin
-            @gate_class.create! do |g|
-              g.feature_key = feature.key
-              g.key = gate.key
-              g.value = thing.value.to_s
+        with_connection(@gate_class) do
+          @gate_class.transaction do
+            clear(feature) if clear_feature
+            @gate_class.where(feature_key: feature.key, key: gate.key).destroy_all
+            begin
+              @gate_class.create! do |g|
+                g.feature_key = feature.key
+                g.key = gate.key
+                g.value = thing.value.to_s
+              end
+            rescue ::ActiveRecord::RecordNotUnique
+              # assume this happened concurrently with the same thing and its fine
+              # see https://github.com/jnunemaker/flipper/issues/544
             end
-          rescue ::ActiveRecord::RecordNotUnique
-            # assume this happened concurrently with the same thing and its fine
-            # see https://github.com/jnunemaker/flipper/issues/544
           end
         end
 
@@ -194,10 +206,12 @@ module Flipper
       end
 
       def enable_multi(feature, gate, thing)
-        @gate_class.create! do |g|
-          g.feature_key = feature.key
-          g.key = gate.key
-          g.value = thing.value.to_s
+        with_connection(@gate_class) do
+          @gate_class.create! do |g|
+            g.feature_key = feature.key
+            g.key = gate.key
+            g.value = thing.value.to_s
+          end
         end
 
         nil
@@ -227,6 +241,10 @@ module Flipper
         end
         result
       end
+
+      def with_connection(model = @feature_class, &block)
+        model.connection_pool.with_connection(&block)
+      end
     end
   end
 end

data/lib/flipper/version.rb

@@ -1,3 +1,3 @@
 module Flipper
-  VERSION = '0.26.0'.freeze
+  VERSION = '0.26.1'.freeze
 end

data/spec/flipper/adapters/active_record_spec.rb

@@ -57,4 +57,126 @@ RSpec.describe Flipper::Adapters::ActiveRecord do
       expect(Flipper.adapter.adapter).to be_a(Flipper::Adapters::ActiveRecord)
     end
   end
+
+  context "ActiveRecord connection_pool" do
+    before do
+      ActiveRecord::Base.clear_active_connections!
+    end
+
+    context "#features" do
+      it "does not hold onto connections" do
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.features
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+      end
+
+      it "does not release previously held connection" do
+        ActiveRecord::Base.connection # establish a new connection
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.features
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+      end
+    end
+
+    context "#get_all" do
+      it "does not hold onto connections" do
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.get_all
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+      end
+
+      it "does not release previously held connection" do
+        ActiveRecord::Base.connection # establish a new connection
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.get_all
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+      end
+    end
+
+    context "#add / #remove / #clear" do
+      let(:feature) { Flipper::Feature.new(:search, subject) }
+
+      it "does not hold onto connections" do
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.add(feature)
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.remove(feature)
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.clear(feature)
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+      end
+
+      it "does not release previously held connection" do
+        ActiveRecord::Base.connection # establish a new connection
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.add(feature)
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.remove(feature)
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.clear(feature)
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+      end
+    end
+
+    context "#get_multi" do
+      let(:feature) { Flipper::Feature.new(:search, subject) }
+
+      it "does not hold onto connections" do
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.get_multi([feature])
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+      end
+
+      it "does not release previously held connection" do
+        ActiveRecord::Base.connection # establish a new connection
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.get_multi([feature])
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+      end
+    end
+
+    context "#enable/#disable boolean" do
+      let(:feature) { Flipper::Feature.new(:search, subject) }
+      let(:gate) { feature.gate(:boolean)}
+
+      it "does not hold onto connections" do
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.enable(feature, gate, gate.wrap(true))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.disable(feature, gate, gate.wrap(false))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+      end
+
+      it "does not release previously held connection" do
+        ActiveRecord::Base.connection # establish a new connection
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.enable(feature, gate, gate.wrap(true))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.disable(feature, gate, gate.wrap(false))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+      end
+    end
+
+    context "#enable/#disable set" do
+      let(:feature) { Flipper::Feature.new(:search, subject) }
+      let(:gate) { feature.gate(:group) }
+
+      it "does not hold onto connections" do
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.enable(feature, gate, gate.wrap(:admin))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+        subject.disable(feature, gate, gate.wrap(:admin))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(false)
+      end
+
+      it "does not release previously held connection" do
+        ActiveRecord::Base.connection # establish a new connection
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.enable(feature, gate, gate.wrap(:admin))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+        subject.disable(feature, gate, gate.wrap(:admin))
+        expect(ActiveRecord::Base.connection_handler.active_connections?).to be(true)
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: flipper-active_record
 version: !ruby/object:Gem::Version
-  version: 0.26.0
+  version: 0.26.1
 platform: ruby
 authors:
 - John Nunemaker
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-12-05 00:00:00.000000000 Z
+date: 2023-03-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: flipper
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.26.0
+        version: 0.26.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.26.0
+        version: 0.26.1
 - !ruby/object:Gem::Dependency
   name: activerecord
   requirement: !ruby/object:Gem::Requirement