deep_pluck 1.1.4 → 1.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b7bbd3c0df3b323529a252d7d56fa596b4f85923b06b54a04560826b2793b8fe
-  data.tar.gz: 1261968870077b5cb7f4db0c56e1045f513b6caa2a76c028a9235535635704c5
+  metadata.gz: 542e1ba086c4b2a8df7fb4efbf916c3448f207824717272e67bbba581154b3dc
+  data.tar.gz: 6710c826f3ae127d4530863b690541fe048ac6f8b3fc54a18e2da8ecd8629ee4
 SHA512:
-  metadata.gz: dd9196070fb6189bc2e2b8da44976fd0ecd5dead243075a8abe66ae524af674d1f8e99f74abb0d2337bf65e42f85688ef20ba07c4b422f0179750b50d35421d3
-  data.tar.gz: 34302f6ca57406fa7c61635fd0e2465cafef8b41da426b62f23fee1635df506a9d1750b2506653b0d307d2d1a599a83d23986ad9efc772c2ad66dd5792bdbef1
+  metadata.gz: c8c17c52a55de2a7de18ecf762adefe017f05cef401f4e0098d8ce0b5699898ac97918cbede23f802ed8ff4709c1b5655938d64c3511e72494e9777a2ec98e5e
+  data.tar.gz: 21aae7d86d41d53d4b05824ba13a5b7e9c7486ae9f83552146891f62397d8b03e5fc7b6b99c71c60837782d608767a8054e4674af72c2c0fbd655a1e324627a1

data/.github/FUNDING.yml

@@ -0,0 +1,12 @@
+# These are supported funding model platforms
+
+github: [khiav223577] # Replace with up to 4 GitHub Sponsors-enabled usernames e.g., [user1, user2]
+patreon: # Replace with a single Patreon username
+open_collective: # Replace with a single Open Collective username
+ko_fi: # Replace with a single Ko-fi username
+tidelift: # Replace with a single Tidelift platform-name/package-name e.g., npm/babel
+community_bridge: # Replace with a single Community Bridge project-name e.g., cloud-foundry
+liberapay: # Replace with a single Liberapay username
+issuehunt: # Replace with a single IssueHunt username
+otechie: # Replace with a single Otechie username
+custom: # Replace with up to 4 custom sponsorship URLs e.g., ['link1', 'link2']

data/.gitignore

@@ -1,6 +1,7 @@
 /.bundle/
 /.yardoc
 /Gemfile.lock
+/gemfiles/*.gemfile.lock
 /_yardoc/
 /coverage/
 /doc/

data/.rubocop.yml

@@ -1,3 +1,7 @@
+require:
+  - rubocop-performance
+  - rubocop-rails
+
 AllCops:
   DisabledByDefault: true
   Exclude: []
@@ -73,10 +77,6 @@ Layout/EndAlignment:
   Description: 'Align ends correctly.'
   Enabled: true
 
-Lint/EndInMethod:
-  Description: 'END blocks should not be placed inside method definitions.'
-  Enabled: true
-
 Lint/EnsureReturn:
   Description: 'Do not use return in an ensure block.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#no-return-ensure'
@@ -90,7 +90,7 @@ Lint/FormatParameterMismatch:
   Description: 'The number of parameters to format/sprint must match the fields.'
   Enabled: true
 
-Lint/HandleExceptions:
+Lint/SuppressedException:
   Description: "Don't suppress exception."
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#dont-hide-exceptions'
   Enabled: true
@@ -139,7 +139,7 @@ Lint/ShadowingOuterLocalVariable:
                  for block arguments or block local variables.
   Enabled: true
 
-Lint/StringConversionInInterpolation:
+Lint/RedundantStringCoercion:
   Description: 'Checks for Object#to_s usage in string interpolation.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#no-to-s'
   Enabled: true
@@ -148,7 +148,7 @@ Lint/UnderscorePrefixedVariableName:
   Description: 'Do not use prefix `_` for a variable that is used.'
   Enabled: true
 
-Lint/UnneededCopDisableDirective:
+Lint/RedundantCopDisableDirective:
   Description: >-
                  Checks for rubocop:disable comments that can be removed.
                  Note: this cop is not disabled when disabling all cops.
@@ -178,8 +178,8 @@ Lint/UselessAssignment:
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#underscore-unused-vars'
   Enabled: true
 
-Lint/UselessComparison:
-  Description: 'Checks for comparison of something with itself.'
+Lint/BinaryOperatorWithIdenticalOperands:
+  Description: 'This cop checks for places where binary operator has identical operands.'
   Enabled: true
 
 Lint/UselessElseWithoutRescue:
@@ -221,7 +221,7 @@ Metrics/CyclomaticComplexity:
                  of test cases needed to validate a method.
   Enabled: true
 
-Metrics/LineLength:
+Layout/LineLength:
   Description: 'Limit lines to 120 characters.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#80-character-limits'
   Max: 120
@@ -285,7 +285,7 @@ Performance/ReverseEach:
   Reference: 'https://github.com/JuanitoFatas/fast-ruby#enumerablereverseeach-vs-enumerablereverse_each-code'
   Enabled: true
 
-Performance/Sample:
+Style/Sample:
   Description: >-
                   Use `sample` instead of `shuffle.first`,
                   `shuffle.last`, and `shuffle[Fixnum]`.
@@ -353,7 +353,7 @@ Rails/TimeZone:
   Description: 'Checks the correct usage of time zone aware methods.'
   StyleGuide: 'https://github.com/bbatsov/rails-style-guide#time'
   Reference: 'http://danilenko.org/2012/7/6/rails_timezones'
-  Enabled: true
+  Enabled: false
 
 Rails/Validation:
   Description: 'Use validates :attribute, hash of validations.'
@@ -375,21 +375,21 @@ Style/Alias:
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#alias-method'
   Enabled: true
 
-Layout/AlignArray:
+Layout/ArrayAlignment:
   Description: >-
                  Align the elements of an array literal if they span more than
                  one line.
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#align-multiline-arrays'
   Enabled: true
 
-Layout/AlignHash:
+Layout/HashAlignment:
   Description: >-
                  Align the elements of a hash literal if they span more than
                  one line.
   EnforcedHashRocketStyle: table
   Enabled: true
 
-Layout/AlignParameters:
+Layout/ParameterAlignment:
   Description: >-
                  Align the parameters of a method call if they span more
                  than one line.
@@ -448,10 +448,6 @@ Style/BlockDelimiters:
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#single-line-blocks'
   Enabled: true
 
-Style/BracesAroundHashParameters:
-  Description: 'Enforce braces style around hash parameters.'
-  Enabled: true
-
 Style/CaseEquality:
   Description: 'Avoid explicit use of the case equality operator(===).'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#no-case-equality'
@@ -596,7 +592,7 @@ Style/EndBlock:
 Layout/EndOfLine:
   Description: 'Use Unix-style line endings.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#crlf'
-  Enabled: true
+  Enabled: false
 
 Style/EvenOdd:
   Description: 'Favor the use of Fixnum#even? && Fixnum#odd?'
@@ -617,14 +613,14 @@ Layout/InitialIndentation:
     Checks the indentation of the first non-blank non-comment line in a file.
   Enabled: true
 
-Layout/IndentFirstArgument:
+Layout/FirstArgumentIndentation:
   Description: 'Checks the indentation of the first parameter in a method call.'
   Enabled: true
 
-Style/FlipFlop:
+Lint/FlipFlop:
   Description: 'Checks for flip flops'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#no-flip-flops'
-  Enabled: true
+  Enabled: false
 
 Style/For:
   Description: 'Checks use of for or each in multiline loops.'
@@ -634,9 +630,12 @@ Style/For:
 Style/FormatString:
   Description: 'Enforce the use of Kernel#sprintf, Kernel#format or String#%.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#sprintf'
-  EnforcedStyle: sprintf
+  EnforcedStyle: percent
   Enabled: true
 
+Style/FormatStringToken:
+  Enabled: false
+
 Style/GlobalVars:
   Description: 'Do not introduce global variables.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#instance-vars'
@@ -676,14 +675,14 @@ Layout/IndentationWidth:
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#spaces-indentation'
   Enabled: true
 
-Layout/IndentFirstArrayElement:
+Layout/FirstArrayElementIndentation:
   Description: >-
                  Checks the indentation of the first element in an array
                  literal.
   EnforcedStyle: consistent
   Enabled: true
 
-Layout/IndentFirstHashElement:
+Layout/FirstHashElementIndentation:
   Description: 'Checks the indentation of the first key in a hash literal.'
   EnforcedStyle: consistent
   Enabled: true
@@ -862,7 +861,7 @@ Style/PercentQLiterals:
 Style/PerlBackrefs:
   Description: 'Avoid Perl-style regex back references.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#no-perl-regexp-last-matchers'
-  Enabled: true
+  Enabled: false
 
 Naming/PredicateName:
   Description: 'Check the names of predicate methods.'
@@ -940,11 +939,11 @@ Style/SingleLineBlockParams:
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#reduce-blocks'
   Methods:
     - reduce:
-        - acc
-        - elem
+        - sum
+        - v
     - inject:
-        - acc
-        - elem
+        - sum
+        - v
   Enabled: true
 
 Style/SingleLineMethods:
@@ -1118,12 +1117,12 @@ Style/SymbolProc:
   Description: 'Use symbols as procs instead of blocks when possible.'
   Enabled: true
 
-Layout/Tab:
+Layout/IndentationStyle:
   Description: 'No hard tabs.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#spaces-indentation'
   Enabled: true
 
-Layout/TrailingBlankLines:
+Layout/TrailingEmptyLines:
   Description: 'Checks trailing blank lines and final newline.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#newline-eof'
   Enabled: true
@@ -1178,11 +1177,11 @@ Style/UnlessElse:
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#no-else-with-unless'
   Enabled: true
 
-Style/UnneededCapitalW:
+Style/RedundantCapitalW:
   Description: 'Checks for %W when interpolation is not needed.'
   Enabled: true
 
-Style/UnneededPercentQ:
+Style/RedundantPercentQ:
   Description: 'Checks for %q/%Q when single quotes or double quotes would do.'
   StyleGuide: 'https://github.com/bbatsov/ruby-style-guide#percent-q'
   Enabled: true

data/.travis.yml

@@ -14,6 +14,7 @@ gemfile:
   - gemfiles/5.1.gemfile
   - gemfiles/5.2.gemfile
   - gemfiles/6.0.gemfile
+  - gemfiles/6.1.gemfile
 matrix:
   exclude:
     - gemfile: gemfiles/3.2.gemfile
@@ -24,6 +25,8 @@ matrix:
       rvm: 2.7
     - gemfile: gemfiles/6.0.gemfile
       rvm: 2.2
+    - gemfile: gemfiles/6.1.gemfile
+      rvm: 2.2
 before_install:
   - if `ruby -e 'exit(RUBY_VERSION.to_f < 2.7)'`; then
       gem i rubygems-update -v '< 3' && update_rubygems;

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## Change Log
 
+### [v1.1.4](https://github.com/khiav223577/deep_pluck/compare/v1.1.3...v1.1.4) 2020/01/13
+- [#36](https://github.com/khiav223577/deep_pluck/pull/36) A workaround to fix mismatched association named. (@khiav223577)
+- [#35](https://github.com/khiav223577/deep_pluck/pull/35) Support Ruby 2.7 (@khiav223577)
+
 ### [v1.1.3](https://github.com/khiav223577/deep_pluck/compare/v1.1.2...v1.1.3) 2019/12/17
 - [#34](https://github.com/khiav223577/deep_pluck/pull/34) Support for plucking directly on a has_one through association (@khiav223577)
 

data/gemfiles/6.1.gemfile

@@ -0,0 +1,11 @@
+source 'https://rubygems.org'
+
+gem 'activerecord', '~> 6.1.1'
+gem 'pluck_all', '~> 2.2.1'
+
+group :test do
+  gem 'simplecov'
+  gem 'sqlite3', '~> 1.4.1'
+end
+
+gemspec path: '../'

data/lib/deep_pluck/data_combiner.rb

@@ -26,8 +26,8 @@ module DeepPluck
       def assign_values_to_parent(collection, parent, children_hash, column_name, foreign_key, reverse: false)
         parent.each do |s|
           next if (id = s[foreign_key]) == nil
-          left   =  reverse ? children_hash[id] : s
-          right  = !reverse ? children_hash[id] : s
+          left = reverse ? children_hash[id] : s
+          right = !reverse ? children_hash[id] : s
           if collection
             left[column_name] << right
           else

data/lib/deep_pluck/model.rb

@@ -1,225 +1,225 @@
-require 'rails_compatibility'
-require 'rails_compatibility/unscope_where'
-require 'deep_pluck/data_combiner'
-
-module DeepPluck
-  class Model
-    # ----------------------------------------------------------------
-    # ● Initialize
-    # ----------------------------------------------------------------
-    def initialize(relation, parent_association_key = nil, parent_model = nil, need_columns: [])
-      if relation.is_a?(ActiveRecord::Base)
-        @model = relation
-        @relation = nil
-        @klass = @model.class
-      else
-        @model = nil
-        @relation = relation
-        @klass = @relation.klass
-      end
-
-      @parent_association_key = parent_association_key
-      @parent_model = parent_model
-      @need_columns = need_columns
-      @associations = {}
-    end
-
-    # ----------------------------------------------------------------
-    # ● Reader
-    # ----------------------------------------------------------------
-    def get_reflect(association_key)
-      @klass.reflect_on_association(association_key.to_sym) || # add to_sym since rails 3 only support symbol
-        fail(ActiveRecord::ConfigurationError, "ActiveRecord::ConfigurationError: Association named \
-          '#{association_key}' was not found on #{@klass.name}; perhaps you misspelled it?"
-      )
-    end
-
-    def with_conditions(reflect, relation)
-      options = reflect.options
-      relation = relation.instance_exec(&reflect.scope) if reflect.respond_to?(:scope) and reflect.scope
-      relation = relation.where(options[:conditions]) if options[:conditions]
-      return relation
-    end
-
-    def get_join_table(reflect)
-      options = reflect.options
-      return options[:through] if options[:through]
-      return (options[:join_table] || reflect.send(:derive_join_table)) if reflect.macro == :has_and_belongs_to_many
-      return nil
-    end
-
-    def get_primary_key(reflect)
-      return (reflect.belongs_to? ? reflect.klass : reflect.active_record).primary_key
-    end
-
-    def get_foreign_key(reflect, reverse: false, with_table_name: false)
-      reflect = reflect.chain.last
-      if reverse and (table_name = get_join_table(reflect)) # reverse = parent
-        key = reflect.chain.last.foreign_key
-      else
-        key = (reflect.belongs_to? == reverse ? get_primary_key(reflect) : reflect.foreign_key)
-        table_name = (reverse ? reflect.klass : reflect.active_record).table_name
-      end
-      return "#{table_name}.#{key}" if with_table_name
-      return key.to_s # key may be symbol if specify foreign_key in association options
-    end
-
-    def get_association_scope(reflect)
-      RailsCompatibility.unscope_where(reflect.association_class.new({}, reflect).send(:association_scope))
-    end
-
-    def use_association_to_query?(reflect)
-      reflect.through_reflection && reflect.chain.first.macro == :has_one
-    end
-
-    # ----------------------------------------------------------------
-    # ● Contruction OPs
-    # ----------------------------------------------------------------
-
-    private
-
-    def add_need_column(column)
-      @need_columns << column
-    end
-
-    def add_association(hash)
-      hash.each do |key, value|
-        model = (@associations[key] ||= Model.new(get_reflect(key).klass.where(''), key, self))
-        model.add(value)
-      end
-    end
-
-    public
-
-    def add(args)
-      return self if args == nil
-      args = [args] if not args.is_a?(Array)
-      args.each do |arg|
-        case arg
-        when Hash ; add_association(arg)
-        else      ; add_need_column(arg)
-        end
-      end
-      return self
-    end
-
-    # ----------------------------------------------------------------
-    # ● Load
-    # ----------------------------------------------------------------
-    private
-
-    def do_query(parent, reflect, relation)
-      parent_key = get_foreign_key(reflect)
-      relation_key = get_foreign_key(reflect, reverse: true, with_table_name: true)
-      ids = parent.map{|s| s[parent_key] }
-      ids.uniq!
-      ids.compact!
-      relation = with_conditions(reflect, relation)
-      query = { relation_key => ids }
-      query[reflect.type] = reflect.active_record.to_s if reflect.type
-
-      return get_association_scope(reflect).where(query) if use_association_to_query?(reflect)
-
-      join_table = get_join_table(reflect)
-      join_table = backtrace_possible_association(relation, join_table)
-
-      return relation.joins(join_table).where(query)
-    end
-
-    # Let city has_many :users, through: :schools
-    # And the query is: City.deep_pluck('users' => :name)
-    # We want to get the users data via `User.joins(:school).where(city_id: city_ids)`
-    # But get_join_table(reflect) returns `:schools` not :school
-    # No idea how to get the right association, so we try singularize or pluralize it.
-    def backtrace_possible_association(relation, join_table)
-      return join_table if relation.reflect_on_association(join_table)
-      join_table.to_s.singularize.to_sym.tap{|s| return s if relation.reflect_on_association(s) }
-      join_table.to_s.pluralize.to_sym.tap{|s| return s if relation.reflect_on_association(s) }
-      return nil
-    end
-
-    def set_includes_data(parent, column_name, model)
-      reflect = get_reflect(column_name)
-      reverse = !reflect.belongs_to?
-      foreign_key = get_foreign_key(reflect, reverse: reverse)
-      primary_key = get_foreign_key(reflect, reverse: !reverse)
-      children = model.load_data{|relation| do_query(parent, reflect, relation) }
-      # reverse = false: Child.where(:id => parent.pluck(:child_id))
-      # reverse = true : Child.where(:parent_id => parent.pluck(:id))
-      return DataCombiner.combine_data(
-        parent,
-        children,
-        primary_key,
-        column_name,
-        foreign_key,
-        reverse,
-        reflect.collection?,
-      )
-    end
-
-    def get_query_columns
-      if @parent_model
-        parent_reflect = @parent_model.get_reflect(@parent_association_key)
-        prev_need_columns = @parent_model.get_foreign_key(parent_reflect, reverse: true, with_table_name: true)
-      end
-      next_need_columns = @associations.map{|key, _| get_foreign_key(get_reflect(key), with_table_name: true) }.uniq
-      return [*prev_need_columns, *next_need_columns, *@need_columns].uniq(&Helper::TO_KEY_PROC)
-    end
-
-    def pluck_values(columns)
-      includes_values = @relation.includes_values
-      @relation.includes_values = []
-
-      result = @relation.pluck_all(*columns)
-
-      @relation.includes_values = includes_values
-      return result
-    end
-
-    def loaded_models
-      return [@model] if @model
-      return @relation if @relation.loaded
-    end
-
-    public
-
-    def load_data
-      columns = get_query_columns
-      key_columns = columns.map(&Helper::TO_KEY_PROC)
-      @relation = yield(@relation) if block_given?
-      @data = loaded_models ? loaded_models.as_json(root: false, only: key_columns) : pluck_values(columns)
-      if @data.size != 0
-        # for delete_extra_column_data!
-        @extra_columns = key_columns - @need_columns.map(&Helper::TO_KEY_PROC)
-        @associations.each do |key, model|
-          set_includes_data(@data, key, model)
-        end
-      end
-      return @data
-    end
-
-    def load_all
-      load_data
-      delete_extra_column_data!
-      return @data
-    end
-
-    def delete_extra_column_data!
-      return if @data.blank?
-      @data.each{|s| s.except!(*@extra_columns) }
-      @associations.each{|_, model| model.delete_extra_column_data! }
-    end
-
-    # ----------------------------------------------------------------
-    # ● Helper methods
-    # ----------------------------------------------------------------
-    module Helper
-      TO_KEY_PROC = proc{|s| Helper.column_to_key(s) }
-      def self.column_to_key(key) # user_achievements.user_id => user_id
-        key = key[/(\w+)[^\w]*\z/]
-        key.gsub!(/[^\w]+/, '')
-        return key
-      end
-    end
-  end
-end
+require 'rails_compatibility'
+require 'rails_compatibility/unscope_where'
+require 'deep_pluck/data_combiner'
+
+module DeepPluck
+  class Model
+    # ----------------------------------------------------------------
+    # ● Initialize
+    # ----------------------------------------------------------------
+    def initialize(relation, parent_association_key = nil, parent_model = nil, need_columns: [])
+      if relation.is_a?(ActiveRecord::Base)
+        @model = relation
+        @relation = nil
+        @klass = @model.class
+      else
+        @model = nil
+        @relation = relation
+        @klass = @relation.klass
+      end
+
+      @parent_association_key = parent_association_key
+      @parent_model = parent_model
+      @need_columns = need_columns
+      @associations = {}
+    end
+
+    # ----------------------------------------------------------------
+    # ● Reader
+    # ----------------------------------------------------------------
+    def get_reflect(association_key)
+      @klass.reflect_on_association(association_key.to_sym) || # add to_sym since rails 3 only support symbol
+        fail(ActiveRecord::ConfigurationError, "ActiveRecord::ConfigurationError: Association named \
+          '#{association_key}' was not found on #{@klass.name}; perhaps you misspelled it?"
+        )
+    end
+
+    def with_conditions(reflect, relation)
+      options = reflect.options
+      relation = relation.instance_exec(&reflect.scope) if reflect.respond_to?(:scope) and reflect.scope
+      relation = relation.where(options[:conditions]) if options[:conditions]
+      return relation
+    end
+
+    def get_join_table(reflect)
+      options = reflect.options
+      return options[:through] if options[:through]
+      return (options[:join_table] || reflect.send(:derive_join_table)) if reflect.macro == :has_and_belongs_to_many
+      return nil
+    end
+
+    def get_primary_key(reflect)
+      return (reflect.belongs_to? ? reflect.klass : reflect.active_record).primary_key
+    end
+
+    def get_foreign_key(reflect, reverse: false, with_table_name: false)
+      reflect = reflect.chain.last
+      if reverse and (table_name = get_join_table(reflect)) # reverse = parent
+        key = reflect.chain.last.foreign_key
+      else
+        key = (reflect.belongs_to? == reverse ? get_primary_key(reflect) : reflect.foreign_key)
+        table_name = (reverse ? reflect.klass : reflect.active_record).table_name
+      end
+      return "#{table_name}.#{key}" if with_table_name
+      return key.to_s # key may be symbol if specify foreign_key in association options
+    end
+
+    def get_association_scope(reflect)
+      RailsCompatibility.unscope_where(reflect.association_class.new({}, reflect).send(:association_scope))
+    end
+
+    def use_association_to_query?(reflect)
+      reflect.through_reflection && reflect.chain.first.macro == :has_one
+    end
+
+    # ----------------------------------------------------------------
+    # ● Contruction OPs
+    # ----------------------------------------------------------------
+
+    private
+
+    def add_need_column(column)
+      @need_columns << column
+    end
+
+    def add_association(hash)
+      hash.each do |key, value|
+        model = (@associations[key] ||= Model.new(get_reflect(key).klass.where(''), key, self))
+        model.add(value)
+      end
+    end
+
+    public
+
+    def add(args)
+      return self if args == nil
+      args = [args] if not args.is_a?(Array)
+      args.each do |arg|
+        case arg
+        when Hash ; add_association(arg)
+        else      ; add_need_column(arg)
+        end
+      end
+      return self
+    end
+
+    # ----------------------------------------------------------------
+    # ● Load
+    # ----------------------------------------------------------------
+    private
+
+    def do_query(parent, reflect, relation)
+      parent_key = get_foreign_key(reflect)
+      relation_key = get_foreign_key(reflect, reverse: true, with_table_name: true)
+      ids = parent.map{|s| s[parent_key] }
+      ids.uniq!
+      ids.compact!
+      relation = with_conditions(reflect, relation)
+      query = { relation_key => ids }
+      query[reflect.type] = reflect.active_record.to_s if reflect.type
+
+      return get_association_scope(reflect).where(query) if use_association_to_query?(reflect)
+
+      join_table = get_join_table(reflect)
+      join_table = backtrace_possible_association(relation, join_table)
+
+      return relation.joins(join_table).where(query)
+    end
+
+    # Let city has_many :users, through: :schools
+    # And the query is: City.deep_pluck('users' => :name)
+    # We want to get the users data via `User.joins(:school).where(city_id: city_ids)`
+    # But get_join_table(reflect) returns `:schools` not :school
+    # No idea how to get the right association, so we try singularize or pluralize it.
+    def backtrace_possible_association(relation, join_table)
+      return join_table if relation.reflect_on_association(join_table)
+      join_table.to_s.singularize.to_sym.tap{|s| return s if relation.reflect_on_association(s) }
+      join_table.to_s.pluralize.to_sym.tap{|s| return s if relation.reflect_on_association(s) }
+      return nil
+    end
+
+    def set_includes_data(parent, column_name, model)
+      reflect = get_reflect(column_name)
+      reverse = !reflect.belongs_to?
+      foreign_key = get_foreign_key(reflect, reverse: reverse)
+      primary_key = get_foreign_key(reflect, reverse: !reverse)
+      children = model.load_data{|relation| do_query(parent, reflect, relation) }
+      # reverse = false: Child.where(:id => parent.pluck(:child_id))
+      # reverse = true : Child.where(:parent_id => parent.pluck(:id))
+      return DataCombiner.combine_data(
+        parent,
+        children,
+        primary_key,
+        column_name,
+        foreign_key,
+        reverse,
+        reflect.collection?,
+      )
+    end
+
+    def get_query_columns
+      if @parent_model
+        parent_reflect = @parent_model.get_reflect(@parent_association_key)
+        prev_need_columns = @parent_model.get_foreign_key(parent_reflect, reverse: true, with_table_name: true)
+      end
+      next_need_columns = @associations.map{|key, _| get_foreign_key(get_reflect(key), with_table_name: true) }.uniq
+      return [*prev_need_columns, *next_need_columns, *@need_columns].uniq(&Helper::TO_KEY_PROC)
+    end
+
+    def pluck_values(columns)
+      includes_values = @relation.includes_values
+      @relation.includes_values = []
+
+      result = @relation.pluck_all(*columns)
+
+      @relation.includes_values = includes_values
+      return result
+    end
+
+    def loaded_models
+      return [@model] if @model
+      return @relation if @relation.loaded
+    end
+
+    public
+
+    def load_data
+      columns = get_query_columns
+      key_columns = columns.map(&Helper::TO_KEY_PROC)
+      @relation = yield(@relation) if block_given?
+      @data = loaded_models ? loaded_models.as_json(root: false, only: key_columns) : pluck_values(columns)
+      if @data.size != 0
+        # for delete_extra_column_data!
+        @extra_columns = key_columns - @need_columns.map(&Helper::TO_KEY_PROC)
+        @associations.each do |key, model|
+          set_includes_data(@data, key, model)
+        end
+      end
+      return @data
+    end
+
+    def load_all
+      load_data
+      delete_extra_column_data!
+      return @data
+    end
+
+    def delete_extra_column_data!
+      return if @data.blank?
+      @data.each{|s| s.except!(*@extra_columns) }
+      @associations.each{|_, model| model.delete_extra_column_data! }
+    end
+
+    # ----------------------------------------------------------------
+    # ● Helper methods
+    # ----------------------------------------------------------------
+    module Helper
+      TO_KEY_PROC = proc{|s| Helper.column_to_key(s) }
+      def self.column_to_key(key) # user_achievements.user_id => user_id
+        key = key[/(\w+)[^\w]*\z/]
+        key.gsub!(/[^\w]+/, '')
+        return key
+      end
+    end
+  end
+end

data/lib/deep_pluck/version.rb

@@ -1,3 +1,3 @@
 module DeepPluck
-  VERSION = '1.1.4'
+  VERSION = '1.1.5'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: deep_pluck
 version: !ruby/object:Gem::Version
-  version: 1.1.4
+  version: 1.1.5
 platform: ruby
 authors:
 - khiav reoy
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-01-13 00:00:00.000000000 Z
+date: 2021-01-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -123,6 +123,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".github/FUNDING.yml"
 - ".gitignore"
 - ".rubocop.yml"
 - ".travis.yml"
@@ -140,6 +141,7 @@ files:
 - gemfiles/5.1.gemfile
 - gemfiles/5.2.gemfile
 - gemfiles/6.0.gemfile
+- gemfiles/6.1.gemfile
 - lib/deep_pluck.rb
 - lib/deep_pluck/data_combiner.rb
 - lib/deep_pluck/model.rb