Module: ForestAdminDatasourceMongoid::Utils::AddNullValues

Included in:
Collection
Defined in:
lib/forest_admin_datasource_mongoid/utils/add_null_values.rb

Instance Method Summary collapse

Instance Method Details

#add_null_values(records, projection) ⇒ Object 



51
52
53
 
# File 'lib/forest_admin_datasource_mongoid/utils/add_null_values.rb', line 51

def add_null_values(records, projection)
  records.filter_map { |record| add_null_values_on_record(record, projection) }
end

#add_null_values_on_record(record, projection) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/forest_admin_datasource_mongoid/utils/add_null_values.rb', line 21

def add_null_values_on_record(record, projection)
  return nil if record.nil?

  result = record.dup

  projection.each do |field|
    field_prefix = field.split(':').first
    result[field_prefix] ||= nil
  end

  nested_prefixes = projection.select { |field| field.include?(':') }.map { |field| field.split(':').first }.uniq

  nested_prefixes.each do |nested_prefix|
    child_paths = projection.filter { |field| field.start_with?("#{nested_prefix}:") }
                            .map { |field| field[(nested_prefix.size + 1)..] }

    next unless result[nested_prefix] && !result[nested_prefix].nil?

    if result[nested_prefix].is_a?(Array)
      result[nested_prefix] = result[nested_prefix].map do |child_record|
        add_null_values_on_record(child_record, child_paths)
      end
    elsif result[nested_prefix].is_a?(Hash)
      result[nested_prefix] = add_null_values_on_record(result[nested_prefix], child_paths)
    end
  end

  remove_not_exist_record(result)
end

#remove_not_exist_record(record) ⇒ Object

Filter out records that have been tagged as not existing If the key FOREST_RECORD_DOES_NOT_EXIST is present in the record, the record is removed If a nested object has a key with FOREST_RECORD_DOES_NOT_EXIST, the nested object is removed



7
8
9
10
11
12
13
14
15
16
17
18
19
 
# File 'lib/forest_admin_datasource_mongoid/utils/add_null_values.rb', line 7

def remove_not_exist_record(record)
  return nil if record.nil? || record[Pipeline::ConditionGenerator::FOREST_RECORD_DOES_NOT_EXIST]

  record.transform_values! do |value|
    if value.is_a?(Hash) && value[Pipeline::ConditionGenerator::FOREST_RECORD_DOES_NOT_EXIST]
      nil
    else
      value
    end
  end

  record
end