summaryrefslogtreecommitdiff
path: root/app/models/data_list.rb
blob: 2cee34478861dfd1c83b0cd077e3abf2887a305b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# frozen_string_literal: true

class DataList
  def initialize(batch_ids, data_fields_hash, klass)
    @batch_ids = batch_ids
    @data_fields_hash = data_fields_hash
    @klass = klass
  end

  def to_array
    [klass, columns, values]
  end

  private

  attr_reader :batch_ids, :data_fields_hash, :klass

  def columns
    data_fields_hash.keys << 'service_id'
  end

  def values
    batch_ids.map do |row|
      data_fields_hash.values << row['id']
    end
  end
end