紅寶石：實現函數返回一個枚舉

Question

對於個人Hash執行，我想創建的map，map!，即一個拍着版本，改變當前對象，就像存在Array的一個版本。當一個塊已定，那麼執行下面似乎正常工作：

class MyHash < Hash 

    def map! &block 
    self.map { |ek, ev| self[ek] = block[ev] } if block_given? 
    # how to return the Enumerator if block not given? 
    end 

end 

當不使用塊調用，hash.map返回Enumerator，就像下面的例子：

h.map 
=> #<Enumerator: {1=>2, 12=>21, 6=>3}:map> 

我要補充到我的函數，使其返回枚舉器時，塊沒有給出？

Answer 1

我不能讓你的代碼做什麼，但你可以嘗試返回self.each：

def map! &block 
    if block_given? 
     self.map { |ek, ev| self[ek] = block[ev] } 
    else 
     self.each 
    end 
end 

Answer 2

這是非常容易使用Object#to_enum

class MyHash < Hash 
    def map! 
    if block_given? 
     each { |k, v| self[k] = yield v } 
    else 
     to_enum :map! 
    end 
    end 
end