lib/registry.rb in regstry-1.0.11 vs lib/registry.rb in regstry-1.0.12
- old
+ new
@@ -75,13 +75,15 @@
end
module ClassMethods
attr_reader :registry_key
- def register(key, obj = self)
- key = key.to_sym
- obj.instance_variable_set(:@registry_key, key)
- registry[key] = obj
+ def register(*args)
+ obj = !args.last.is_a?(Symbol) ? args.pop : self
+ args.reverse.map(&:to_sym).each do |key|
+ obj.instance_variable_set(:@registry_key, key)
+ registry[key] = obj
+ end
end
def unregister(obj = self)
return unless obj.respond_to?(:registry_key) && obj.registry_key
registry.delete(obj.registry_key)