summaryrefslogtreecommitdiff
path: root/lib/humble/result_set.rb
blob: 0a0d4ee0de8dc7ee227bb81d5be4fd5de83863ce (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
module Humble
  class ResultSet
    include Enumerable

    def initialize(rows, mapper)
      @rows = rows
      @mapper = mapper
    end

    def each(&block)
      @rows.each do |row|
        block.call(@mapper.map_from(row))
      end
    end

    def include?(item)
      self.find do |x|
        x == item
      end
    end

    def inspect
      "[#{self.map { |x| x.inspect }.join(", ")}]"
    end
  end
end