Contents

module Rasti
  module DB
    module Relations
      class OneToMany < Base

        def foreign_key
          @foreign_key ||= options[:foreign_key] || source_collection_class.foreign_key
        end

        def graph_to(rows, db, schema=nil, relations=[])
          pks = rows.map { |row| row[source_collection_class.primary_key] }.uniq

          target_collection = target_collection_class.new db, schema

          relation_rows = target_collection.where(foreign_key => pks)
                                           .graph(*relations)
                                           .group_by { |r| r.public_send(foreign_key) }

          rows.each do |row| 
            row[name] = build_graph_result relation_rows.fetch(row[source_collection_class.primary_key], [])
          end
        end

        def join_to(dataset, schema=nil, prefix=nil)
          relation_alias = join_relation_name prefix

          qualified_relation_source = prefix ? Sequel[prefix] : qualified_source_collection_name(schema)

          relation_condition = {
            Sequel[relation_alias][foreign_key] => qualified_relation_source[source_collection_class.primary_key]
          }

          dataset.join(qualified_target_collection_name(schema).as(relation_alias), relation_condition)
        end

        def apply_filter(dataset, schema=nil, primary_keys=[])
          target_name = qualified_target_collection_name schema

          dataset.join(target_name, foreign_key => source_collection_class.primary_key)
                 .where(Sequel[target_name][target_collection_class.primary_key] => primary_keys)
                 .select_all(qualified_source_collection_name(schema))
                 .distinct
        end

        private

        def build_graph_result(rows)
          rows
        end

      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
rasti-db-1.4.0	lib/rasti/db/relations/one_to_many.rb
rasti-db-1.3.1	lib/rasti/db/relations/one_to_many.rb
rasti-db-1.3.0	lib/rasti/db/relations/one_to_many.rb
rasti-db-1.2.0	lib/rasti/db/relations/one_to_many.rb
rasti-db-1.1.1	lib/rasti/db/relations/one_to_many.rb
rasti-db-1.1.0	lib/rasti/db/relations/one_to_many.rb