Реализация отложенных перечислимых в Ruby

 def map(&block) Lax.new(self) end 

То, что мы действительно хотим, это заполнить объект урожай с элементами, которые модифицируются методом map . Это означает, что мы как-то должны передать две вещи в экземпляр Lax на map : объект- yielder и элемент для отображения. Мы можем передать их через аргумент блока следующим образом:

 def map(&block) Lax.new(self) do |yielder, val| yielder << block.call(val) end end 

Что это делает? Похоже, что блок map вызывается с помощью val , и этот элемент затем передается в урожай. Это не совсем точно, хотя. Это больше похоже на инструкции о том, как обрабатывать сопоставленное значение, хранящееся в получателе.

Поскольку Lax может получить блок, нам нужно изменить конструктор для обработки этого случая:

 class Lax < Enumerator def initialize(receiver) super() do |yielder| receiver.each do |val| if block_given? yield(yielder, val) else yielder << val end end end end end 

Когда предоставляется блок, как в случае с версией map Lax , мы получаем блок, передавая объект урожайности и элемент val .

Это работает? Давай выясним!

 p 1.upto(Float::INFINITY).lax .map { |x| x*x } .map { |x| x+1 } .first(5) 

Enumerable#first(5) возвращает первые 5 элементов перечислимого, что дает вам [2, 5, 10, 17, 26] .

Реализация Ленивый take

Теперь, когда мы увидели, как реализован map , давайте попробуем реализовать take . Как следует из его названия, Enumerable#take(n) возвращает первые n элементов из Enum. Ленивая версия вернет экземпляр Lax завернутый в Enumerable#take .

 class Lax < Enumerator # initialize and map goes here def take(n) taken = 0 Lax.new(self) do |yielder, val| if taken < n yielder << val taken += 1 else raise StopIteration end end end end 

Логика для take должна быть достаточно легкой для понимания. Здесь интересно то, как take сигналы об окончании итерации — она ​​вызывает исключение StopIteration .

Не сердитесь, потому что именно так это реализует Ruby. Мы можем обработать это исключение внутри конструктора и просто ничего не делать, когда обнаруживаем исключение StopIteration :

 class Lax < Enumerator def initialize(receiver) super() do |yielder| begin receiver.each do |val| if block_given? yield(yielder, val) else yielder << val end end rescue StopIteration end end end # map and take goes here ... end 

Прежде чем мы возьмем код для другого вращения, вот что у вас должно быть:

 module Enumerable def lax Lax.new(self) end end class Lax < Enumerator def initialize(receiver) super() do |yielder| begin receiver.each do |val| if block_given? yield(yielder, val) else yielder << val end end rescue StopIteration end end end def map(&block) Lax.new(self) do |yielder, val| yielder << block.call(val) end end def take(n) taken = 0 Lax.new(self) do |yielder, val| if taken < n yielder << val taken += 1 else raise StopIteration end end end end 

С этим, давайте попробуем код:

 p 1.upto(Float::INFINITY).lax .map { |x| x*x } .map { |x| x+1 } .take(10) .to_a 

Если вы получили [2, 5, 10, 17, 26] , то это сработало!

Резюме

Ленивые перечисления кажутся волшебными, когда вы впервые сталкиваетесь с ними. На самом деле, даже когда вы начнете разгадывать ленивые числа, вы, несомненно, найдете интересные кусочки Ruby, с которыми вы, возможно, не столкнулись бы иначе.

До написания этой главы я никогда не сталкивался с Enumerator::Yielder , Enumerator::Generator и Fiber . К счастью, реализация Enumerable в Rubinius очень помогла. Узнав, как поведение Fiber.yield стало для меня моментом «АГА!».