루비 이넘레이터 생성하기

루비는 강력한 프로그래밍 언어로, 다양한 데이터 구조를 다루는 데 유용한 도구를 제공합니다. 그 중 하나가 바로 이넘레이터(Enumerator)입니다. 이넘레이터는 컬렉션을 순회하는 데 도움을 주며, 특히 반복 작업을 간편하게 만들어 줍니다. 이번 글에서는 루비에서 이넘레이터를 생성하고 사용하는 방법에 대해 알아보겠습니다.

이넘레이터란?

이넘레이터는 루비에서 컬렉션을 순회할 수 있는 객체입니다. 이넘레이터를 사용하면 배열, 해시, 또는 다른 컬렉션을 쉽게 반복할 수 있습니다. 이넘레이터는 특히 무한한 시퀀스를 생성하거나, 복잡한 데이터 구조를 다룰 때 유용합니다.

이넘레이터의 기본 사용법

이넘레이터를 생성하는 가장 간단한 방법은 Enumerator.new 메서드를 사용하는 것입니다. 다음은 기본적인 이넘레이터 생성 예제입니다:

enumerator = Enumerator.new do |yielder|
  yielder << 1
  yielder << 2
  yielder << 3
end

enumerator.each do |value|
  puts value
end

위의 코드에서 yielder는 이넘레이터가 값을 생성할 때 사용하는 객체입니다. yielder << value를 통해 값을 추가할 수 있습니다. 이 코드를 실행하면 1, 2, 3이 출력됩니다.

이넘레이터의 다양한 생성 방법

이넘레이터는 여러 가지 방법으로 생성할 수 있습니다. 여기서는 몇 가지 방법을 소개하겠습니다.

1. 배열에서 이넘레이터 생성하기

배열에서 이넘레이터를 생성하는 것은 매우 간단합니다. to_enum 메서드를 사용하면 됩니다:

array = [1, 2, 3, 4, 5]
enumerator = array.to_enum

enumerator.each do |value|
  puts value
end

이 코드는 배열의 각 요소를 순회하여 출력합니다.

2. 무한 이넘레이터 생성하기

무한 이넘레이터를 생성하는 것도 가능합니다. 다음은 무한한 자연수를 생성하는 예제입니다:

natural_numbers = Enumerator.new do |yielder|
  number = 1
  loop do
    yielder << number
    number += 1
  end
end

# 처음 10개의 자연수 출력
natural_numbers.take(10).each do |value|
  puts value
end

위의 코드에서 loop do를 사용하여 무한 루프를 생성하고, yielder << number를 통해 자연수를 계속 추가합니다. take(10) 메서드를 사용하여 처음 10개의 자연수를 가져옵니다.

3. 블록을 사용한 이넘레이터 생성하기

이넘레이터를 생성할 때 블록을 사용할 수도 있습니다. 다음은 블록을 사용하여 이넘레이터를 생성하는 예제입니다:

enumerator = Enumerator.new do |yielder|
  (1..5).each do |i|
    yielder << i * 2
  end
end

enumerator.each do |value|
  puts value
end

이 코드는 1부터 5까지의 숫자를 두 배로 만들어 출력합니다.

이넘레이터의 유용한 메서드

이넘레이터는 다양한 메서드를 제공하여 데이터를 쉽게 다룰 수 있도록 도와줍니다. 여기서는 몇 가지 유용한 메서드를 소개하겠습니다.

next: 이넘레이터의 다음 값을 반환합니다.
peek: 이넘레이터의 다음 값을 반환하지만, 이넘레이터의 상태는 변경하지 않습니다.
rewind: 이넘레이터를 처음 상태로 되돌립니다.
with_index: 이넘레이터의 각 요소에 인덱스를 추가하여 반환합니다.

next 메서드 사용하기

next 메서드를 사용하여 이넘레이터의 다음 값을 가져오는 예제입니다:

enumerator = [1, 2, 3].to_enum

puts enumerator.next  # 1
puts enumerator.next  # 2
puts enumerator.next  # 3

with_index 메서드 사용하기

with_index 메서드를 사용하여 인덱스와 함께 값을 출력하는 예제입니다:

enumerator = ['a', 'b', 'c'].to_enum

enumerator.with_index do |value, index|
  puts "#{index}: #{value}"
end

이 코드는 각 요소와 그 인덱스를 출력합니다.

이넘레이터의 활용 예제

이넘레이터는 다양한 상황에서 유용하게 사용될 수 있습니다. 다음은 몇 가지 활용 예제입니다.

1. 파일 읽기

이넘레이터를 사용하여 파일을 한 줄씩 읽는 예제입니다:

file = File.open("example.txt")
enumerator = file.each_line

enumerator.each do |line|
  puts line.chomp
end

file.close

이 코드는 example.txt 파일의 각 줄을 읽어 출력합니다.

2. 데이터베이스 쿼리 결과 처리

이넘레이터를 사용하여 데이터베이스 쿼리 결과를 처리하는 예제입니다:

require 'sqlite3'

db = SQLite3::Database.new "test.db"
enumerator = db.execute("SELECT name FROM users").to_enum

enumerator.each do |row|
  puts row[0]
end

이 코드는 데이터베이스에서 사용자 이름을 가져와 출력합니다.

결론

이넘레이터는 루비에서 컬렉션을 다루는 데 매우 유용한 도구입니다. 이넘레이터를 사용하면 반복 작업을 간편하게 처리할 수 있으며, 다양한 데이터 구조를 쉽게 다룰 수 있습니다. 이번 글에서 소개한 내용을 바탕으로 이넘레이터를 활용하여 더 효율적인 코드를 작성해 보세요!