如何在 ruby​​ 中编写一个采用散列样式、方括号括起来的参数的方法，比如 mymethod[arg]？

Question

我想用 ruby​​ 编写一个方法来替换当前只返回散列的方法。

所以目前我们有这个

def some_stats
  {
    :foo => "bar",
    :zoo => "zar"
  }
end

可以这样调用：

some_stats[:foo]

我想用一个用 :foo 调用的方法来替换它，它会做一些更复杂的事情，只是构建一个散列。 但是，我希望以相同的方式调用它，并使用方括号表示法。 例如

#this is a bit pseudo-cody but hopefully you get the idea.
def some_stats[](key)
  if key.to_s =~ /something/
    #do something
    return something
  else
    #do something else
    return something_else
  end
end

这仍然会像some_stats[:foo]一样被调用

我无法计算出定义方法名称的语法：上面的def some_stats[](key)不起作用。 不过我觉得这应该是可能的。

Answer 1

你很接近。 object[sth]只是object.[](sth)的语法糖。 因此，要执行您需要的操作，您必须定义some_stats方法，该方法返回定义[]方法的对象：

class Stats
  def [](key)
    if key.to_s =~ /something/
     #do something
     return something
  else
    #do something else
    return something_else
  end
end

def some_stats
  Stats.new
end

some_stats[:something_new] #=> something
some_stats[:not_new] #=> something_else

Answer 2

您的原始方法不是用方括号调用的； 它在没有任何参数的情况下被调用，但返回一个对象（在你的情况下是一个Hash ），它定义了方法:[] ，因此理解方括号。 您的方法调用相当于

(some_stats())[:foo]

或者，更明确地说，

x = some_stats()
x[:foo]

因此你需要 - 在你的方法中返回一个提供[]方法的类的实例。 假设您编写了一个名为MyClass的类，它定义了一个[] 。 然后你可以写一个函数

def some_stats
  if _your condition goes here_
    { ... } # Return your Hash
  else
    Myclass.new
  end
end

这将用作

some_stats[:whatever]

但是，要真正有用，您的条件将使用该方法的参数，即

def some_stats(param)
  if param < 4000
    { ... } # Return your Hash
  else
    Myclass.new
  end
end

你会调用它

some_stats(4711)[:whatever]