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)