有沒有辦法將大小傳遞給`enum_for` /`to_enum`？

Question

我工作了很多的元素的2D網格，我有時需要枚舉這些元素，所以我有一個類，看起來像這樣：

class Grid 
    def initialize(width, height) 
    @elements = Array.new(width * height, DEFAULT_ELEMENT) 
    end 

    def each(&block) 
    return enum_for(:each) unless block_given? 

    @elements.each(&block) 

    return self 
    end 
end 

這一直永遠的罰款，但今天才我是否意識到由#each返回的Enumerator不會延遲計算大小。我可以手動定義一個Enumerator，但除了each之外，我還有不少調查員，那樣會很乏味。

所以我想知道是否有像enum_for(:method, lazy_size)這樣的東西，以便我可以寫enum_for(:each width * height)。

Answer 1

你可以傳遞一個塊enum_for計算枚舉的大小：

enum_for(:each) { width * height } 

Answer 2

如果您計劃支持各種類型枚舉的，可能是你應該考慮實施Enumerable混入

DEFAULT_ELEMENT = ":" 

class Grid 
    include Enumerable 

    def initialize(width, height) 
    @elements = Array.new(width * height, DEFAULT_ELEMENT) 
    end 

    def each 
    @elements.each {|i| yield i} 
    end 

end 


p Grid.new(2,2).map {|i| "#{i}-)"} 
#=> [":-)", ":-)", ":-)", ":-)"] 

p Grid.new(4,6).count 
#=> 24