composite_primary_keys 12.0.9 → 13.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2465cca0fa83a628e24f3221e88095690295c99b2bf46818bd2ebd76df8300b5
-  data.tar.gz: a22c2b61575c26e23e51b314c05c158663c26ce7a9c64fb56019d596edc6a844
+  metadata.gz: eeb7ac69a866392d1d19aec7ae67f99af052e72e5585375c2cb73990f442cc28
+  data.tar.gz: 6552f1e74d6ab9a40aadd66ea78e8d6c489e6d793a47cc52ab2582f60dfbced7
 SHA512:
-  metadata.gz: 3147006b49f98c3d1e88dbec56590fad63c13506634427361e14e06d611e75666c89c48c3e9151d174dc075d850a238285be6bc62739d8971d7bf190bc6e9db9
-  data.tar.gz: 34d811c5ab6afb7d5b4b9dc871b2df660f42abaf146ffc7faa60c9d4d2edfb8c1a8fa6e49b40488a6090a7bb64218fafb6b0e965744b4daa0d11c0fc34605abf
+  metadata.gz: 1b0bd53df9177d3569e4646124ff2aaf6bacf4d83a78f287a9e954ed388a37ed4872f4024204e34f7402b7989b401586c53636a551d745e5aaf4871927bb66fb
+  data.tar.gz: 8c2eacaaff041c6c4255aed0402f63038d6e0d781665208cbb9b3bdbcc86666087f69822400384f9dfc44215c23be46ddc1c3c69b7ee0008db636d8ff1e0a29c

data/History.rdoc

@@ -1,3 +1,17 @@
+== 13.0.2 (2022-01-09)
+* Fix scoped associations take #2 (Charlie Savage)
+
+== 13.0.1 (2021-11-14)
+* Fix invalid sql generation for some cases of scoped associations (Ryan Mulligan)
+* Fix unintentional connection to database (Kazuhiro Masuda)
+* Zip values then keys - fixes #548 (Charlie Savage)
+
+== 13.0.0 (2021-05-09)
+* Update to ActiveRecord 6.1 (Javier Julio, Charlie Savage, Sammy Larbi)
+
+== 12.0.10 (2021-05-09)
+* Support using the #id method on records with primary keys that also have an :id attribute.
+
 == 12.0.9 (2021-02-22)
 * Third time is hopefully the charm on MySQL/MariaDB auto increment fix
 
@@ -511,7 +525,7 @@ by replacing quoted identifiers with all-caps equivalents on Oracle (Rhett Sutph
 * Update Oracle tests (Rhett Sutphin)
 
 == 4.1.1 2011-08-31
-* Support for AR 3.1.1 
+* Support for AR 3.1.1
 * Make polymorphic belongs_to work in rails 3.1.1 (Tom Hughes)
 * Eliminate relative paths from the test suite (Rhett Sutphin)
 * Minor improvements to the CPK test runner w/o relative path changes (Rhett Sutphin)

data/README.rdoc

@@ -20,6 +20,7 @@ Every major version of ActiveRecord has included numerous internal changes.  As
 CPK has to be rewritten for each version of ActiveRecord.  To help keep
 things straight, here is the mapping:
 
+    Version 13.x is designed to work with ActiveRecord 6.1.x
     Version 12.x is designed to work with ActiveRecord 6.0.x
     Version 11.x is designed to work with ActiveRecord 5.2.x
     Version 10.x is designed to work with ActiveRecord 5.1.x

data/Rakefile

@@ -1,37 +1,37 @@
-require 'rubygems'
-require 'rake'
-require 'rake/clean'
-require 'rake/testtask'
-require 'rubygems/package_task'
-
-# Set global variable so other tasks can access them
-::PROJECT_ROOT = File.expand_path(".")
-::GEM_NAME = 'composite_primary_keys'
-
-require File.join(PROJECT_ROOT, 'lib', 'composite_primary_keys')
-require File.join(PROJECT_ROOT, 'test', 'connections', 'connection_spec')
-
-# Read the spec file
-spec = Gem::Specification.load("#{GEM_NAME}.gemspec")
-
-# Setup Rake tasks for managing the gem
-Gem::PackageTask.new(spec).define
-
-# Now load in other task files
-Dir.glob('tasks/**/*.rake').each do |rake_file|
-  load File.join(File.dirname(__FILE__), rake_file)
-end
-
-# Set up test tasks for each supported connection adapter
-%w(mysql sqlite oracle oracle_enhanced postgresql ibm_db sqlserver).each do |adapter|
-  namespace adapter do
-    desc "Run tests using the #{adapter} adapter"
-    task "test" do
-      ENV["ADAPTER"] = adapter
-      Rake::TestTask.new("subtest_#{adapter}") do |t|
-        t.libs << "test"
-      end
-      Rake::Task["subtest_#{adapter}"].invoke
-    end
-  end
-end
+require 'rubygems'
+require 'rake'
+require 'rake/clean'
+require 'rake/testtask'
+require 'rubygems/package_task'
+
+# Set global variable so other tasks can access them
+::PROJECT_ROOT = File.expand_path(".")
+::GEM_NAME = 'composite_primary_keys'
+
+require File.join(PROJECT_ROOT, 'lib', 'composite_primary_keys')
+require File.join(PROJECT_ROOT, 'test', 'connections', 'connection_spec')
+
+# Read the spec file
+spec = Gem::Specification.load("#{GEM_NAME}.gemspec")
+
+# Setup Rake tasks for managing the gem
+Gem::PackageTask.new(spec).define
+
+# Now load in other task files
+Dir.glob('tasks/**/*.rake').each do |rake_file|
+  load File.join(File.dirname(__FILE__), rake_file)
+end
+
+# Set up test tasks for each supported connection adapter
+%w(mysql sqlite oracle oracle_enhanced postgresql ibm_db sqlserver).each do |adapter|
+  namespace adapter do
+    desc "Run tests using the #{adapter} adapter"
+    task "test" do
+      ENV["ADAPTER"] = adapter
+      Rake::TestTask.new("subtest_#{adapter}") do |t|
+        t.libs << "test"
+      end
+      Rake::Task["subtest_#{adapter}"].invoke
+    end
+  end
+end

data/lib/composite_primary_keys/active_model/attribute_assignment.rb

@@ -1,19 +1,19 @@
-module ActiveModel
-  module AttributeAssignment
-    def _assign_attribute(k, v)
-      # CPK. This is super ugly, but if a table has a composite key where one of the fields is named :id we need
-      # to handle it as a single value. Otherwise, we would call the id=(value) method which is expecting
-      # and array of values.
-      if k == 'id' && self.kind_of?(ActiveRecord::Base) && self.composite? && !self.column_for_attribute(k).null
-        self._write_attribute(k, v)
-      else
-        setter = :"#{k}="
-        if respond_to?(setter)
-          public_send(setter, v)
-        else
-          raise UnknownAttributeError.new(self, k)
-        end
-      end
-    end
-  end
-end
+module ActiveModel
+  module AttributeAssignment
+    def _assign_attribute(k, v)
+      # CPK. This is super ugly, but if a table has a composite key where one of the fields is named :id we need
+      # to handle it as a single value. Otherwise, we would call the id=(value) method which is expecting
+      # and array of values.
+      if k == 'id' && !v.kind_of?(Array) && self.kind_of?(ActiveRecord::Base) && self.composite? && !self.column_for_attribute(k).null
+        self._write_attribute(k, v)
+      else
+        setter = :"#{k}="
+        if respond_to?(setter)
+          public_send(setter, v)
+        else
+          raise UnknownAttributeError.new(self, k)
+        end
+      end
+    end
+  end
+end

data/lib/composite_primary_keys/arel/sqlserver.rb

@@ -1,37 +1,37 @@
-module Arel
-  module Visitors
-    class SQLServer < Arel::Visitors::ToSql
-      def make_Fetch_Possible_And_Deterministic o
-        return if o.limit.nil? && o.offset.nil?
-        t = table_From_Statement o
-        pk = primary_Key_From_Table t
-        return unless pk
-        if o.orders.empty?
-          # Prefer deterministic vs a simple `(SELECT NULL)` expr.
-          # CPK
-          #o.orders = [pk.asc]
-          o.orders = pk.map {|a_pk| a_pk.asc}
-        end
-      end
-
-      def primary_Key_From_Table t
-        return unless t
-        column_name = @connection.schema_cache.primary_keys(t.name) ||
-          @connection.schema_cache.columns_hash(t.name).first.try(:second).try(:name)
-
-        # CPK
-        # column_name ? t[column_name] : nil
-        case column_name
-          when Array
-            column_name.map do |name|
-              t[name]
-            end
-          when NilClass
-            nil
-          else
-            [t[column_name]]
-        end
-      end
-    end
-  end
-end
+module Arel
+  module Visitors
+    class SQLServer < Arel::Visitors::ToSql
+      def make_Fetch_Possible_And_Deterministic o
+        return if o.limit.nil? && o.offset.nil?
+        t = table_From_Statement o
+        pk = primary_Key_From_Table t
+        return unless pk
+        if o.orders.empty?
+          # Prefer deterministic vs a simple `(SELECT NULL)` expr.
+          # CPK
+          #o.orders = [pk.asc]
+          o.orders = pk.map {|a_pk| a_pk.asc}
+        end
+      end
+
+      def primary_Key_From_Table t
+        return unless t
+        column_name = @connection.schema_cache.primary_keys(t.name) ||
+          @connection.schema_cache.columns_hash(t.name).first.try(:second).try(:name)
+
+        # CPK
+        # column_name ? t[column_name] : nil
+        case column_name
+          when Array
+            column_name.map do |name|
+              t[name]
+            end
+          when NilClass
+            nil
+          else
+            [t[column_name]]
+        end
+      end
+    end
+  end
+end

data/lib/composite_primary_keys/arel/to_sql.rb

@@ -1,18 +1,18 @@
-module Arel
-  module Visitors
-    class ToSql
-      def visit_CompositePrimaryKeys_CompositeKeys o, collector
-        values = o.map do |key|
-          case key
-            when Arel::Attributes::Attribute
-              "#{key.relation.name}.#{key.name}"
-            else
-              key
-          end
-        end
-        collector << "(#{values.join(', ')})"
-        collector
-      end
-    end
-  end
-end
+module Arel
+  module Visitors
+    class ToSql
+      def visit_CompositePrimaryKeys_CompositeKeys o, collector
+        values = o.map do |key|
+          case key
+            when Arel::Attributes::Attribute
+              "#{key.relation.name}.#{key.name}"
+            else
+              key
+          end
+        end
+        collector << "(#{values.join(', ')})"
+        collector
+      end
+    end
+  end
+end

data/lib/composite_primary_keys/associations/association.rb

@@ -1,23 +1,23 @@
-module ActiveRecord
-  module Associations
-    class Association
-      def creation_attributes
-        attributes = {}
-
-        if (reflection.has_one? || reflection.collection?) && !options[:through]
-          # CPK
-          # attributes[reflection.foreign_key] = owner[reflection.active_record_primary_key]
-          Array(reflection.foreign_key).zip(Array(reflection.active_record_primary_key)).each do |key1, key2|
-            attributes[key1] = owner[key2]
-          end
-
-          if reflection.options[:as]
-            attributes[reflection.type] = owner.class.base_class.name
-          end
-        end
-
-        attributes
-      end
-    end
-  end
-end
+module ActiveRecord
+  module Associations
+    class Association
+      def creation_attributes
+        attributes = {}
+
+        if (reflection.has_one? || reflection.collection?) && !options[:through]
+          # CPK
+          # attributes[reflection.foreign_key] = owner[reflection.active_record_primary_key]
+          Array(reflection.foreign_key).zip(Array(reflection.active_record_primary_key)).each do |key1, key2|
+            attributes[key1] = owner[key2]
+          end
+
+          if reflection.options[:as]
+            attributes[reflection.type] = owner.class.base_class.name
+          end
+        end
+
+        attributes
+      end
+    end
+  end
+end

data/lib/composite_primary_keys/associations/association_scope.rb

@@ -1,69 +1,67 @@
-module ActiveRecord
-  module Associations
-    class AssociationScope
-      def self.get_bind_values(owner, chain)
-        binds = []
-        last_reflection = chain.last
-
-        # CPK
-        # binds << last_reflection.join_id_for(owner)
-        values = last_reflection.join_id_for(owner)
-        binds += Array(values)
-
-        if last_reflection.type
-          binds << owner.class.polymorphic_name
-        end
-
-        chain.each_cons(2).each do |reflection, next_reflection|
-          if reflection.type
-            binds << next_reflection.klass.polymorphic_name
-          end
-        end
-        binds
-      end
-
-      def last_chain_scope(scope, reflection, owner)
-        join_keys = reflection.join_keys
-        key = join_keys.key
-        foreign_key = join_keys.foreign_key
-
-        table = reflection.aliased_table
-
-        # CPK
-        # value = transform_value(owner[foreign_key])
-        # scope = apply_scope(scope, table, key, value)
-        Array(key).zip(Array(foreign_key)).each do |a_join_key, a_foreign_key|
-          value = transform_value(owner[a_foreign_key])
-          scope = apply_scope(scope, table, a_join_key, value)
-        end
-
-        if reflection.type
-          polymorphic_type = transform_value(owner.class.polymorphic_name)
-          scope = apply_scope(scope, table, reflection.type, polymorphic_type)
-        end
-
-        scope
-      end
-
-      def next_chain_scope(scope, reflection, next_reflection)
-        join_keys = reflection.join_keys
-        key = join_keys.key
-        foreign_key = join_keys.foreign_key
-
-        table = reflection.aliased_table
-        foreign_table = next_reflection.aliased_table
-
-        # CPK
-        # constraint = table[key].eq(foreign_table[foreign_key])
-        constraint = cpk_join_predicate(table, key, foreign_table, foreign_key)
-
-        if reflection.type
-          value = transform_value(next_reflection.klass.polymorphic_name)
-          scope = apply_scope(scope, table, reflection.type, value)
-        end
-
-        scope.joins!(join(foreign_table, constraint))
-      end
-    end
-  end
+module ActiveRecord
+  module Associations
+    class AssociationScope
+      def self.get_bind_values(owner, chain)
+        binds = []
+        last_reflection = chain.last
+
+        # CPK
+        # binds << last_reflection.join_id_for(owner)
+        values = last_reflection.join_id_for(owner)
+        binds += Array(values)
+
+        if last_reflection.type
+          binds << owner.class.polymorphic_name
+        end
+
+        chain.each_cons(2).each do |reflection, next_reflection|
+          if reflection.type
+            binds << next_reflection.klass.polymorphic_name
+          end
+        end
+        binds
+      end
+
+      def last_chain_scope(scope, reflection, owner)
+        key = reflection.join_primary_key
+        foreign_key = reflection.join_foreign_key
+
+        table = reflection.aliased_table
+
+        # CPK
+        # value = transform_value(owner[foreign_key])
+        # scope = apply_scope(scope, table, key, value)
+        Array(key).zip(Array(foreign_key)).each do |a_join_key, a_foreign_key|
+          value = transform_value(owner[a_foreign_key])
+          scope = apply_scope(scope, table, a_join_key, value)
+        end
+
+        if reflection.type
+          polymorphic_type = transform_value(owner.class.polymorphic_name)
+          scope = apply_scope(scope, table, reflection.type, polymorphic_type)
+        end
+
+        scope
+      end
+
+      def next_chain_scope(scope, reflection, next_reflection)
+        key = reflection.join_primary_key
+        foreign_key = reflection.join_foreign_key
+
+        table = reflection.aliased_table
+        foreign_table = next_reflection.aliased_table
+
+        # CPK
+        # constraint = table[key].eq(foreign_table[foreign_key])
+        constraint = cpk_join_predicate(table, key, foreign_table, foreign_key)
+
+        if reflection.type
+          value = transform_value(next_reflection.klass.polymorphic_name)
+          scope = apply_scope(scope, table, reflection.type, value)
+        end
+
+        scope.joins!(join(foreign_table, constraint))
+      end
+    end
+  end
 end

data/lib/composite_primary_keys/associations/collection_association.rb

@@ -1,32 +1,32 @@
-module CompositePrimaryKeys
-  module CollectionAssociation
-    def ids_writer(ids)
-      primary_key = reflection.association_primary_key
-      pk_type = klass.type_for_attribute(primary_key)
-      ids = Array(ids).reject(&:blank?)
-      ids.map! { |i| pk_type.cast(i) }
-
-      # CPK-
-      if primary_key.is_a?(Array)
-        predicate = CompositePrimaryKeys::Predicates.cpk_in_predicate(klass.arel_table, reflection.association_primary_key, ids)
-        records = klass.where(predicate).index_by do |r|
-          reflection.association_primary_key.map{ |k| r.send(k) }
-        end.values_at(*ids)
-      else
-        records = klass.where(primary_key => ids).index_by do |r|
-          r.public_send(primary_key)
-        end.values_at(*ids).compact
-      end
-
-      if records.size != ids.size
-        found_ids = records.map { |record| record.public_send(primary_key) }
-        not_found_ids = ids - found_ids
-        klass.all.raise_record_not_found_exception!(ids, records.size, ids.size, primary_key, not_found_ids)
-      else
-        replace(records)
-      end
-    end
-  end
-end
-
+module CompositePrimaryKeys
+  module CollectionAssociation
+    def ids_writer(ids)
+      primary_key = reflection.association_primary_key
+      pk_type = klass.type_for_attribute(primary_key)
+      ids = Array(ids).reject(&:blank?)
+      ids.map! { |i| pk_type.cast(i) }
+
+      # CPK-
+      if primary_key.is_a?(Array)
+        predicate = CompositePrimaryKeys::Predicates.cpk_in_predicate(klass.arel_table, reflection.association_primary_key, ids)
+        records = klass.where(predicate).index_by do |r|
+          reflection.association_primary_key.map{ |k| r.send(k) }
+        end.values_at(*ids)
+      else
+        records = klass.where(primary_key => ids).index_by do |r|
+          r.public_send(primary_key)
+        end.values_at(*ids).compact
+      end
+
+      if records.size != ids.size
+        found_ids = records.map { |record| record.public_send(primary_key) }
+        not_found_ids = ids - found_ids
+        klass.all.raise_record_not_found_exception!(ids, records.size, ids.size, primary_key, not_found_ids)
+      else
+        replace(records)
+      end
+    end
+  end
+end
+
 ActiveRecord::Associations::CollectionAssociation.prepend CompositePrimaryKeys::CollectionAssociation

data/lib/composite_primary_keys/associations/foreign_association.rb

@@ -1,15 +1,15 @@
-# frozen_string_literal: true
-
-module ActiveRecord::Associations
-  module ForeignAssociation # :nodoc:
-    def foreign_key_present?
-      if reflection.klass.primary_key
-        # CPK
-        # owner.attribute_present?(reflection.active_record_primary_key)
-        Array(reflection.active_record_primary_key).all? {|key| owner.attribute_present?(key)}
-      else
-        false
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+module ActiveRecord::Associations
+  module ForeignAssociation # :nodoc:
+    def foreign_key_present?
+      if reflection.klass.primary_key
+        # CPK
+        # owner.attribute_present?(reflection.active_record_primary_key)
+        Array(reflection.active_record_primary_key).all? {|key| owner.attribute_present?(key)}
+      else
+        false
+      end
+    end
+  end
+end

data/lib/composite_primary_keys/associations/has_many_association.rb

@@ -1,35 +1,35 @@
-module ActiveRecord
-  module Associations
-    class HasManyAssociation
-      def delete_records(records, method)
-        if method == :destroy
-          records.each(&:destroy!)
-          update_counter(-records.length) unless reflection.inverse_updates_counter_cache?
-        # CPK
-        elsif self.reflection.klass.composite?
-          predicate = cpk_in_predicate(self.scope.table, self.reflection.klass.primary_keys, records.map(&:id))
-          scope = self.scope.where(predicate)
-          update_counter(-delete_count(method, scope))
-        else
-          scope = self.scope.where(reflection.klass.primary_key => records)
-          update_counter(-delete_count(method, scope))
-        end
-      end
-
-      def delete_count(method, scope)
-        if method == :delete_all
-          scope.delete_all
-        else
-          # CPK
-          # scope.update_all(nullified_owner_attributes)
-          conds = Array(reflection.foreign_key).inject(Hash.new) do |mem, key|
-            mem[key] = nil
-            mem
-          end
-          conds[reflection.type] = nil if reflection.type.present?
-          scope.update_all(conds)
-        end
-      end
-    end
-  end
-end
+module ActiveRecord
+  module Associations
+    class HasManyAssociation
+      def delete_records(records, method)
+        if method == :destroy
+          records.each(&:destroy!)
+          update_counter(-records.length) unless reflection.inverse_updates_counter_cache?
+        # CPK
+        elsif self.reflection.klass.composite?
+          predicate = cpk_in_predicate(self.scope.table, self.reflection.klass.primary_keys, records.map(&:id))
+          scope = self.scope.where(predicate)
+          update_counter(-delete_count(method, scope))
+        else
+          scope = self.scope.where(reflection.klass.primary_key => records)
+          update_counter(-delete_count(method, scope))
+        end
+      end
+
+      def delete_count(method, scope)
+        if method == :delete_all
+          scope.delete_all
+        else
+          # CPK
+          # scope.update_all(nullified_owner_attributes)
+          conds = Array(reflection.foreign_key).inject(Hash.new) do |mem, key|
+            mem[key] = nil
+            mem
+          end
+          conds[reflection.type] = nil if reflection.type.present?
+          scope.update_all(conds)
+        end
+      end
+    end
+  end
+end