自定义ruby注入方法未产生预期输出



请原谅,我看到论坛上有很多人问过这个问题,但没有一个能输出所需的输出,我已经读了几天关于注入方法如何工作的不同解释。是否可以检查我的代码,看看我做错了什么或缺少什么?我正在学习这门语言,不想因为缺乏经验而感到沮丧。

我的代码:


def my_inject(initial = nil, sym = nil, &block)
check = initial.nil?
acc = check ? 0 : 1

if block_given?
my_each { |x| acc = yield(acc, x) }
end
acc
end
alias_method :my_reduce, :my_inject

我一直在使用ruby.doc.org中的测试用例进行测试。测试用例:

# Sum some numbers
(5..10).reduce(:+)                             #=> 45
# Same using a block and inject
(5..10).inject { |sum, n| sum + n }            #=> 45
# Multiply some numbers
(5..10).reduce(1, :*)                          #=> 151200
# Same using a block
(5..10).inject(1) { |product, n| product * n } #=> 151200
# find the longest word
longest = %w{ cat sheep bear }.inject do |memo, word|
memo.length > word.length ? memo : word
end
longest                                        #=> "sheep"

我的代码适用于除了最后一个返回绵羊的块之外的所有具有块的代码。我很感激能深入了解我做错了什么。

好的,这就是我现在拥有的:

def my_inject(initial = nil, sym = nil, &block)
acc = initial.nil? ? 0 : 1
sym ||= initial if initial.is_a? Symbol
if !block_given?
block = sym.to_proc
my_each { |x| acc = block.call(acc, x) }
else
my_each { |x| acc = yield(acc, x) }
end
acc
end
alias_method :my_reduce, :my_inject

它适用于前四个测试用例,但不适用于第五个,而且第一个测试用例返回46而不是45,这是因为我使用了检查,但我尝试过其他不起作用的测试用例

我能够用代码解决它。

def my_inject(initial = nil, second = nil)
arr = is_a?(Array) ? self : to_a
sym = initial if initial.is_a?(Symbol) || initial.is_a?(String)
acc = initial if initial.is_a? Integer
if initial.is_a?(Integer)
sym = second if second.is_a?(Symbol) || second.is_a?(String)
end
if sym
arr.my_each { |x| acc = acc ? acc.send(sym, x) : x }
elsif block_given?
arr.my_each { |x| acc = acc ? yield(acc, x) : x }
end
acc
end
alias my_reduce my_injec

最新更新