Ruby의 Enumerableeach 진도를 표시하는 방법

컨디션

이루어지다

module Enumerable
  def each_with_progress(&block)
    self.each_with_index do |element, index|
      print "\r#{index + 1}/#{self.length}"
      block.call(element, index)
    end
    print "\r\n"
  end
end

호출단 코드

puts "# program start"

['a','b','c'].each_with_progress do |elem|
  sleep 1
  print " #{elem}"
end

['a','b','c'].each_with_progress do |elem, index|
  sleep 1
  print " #{index} -> #{elem}"
end

puts "# program end"

실행 결과

$ ruby index.rb
# program start
3/3 c
3/3 2 -> c
# program end

Enumerable를 통한 구현

module Enumerable
  def each_with_progress(&block)
    length = self.respond_to?(:length) ? self.length : self.count

    self.each_with_index do |element, index|
      print "\r#{index + 1}/#{length}"
      block.call(element, index)
    end
    print "\r\n"
  end
end

호출단 코드

puts "# program start"

sum = 0
(1..10000).each_with_progress do |elem|
  sum += elem
end
puts sum

puts "# program end"

실행 결과

$ ruby index.rb
# program start
10000/10000
50005000
# program end

참고 자료