从方法外部获取方法内定义的局部变量名称

时间:2016-06-11 02:13:02

标签: ruby scope metaprogramming

Ruby中是否有可能使用元编程从方法外部获取方法中定义的所有局部变量名称?

def foo
  var = 100
  arr = [1,2]
end

foo.local_variables

我只需要检索变量名,而不是它们的值。

为什么我要找那些变量名:我有一个类,我想为“#initialize”中初始化的所有实例变量动态生成setter。

为了做到这一点,在“初始化”中,我使用的是instance_variables.each do |inst_var_name| define_singleton_method(...) do ...

这是有效的:每个实例对象都将其setter作为singleton方法。但是,我正在寻找一个解决方案,我可以从“initialize”方法之外读取实例变量名称,以便能够为它们使用“#define_method”,并创建常规方法而不是单例方法。

2 个答案:

答案 0 :(得分:1)

您可以(重新) - 解析方法并检查S-EXPR树。请参阅下面的概念证明。您可以使用Method#source_location获取定义方法的文件,然后读取该文件。我的代码肯定有改进的余地,但应该让你开始。它是一个功能齐全的代码,只需要ruby解析器gem(https://github.com/whitequark/parser)。

require 'parser/current'
node = Parser::CurrentRuby.parse(DATA.read) # DATA is everything that comes after __END__

def find_definition(node, name)
  return node if definition_node?(node, name)
  if node.respond_to?(:children)
    node.children.find do |child|
      find_definition(child, name)
    end
  end
end

def definition_node?(node, name)
  return false if !node.respond_to?(:type)
  node.type == :def && node.children.first == name
end

def collect_lvasgn(node)
  result = []
  return result if !node.respond_to?(:children)

  node.children.each do |child|
    if child.respond_to?(:type) && child.type == :lvasgn
      result << child.children.first
    else
      result += collect_lvasgn(child)
    end
  end

  result
end

definition = find_definition(node, :foo)
puts collect_lvasgn(definition)

__END__

def foo
  var = 100
  arr = [1,2]
  if something
    this = 3 * var
  end
end

def bar
  var = 200
  arr = [3, 4]
end

你介意告诉我们你为什么要找到这些变量吗?

答案 1 :(得分:0)

这些变量只是简单地存在,就像约旦所指出的那样,它们只有在方法正在积极执行时才存在,并在之后立即丢弃。

如果你想要检测这些,你需要在方法中进行。该方法本身不允许这种反射。