提交 5b3d4f07 编写于 作者: A Aaron Patterson

switch WatchStack to use composition, tighten up API

上级 88daf082
......@@ -71,14 +71,20 @@ module Dependencies #:nodoc:
#
# This is handled by walking back up the watch stack and adding the constants
# found by child.rb to the list of original constants in parent.rb
class WatchStack < Hash
class WatchStack
include Enumerable
# @watching is a stack of lists of constants being watched. For instance,
# if parent.rb is autoloaded, the stack will look like [[Object]]. If parent.rb
# then requires namespace/child.rb, the stack will look like [[Object], [Namespace]].
def initialize
@watching = []
super { |h,k| h[k] = [] }
@stack = Hash.new { |h,k| h[k] = [] }
end
def each(&block)
@stack.each(&block)
end
# return a list of new constants found since the last call to watch_namespaces
......@@ -89,7 +95,7 @@ def new_constants
@watching.last.each do |namespace|
# Retrieve the constants that were present under the namespace when watch_namespaces
# was originally called
original_constants = self[namespace].last
original_constants = @stack[namespace].last
mod = Inflector.constantize(namespace) if Dependencies.qualified_const_defined?(namespace)
next unless mod.is_a?(Module)
......@@ -102,7 +108,7 @@ def new_constants
# element of self[Object] will be an Array of the constants that were present
# before parent.rb was required. The second element will be an Array of the
# constants that were present before child.rb was required.
self[namespace].each do |namespace_constants|
@stack[namespace].each do |namespace_constants|
namespace_constants.concat(new_constants)
end
......@@ -126,13 +132,14 @@ def watch_namespaces(namespaces)
Inflector.constantize(module_name).local_constant_names : []
watching << module_name
self[module_name] << original_constants
@stack[module_name] << original_constants
end
@watching << watching
end
private
def pop_modules(modules)
modules.each { |mod| self[mod].pop }
modules.each { |mod| @stack[mod].pop }
end
end
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册