ree_lib 1.0.60 → 1.0.62

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 77d773db4104deb076fe0ea9d9aa6b8e0fd1218d71731e6d51a70a50be09d569
-  data.tar.gz: ac943abd83280ff44db19e50991e1ca51aa92cd1f140ccaa9e82cf706fc9dc8e
+  metadata.gz: e69b23ef012a6748604d4b82d1d81c791902ee23bc6179d231a340e48289a5d7
+  data.tar.gz: 6233c958a32bfdeea0efaf5b5692d0e824a5e04d9d334bf002d519b0789d6cc3
 SHA512:
-  metadata.gz: 6eb4b4567e977e12c59f5246f819dcba9702ccb77fb5e776584f193f6cff18c14d4e7e40d75c0377b0fc905720936f0feb888e2e953b288466f7d777ed7bd633
-  data.tar.gz: 26ee7d50cf9e346c4cb9614832282d586d0ac767dcd0b65bdb056511ed2e3058f5d1186003e6f30b89a9a661768c7b7153e4d2a3f079185b441f9b7ca5582485
+  metadata.gz: eee96a99fe5217675f4ff9c3db4915571ae444c7af4a33ca763b786b51a07f77f55fe794aa004688ccfd0f6c00f478631ba100181625de5d5f8d503f62b85edb
+  data.tar.gz: 1abc64ad140c4901e52f62b9a74f30e16f9fd4364a5e066f5321597c4246b7c5d7e2a0cba0515396b1a18aacd27d786321b17fa29004ed7336fac22a8ee75b06

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ree_lib (1.0.60)
+    ree_lib (1.0.62)
       binding_of_caller (~> 1.0.0)
       i18n (~> 1.12.0)
       loofah (~> 2.18.0)

data/lib/ree_lib/packages/ree_dao/package/ree_dao/association.rb

@@ -2,16 +2,16 @@ module ReeDao
   class Association
     include Ree::LinkDSL
 
-    link :demodulize, from: :ree_string
     link :group_by, from: :ree_array
     link :index_by, from: :ree_array
-    link :underscore, from: :ree_string
+    link :transform_values, from: :ree_hash
 
-    attr_reader :parent, :list, :global_opts
+    attr_reader :parent, :parent_dao_name, :list, :global_opts
 
-    contract(ReeDao::Associations, Array, Ksplat[RestKeys => Any] => Any)
-    def initialize(parent, list, **global_opts)
+    contract(ReeDao::Associations, Symbol, Array, Ksplat[RestKeys => Any] => Any)
+    def initialize(parent, parent_dao_name, list, **global_opts)
       @parent = parent
+      @parent_dao_name = parent_dao_name
       @list = list
       @global_opts = global_opts
     end
@@ -37,13 +37,18 @@ module ReeDao
       Optblock => Nilor[Array]
     )
     def load_association(assoc_type, assoc_name, **opts, &block)
+      opts[:autoload_children] ||= false
+
       assoc_index = load_association_by_type(
         assoc_type,
         assoc_name,
         **opts
       )
 
-      process_block(assoc_index, opts[:autoload_children] ||= false, &block) if block_given?
+      dao = find_dao(assoc_name, parent, opts[:scope])
+      dao_name = dao.first_source_table
+
+      process_block(assoc_index, opts[:autoload_children], opts[:to_dto], dao_name, &block) if block_given?
 
       list
     end
@@ -57,6 +62,7 @@ module ReeDao
       case type
       when :belongs_to
         one_to_one(
+          parent_dao_name,
           assoc_name,
           list,
           scope: opts[:scope],
@@ -67,44 +73,46 @@ module ReeDao
         )
       when :has_one
         one_to_one(
+          parent_dao_name,
           assoc_name,
           list,
           scope: opts[:scope],
           primary_key: opts[:primary_key],
           foreign_key: opts[:foreign_key],
+          to_dto: opts[:to_dto],
           setter: opts[:setter],
           reverse: true
         )
       when :has_many
         one_to_many(
+          parent_dao_name,
           assoc_name,
           list,
           scope: opts[:scope],
           primary_key: opts[:primary_key],
           foreign_key: opts[:foreign_key],
-          setter: opts[:setter]
-        )
-      else
-        one_to_many(
-          assoc_name,
-          list,
-          scope: opts[:scope],
-          primary_key: opts[:primary_key],
-          foreign_key: opts[:foreign_key],
+          to_dto: opts[:to_dto],
           setter: opts[:setter]
         )
       end
     end
 
-    contract(Or[Hash, Array], Bool, Block => Any)
-    def process_block(assoc, autoload_children, &block)
+    contract(Or[Hash, Array], Bool, Nilor[Proc], Symbol, Block => Any)
+    def process_block(assoc, autoload_children, to_dto, parent_dao_name, &block)
       assoc_list = assoc.is_a?(Array) ? assoc : assoc.values.flatten
 
+      if to_dto
+        assoc_list = assoc_list.map do |item|
+          to_dto.call(item)
+        end
+      end
+
       if ReeDao::Associations.sync_mode?
         ReeDao::Associations.new(
           parent.agg_caller,
           assoc_list,
           parent.local_vars,
+          parent_dao_name,
           autoload_children,
           **global_opts
         ).instance_exec(assoc_list, &block)
@@ -113,6 +121,7 @@ module ReeDao
           parent.agg_caller,
           assoc_list,
           parent.local_vars,
+          parent_dao_name,
           autoload_children,
           **global_opts
         ).instance_exec(assoc_list, &block)
@@ -132,16 +141,18 @@ module ReeDao
 
     contract(
       Symbol,
-      Array,      
+      Symbol,
+      Array,
       Kwargs[
         primary_key: Nilor[Symbol],
         foreign_key: Nilor[Symbol],
         scope: Nilor[Sequel::Dataset, Array],
         setter: Nilor[Or[Symbol, Proc]],
+        to_dto: Nilor[Proc],
         reverse: Bool
       ] => Or[Hash, Array]
     )
-    def one_to_one(assoc_name, list, scope: nil, primary_key: :id, foreign_key: nil, setter: nil, reverse: true)
+    def one_to_one(parent_assoc_name, assoc_name, list, scope: nil, primary_key: :id, foreign_key: nil, setter: nil, to_dto: nil, reverse: true)
       return {} if list.empty?
 
       primary_key ||= :id
@@ -150,13 +161,12 @@ module ReeDao
         items = scope
       else
         assoc_dao = find_dao(assoc_name, parent, scope)
-  
+
         if reverse
           if !foreign_key
-            name = underscore(demodulize(list.first.class.name))
-            foreign_key = "#{name}_id".to_sym
+            foreign_key = "#{parent_assoc_name.to_s.gsub(/s$/,'')}_id".to_sym
           end
-  
+
           root_ids = list.map(&:id).uniq
         else
           if !foreign_key
@@ -165,13 +175,14 @@ module ReeDao
 
           root_ids = list.map(&:"#{foreign_key}").compact
         end
-  
+
         scope ||= assoc_dao
         scope = scope.where((reverse ? foreign_key : :id) => root_ids)
-        
+
         items = add_scopes(scope, global_opts[assoc_name])
       end
 
+
       assoc = index_by(items) { _1.send(reverse ? foreign_key : :id) }
 
       populate_association(
@@ -180,23 +191,26 @@ module ReeDao
         assoc_name,
         setter: setter,
         reverse: reverse,
-        primary_key: primary_key
+        primary_key: primary_key,
+        to_dto: to_dto
       )
 
       assoc
     end
 
     contract(
+      Symbol,
       Symbol,
       Array,
       Kwargs[
         foreign_key: Nilor[Symbol],
         primary_key: Nilor[Symbol],
         scope: Nilor[Sequel::Dataset, Array],
-        setter: Nilor[Or[Symbol, Proc]]
+        setter: Nilor[Or[Symbol, Proc]],
+        to_dto: Nilor[Proc]
       ] => Or[Hash, Array]
     )
-    def one_to_many(assoc_name, list, primary_key: nil, foreign_key: nil, scope: nil, setter: nil)
+    def one_to_many(parent_assoc_name, assoc_name, list, primary_key: nil, foreign_key: nil, scope: nil, setter: nil, to_dto: nil)
       return {} if list.empty?
 
       primary_key ||= :id
@@ -206,14 +220,14 @@ module ReeDao
       else
         assoc_dao = nil
         assoc_dao = find_dao(assoc_name, parent, scope)
-  
-        foreign_key ||= "#{underscore(demodulize(list.first.class.name))}_id".to_sym
-  
+
+        foreign_key ||= "#{parent_assoc_name.to_s.gsub(/s$/, '')}_id".to_sym
+
         root_ids = list.map(&:"#{primary_key}")
-  
+
         scope ||= assoc_dao
         scope = scope.where(foreign_key => root_ids)
-  
+
         items = add_scopes(scope, global_opts[assoc_name])
       end
 
@@ -229,6 +243,7 @@ module ReeDao
         assoc_name,
         setter: setter,
         primary_key: primary_key,
+        to_dto: to_dto,
         multiple: true
       )
 
@@ -243,10 +258,11 @@ module ReeDao
         primary_key: Nilor[Symbol],
         reverse: Nilor[Bool],
         setter: Nilor[Or[Symbol, Proc]],
+        to_dto: Nilor[Proc],
         multiple: Bool
       ] => Any
     )
-    def populate_association(list, association_index, assoc_name, primary_key: nil, reverse: nil, setter: nil, multiple: false)
+    def populate_association(list, association_index, assoc_name, primary_key: nil, reverse: nil, setter: nil, to_dto: nil, multiple: false)
       assoc_setter = if setter
         setter
       else
@@ -255,16 +271,35 @@ module ReeDao
 
       list.each do |item|
         if setter && setter.is_a?(Proc)
-          self.instance_exec(item, association_index, &assoc_setter)
+          if to_dto
+            assoc_index = transform_values(association_index) do |key, value|
+              to_dto.call(value)
+            end
+
+            self.instance_exec(item, assoc_index, &assoc_setter)
+          else
+            self.instance_exec(item, association_index, &assoc_setter)
+          end
         else
           key = if reverse.nil?
             primary_key
           else
             reverse ? primary_key : "#{assoc_name}_id"
           end
+
           value = association_index[item.send(key)]
 
-          value = [] if value.nil? && multiple
+          if to_dto && !value.nil?
+            if value.is_a?(Array)
+              value = value.map { to_dto.call(_1) }
+            else
+              value = to_dto.call(value)
+            end
+          end
+
+          if value.nil? && multiple
+            value = []
+          end
 
           begin
             item.send(assoc_setter, value)

data/lib/ree_lib/packages/ree_dao/package/ree_dao/associations.rb

@@ -2,15 +2,16 @@ module ReeDao
   class Associations
     include Ree::LinkDSL
 
-    attr_reader :agg_caller, :list, :local_vars, :only, :except, :autoload_children, :global_opts
+    attr_reader :agg_caller, :list, :local_vars, :only, :except, :parent_dao_name, :autoload_children, :global_opts
 
-    def initialize(agg_caller, list, local_vars, autoload_children = false, **opts)
+    def initialize(agg_caller, list, local_vars, parent_dao_name, autoload_children = false, **opts)
       @agg_caller = agg_caller
       @list = list
       @local_vars = local_vars
       @global_opts = opts || {}
       @only = opts[:only] if opts[:only]
       @except = opts[:except] if opts[:except]
+      @parent_dao_name = parent_dao_name
       @autoload_children = autoload_children
 
       raise ArgumentError.new("you can't use both :only and :except arguments at the same time") if @only && @except
@@ -83,7 +84,7 @@ module ReeDao
       if self.class.sync_mode?
         return if association_is_not_included?(assoc_name) || list.empty?
         
-        association = Association.new(self, list, **global_opts)
+        association = Association.new(self, parent_dao_name, list, **global_opts)
         if assoc_type == :field
           association.handle_field(assoc_name, opts)
         else
@@ -94,7 +95,7 @@ module ReeDao
           return { association_threads: @assoc_threads, field_threads: @field_threads } 
         end
 
-        association = Association.new(self, list, **global_opts)
+        association = Association.new(self, parent_dao_name, list, **global_opts)
 
         if assoc_type == :field
           field_proc = opts

data/lib/ree_lib/packages/ree_dao/package/ree_dao/functions/load_agg.rb

@@ -4,6 +4,8 @@ class ReeDao::LoadAgg
   include Ree::FnDSL
 
   fn :load_agg do
+    link :demodulize, from: :ree_string
+    link :underscore, from: :ree_string
     link "ree_dao/associations", -> { Associations }
     link "ree_dao/contract/dao_dataset_contract", -> { DaoDatasetContract }
     link "ree_dao/contract/entity_contract", -> { EntityContract }
@@ -34,6 +36,12 @@ class ReeDao::LoadAgg
       ids_or_scope
     end
 
+    if dao
+      dao_name = dao.first_source_table
+    else
+      dao_name = underscore(demodulize(scope.first.class.name)).to_sym
+    end
+
     list = scope.is_a?(Sequel::Dataset) ? scope.all : scope
 
     if opts[:to_dto]
@@ -42,7 +50,7 @@ class ReeDao::LoadAgg
       end
     end
 
-    load_associations(list, **opts, &block) if block_given?
+    load_associations(dao_name, list, **opts, &block) if block_given?
 
     if ids_or_scope.is_a?(Array)
       list.sort_by { ids_or_scope.index(_1.id) }
@@ -53,7 +61,7 @@ class ReeDao::LoadAgg
 
   private
 
-  def load_associations(list, **opts, &block)
+  def load_associations(dao_name, list, **opts, &block)
     return if list.empty?
 
     local_vars = block.binding.eval(<<-CODE, __FILE__, __LINE__ + 1)
@@ -63,7 +71,7 @@ class ReeDao::LoadAgg
 
     agg_caller = block.binding.eval("self")
 
-    associations = Associations.new(agg_caller, list, local_vars, **opts).instance_exec(list, &block)
+    associations = Associations.new(agg_caller, list, local_vars, dao_name, **opts).instance_exec(list, &block)
 
     if ReeDao.load_sync_associations_enabled?
       associations

data/lib/ree_lib/packages/ree_dao/schemas/ree_dao/functions/load_agg.schema.json

@@ -38,6 +38,21 @@
     }
   ],
   "links": [
+    {
+      "target": "demodulize",
+      "package_name": "ree_string",
+      "as": "demodulize",
+      "imports": [
+
+      ]
+    },
+    {
+      "target": "underscore",
+      "package_name": "ree_string",
+      "as": "underscore",
+      "imports": [
 
+      ]
+    }
   ]
 }

data/lib/ree_lib/packages/ree_dao/spec/ree_dao/functions/load_agg/load_agg_benchmark_spec.rb

@@ -129,7 +129,7 @@ RSpec.describe :load_agg do
     end
 
     def call(ids_or_scope)
-      load_agg(ids_or_scope, users) do
+      load_agg(users, ids_or_scope) do
         belongs_to :organization
 
         has_many :books
@@ -141,7 +141,7 @@ RSpec.describe :load_agg do
         has_many :skills
         has_many :dreams
 
-        has_one :passport, foreign_key: :user_id, scope: user_passports
+        has_one :passport, -> { {foreign_key: :user_id, scope: user_passports} }
       end
     end
   end
@@ -251,7 +251,7 @@ RSpec.describe :load_agg do
 
     _users = []
     st_time = Time.now
-    puts "Starting to seed #{NUM_OF_USERS} users..."
+    puts "Seeding #{NUM_OF_USERS} users..."
     NUM_OF_USERS.times do
       u = ReeDaoLoadAggTest::User.new(
         name: Faker::Name.name,

data/lib/ree_lib/packages/ree_dao/spec/ree_dao/functions/load_agg/load_agg_spec.rb

@@ -104,7 +104,7 @@ RSpec.describe :load_agg do
         has_many :books do |books_list|
           has_one :author
           has_many :chapters
-        
+
           has_many :reviews do |reviews_list|
             has_one :review_author
 
@@ -113,7 +113,7 @@ RSpec.describe :load_agg do
 
           field :book_calculatetable_field, -> { change_book_titles(books_list) }
         end
-        
+
         has_one :passport, -> { passport_opts }
         has_one :custom_field, -> { custom_field_opts }
 
@@ -153,6 +153,9 @@ RSpec.describe :load_agg do
 
     aggregate :users_agg_with_dto do
       link :users, from: :ree_dao_load_agg_test
+      link :books, from: :ree_dao_load_agg_test
+      link :authors, from: :ree_dao_load_agg_test
+      link :chapters, from: :ree_dao_load_agg_test
       link :organizations, from: :ree_dao_load_agg_test
       link :load_agg, from: :ree_dao
     end
@@ -160,8 +163,32 @@ RSpec.describe :load_agg do
     def call(ids_or_scope, **opts)
       load_agg(users, ids_or_scope, **opts) do
         belongs_to :organization
+
+        has_many :books, -> { books_opts } do
+          has_one :author, -> { author_opts }
+          has_many :chapters, -> { chapters_opts }
+        end
       end
     end
+
+    private
+
+    def books_opts
+      {
+        to_dto: -> (book) { ReeDaoLoadAggTest::BookDto.new(book) },
+        setter: -> (item, child_index) {
+          item.set_books(child_index[item.id] || [])
+        }
+      }
+    end
+
+    def author_opts
+      { to_dto: -> (author) { ReeDaoLoadAggTest::AuthorDto.new(author) }}
+    end
+
+    def chapters_opts
+      { to_dto: -> (chapter) { ReeDaoLoadAggTest::ChapterDto.new(chapter) }}
+    end
   end
 
   class ReeDaoLoadAggTest::UsersAggAutoloadBooksChildren
@@ -183,7 +210,7 @@ RSpec.describe :load_agg do
         has_many :books, -> { books_opts } do
           has_one :author
           has_many :chapters
-        
+
           has_many :reviews do
             has_one :review_author
           end
@@ -192,7 +219,7 @@ RSpec.describe :load_agg do
     end
 
     private
-    
+
     def books_opts
       { autoload_children: true }
     end
@@ -217,7 +244,7 @@ RSpec.describe :load_agg do
         has_many :books do
           has_one :author
           has_many :chapters
-        
+
           has_many :reviews, -> { { autoload_children: true } } do
             has_one :review_author
           end
@@ -370,6 +397,12 @@ RSpec.describe :load_agg do
     users.put(user_1)
     users.put(user_2)
 
+    book_1 = ReeDaoLoadAggTest::Book.new(user_id: user_1.id, title: "1984")
+    books.put(book_1)
+
+    authors.put(ReeDaoLoadAggTest::Author.new(book_id: book_1.id, name: "George Orwell"))
+    chapters.put(ReeDaoLoadAggTest::Chapter.new(book_id: book_1.id, title: "interlude"))
+
     res = users_agg_with_dto.call(
       users.all,
       to_dto: -> (user) {
@@ -382,7 +415,12 @@ RSpec.describe :load_agg do
       }
     )
 
+    book = res.first.books.first
+
     expect(res.first.class).to eq(ReeDaoLoadAggTest::UserDto)
+    expect(book.class).to eq(ReeDaoLoadAggTest::BookDto)
+    expect(book.author.class).to eq(ReeDaoLoadAggTest::AuthorDto)
+    expect(book.chapters.first.class).to eq(ReeDaoLoadAggTest::ChapterDto)
   }
 
   it {
@@ -402,7 +440,7 @@ RSpec.describe :load_agg do
     books.put(book_1)
     books.put(book_2)
     books.put(book_3)
-    
+
     res = users_agg_only_dataset.call(users.where(name: "John"))
 
     user = res[0]

data/lib/ree_lib/packages/ree_dao/spec/ree_dao/functions/load_agg/ree_dao_load_agg_test.rb

@@ -75,6 +75,9 @@ class ReeDaoLoadAggTest::Book
   attr_accessor :title, :user_id
 end
 
+class ReeDaoLoadAggTest::BookDto < SimpleDelegator
+end
+
 class ReeDaoLoadAggTest::User
   include ReeDto::EntityDSL
 
@@ -159,6 +162,14 @@ class ReeDaoLoadAggTest::UserDto
   def organization
     @organization
   end
+
+  def set_books(books)
+    @books = books; nil
+  end
+
+  def books
+    @books
+  end
 end
 
 class ReeDaoLoadAggTest::Organization
@@ -276,6 +287,9 @@ class ReeDaoLoadAggTest::Chapter
   attr_accessor :title, :book_id
 end
 
+class ReeDaoLoadAggTest::ChapterDto < SimpleDelegator
+end
+
 class ReeDaoLoadAggTest::Author
   include ReeDto::EntityDSL
 
@@ -288,6 +302,9 @@ class ReeDaoLoadAggTest::Author
   attr_accessor :name, :book_id
 end
 
+class ReeDaoLoadAggTest::AuthorDto < SimpleDelegator
+end
+
 class ReeDaoLoadAggTest::Review
   include ReeDto::EntityDSL
 

data/lib/ree_lib/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReeLib
-  VERSION = "1.0.60"
+  VERSION = "1.0.62"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ree_lib
 version: !ruby/object:Gem::Version
-  version: 1.0.60
+  version: 1.0.62
 platform: ruby
 authors:
 - Ruslan Gatiyatov