eager_group 0.9.1 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 119ff48ee90c0324a8e7f55e521a3606485ad1728ec27eaefea3df1f7c153173
-  data.tar.gz: 7415661b2a4ced20c7673c8d90bfb977fbef22c203062f0bb60a8be5fdd1970a
+  metadata.gz: 3717871f8324a6ec0a0886745b12c17f7443398a8658bcb60ce008fa1e924df1
+  data.tar.gz: 0eccab385f9d6ec341ace42258672f663b03598ee164ef22b1b25fb7bf17779a
 SHA512:
-  metadata.gz: f3b85835d577315809f1a699c7a717d17b1f61c14ae0e0371312c12615a5088c45a266028fc3f091801639298e9dd6a1ca1e8d6beb37889a6f5e63aacbdc8fe6
-  data.tar.gz: 2475f41755b2b0161bcc4fe01fd04a5cb1b9f69728edd0e165e8596773306e421426efc191aa34eba812912c1e6b8cc608386a3fc29ca490a068c15b2335e337
+  metadata.gz: c1fb52a045e4f2005a78ef698ff915195e705a9729e1fb6a25275d2516c48e7f016d3249946d3a337b07b70a822f75c0ffe1f2acbb130c78a2d13df03107449b
+  data.tar.gz: 0a7393ea354afaaf1c1002c7641c3a4bd94e26b2407eea9b6cb701da243b67733c34d226711650964251aaa2f5d117db0d0e418e5702ebb4b75f59323c3b6c49

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Next Release
 
+## 0.10.0 (12/28/2022)
+
+* Add STI support, use `class_attribute` to prevent subclasses affect each other
+
 ## 0.9.1 (12/15/2022)
 
 * Fix eager group fetch multi definitions

data/lib/active_record/with_eager_group.rb

@@ -13,6 +13,7 @@ module ActiveRecord
       # which would cause `[:eager_group_definition, scope_arg1, scope_arg2]` not able to preload together with other `eager_group_definitions`.
       # e.g. `Post.eager_group(:approved_comments_count, [:comments_average_rating_by_author, students[0], true])`
       check_argument_not_blank!(args)
+      check_argument_valid!(args)
 
       spawn.eager_group!(*args)
     end
@@ -38,5 +39,31 @@ module ActiveRecord
       raise ArgumentError, "The method .eager_group() must contain arguments." if args.blank?
       args.compact_blank!
     end
+
+    def check_argument_valid!(args)
+      args.each do |eager_group_value|
+        check_eager_group_definitions_exists!(klass, eager_group_value)
+      end
+    end
+
+    def check_eager_group_definitions_exists!(klass, eager_group_value)
+      case eager_group_value
+      when Symbol, String
+        raise ArgumentError, "Unknown eager group definition :#{eager_group_value}" unless klass.eager_group_definitions.has_key?(eager_group_value)
+      when Array
+        definition_name = eager_group_value.first
+        raise ArgumentError, "Unknown eager group definition :#{definition_name}" unless klass.eager_group_definitions.has_key?(definition_name)
+      when Hash
+        eager_group_value.each do |association_name, association_eager_group_values|
+          association_klass = klass.reflect_on_association(association_name).klass
+
+          Array.wrap(association_eager_group_values).each do |association_eager_group_value|
+            check_eager_group_definitions_exists!(association_klass, association_eager_group_value)
+          end
+        end
+      else
+        raise ArgumentError, "Unknown eager_group argument :#{eager_group_value.inspect}"
+      end
+    end
   end
 end

data/lib/eager_group/preloader.rb

@@ -22,10 +22,13 @@ module EagerGroup
 
         if definition_key.is_a?(Hash)
           association_name, definition_key = *definition_key.first
+          next if @records.empty?
+          @klass = @records.first.class.reflect_on_association(association_name).klass
+
           @records = @records.flat_map { |record| record.send(association_name) }
           next if @records.empty?
 
-          @klass = @records.first.class
+
           self.class.new(@klass, @records, Array.wrap(definition_key)).run
         end
 

data/lib/eager_group/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module EagerGroup
-  VERSION = '0.9.1'
+  VERSION = '0.10.0'
 end

data/lib/eager_group.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 require 'active_support/core_ext/module/attribute_accessors'
+require 'active_support/core_ext/class/attribute'
+require 'active_support/core_ext/hash'
 require 'eager_group/version'
 
 module EagerGroup
@@ -9,30 +11,39 @@ module EagerGroup
 
   def self.included(base)
     base.extend ClassMethods
+    base.class_eval do
+      class_attribute :eager_group_definitions, instance_writer: false, default: {}.with_indifferent_access
+    end
   end
 
   module ClassMethods
-    mattr_accessor :eager_group_definitions, default: {}
+    #mattr_accessor :eager_group_definitions, default: {}
+
+    def add_eager_group_definition(ar, definition_name, definition)
+      ar.eager_group_definitions = self.eager_group_definitions.except(definition_name).merge!(definition_name => definition)
+    end
 
     # class Post
     #   define_eager_group :comments_avergage_rating, :comments, :average, :rating
     #   define_eager_group :approved_comments_count, :comments, :count, :*, -> { approved }
     # end
     def define_eager_group(attr, association, aggregate_function, column_name, scope = nil)
-      send :attr_accessor, attr
-      eager_group_definitions[attr] = Definition.new(association, aggregate_function, column_name, scope)
+      add_eager_group_definition(self, attr, Definition.new(association, aggregate_function, column_name, scope))
+      define_definition_accessor(attr)
+    end
 
-      define_method attr,
+    def define_definition_accessor(definition_name)
+      define_method definition_name,
                     lambda { |*args|
-                      query_result_cache = instance_variable_get("@#{attr}")
+                      query_result_cache = instance_variable_get("@#{definition_name}")
                       return query_result_cache if args.blank? && query_result_cache.present?
 
-                      preload_eager_group(attr, *args)
-                      instance_variable_get("@#{attr}")
+                      preload_eager_group(definition_name, *args)
+                      instance_variable_get("@#{definition_name}")
                     }
 
-      define_method "#{attr}=" do |val|
-        instance_variable_set("@#{attr}", val)
+      define_method "#{definition_name}=" do |val|
+        instance_variable_set("@#{definition_name}", val)
       end
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: eager_group
 version: !ruby/object:Gem::Version
-  version: 0.9.1
+  version: 0.10.0
 platform: ruby
 authors:
 - Richard Huang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-12-15 00:00:00.000000000 Z
+date: 2022-12-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -185,7 +185,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.26
+rubygems_version: 3.4.1
 signing_key:
 specification_version: 4
 summary: Fix n+1 aggregate sql functions