[ruby/singleton] Simplify the implementation
(https://github.com/ruby/singleton/pull/7) Remove `__init__` and move logic to `included`.
This commit is contained in:
parent
4e26ae3cb9
commit
bebd05fb51
@ -112,7 +112,7 @@ module Singleton
|
||||
module SingletonClassMethods # :nodoc:
|
||||
|
||||
def clone # :nodoc:
|
||||
Singleton.__init__(super)
|
||||
super.include(Singleton)
|
||||
end
|
||||
|
||||
# By default calls instance(). Override to retain singleton state.
|
||||
@ -121,31 +121,18 @@ module Singleton
|
||||
end
|
||||
|
||||
def instance # :nodoc:
|
||||
return @singleton__instance__ if @singleton__instance__
|
||||
@singleton__mutex__.synchronize {
|
||||
return @singleton__instance__ if @singleton__instance__
|
||||
@singleton__instance__ = new()
|
||||
}
|
||||
@singleton__instance__
|
||||
@singleton__instance__ || @singleton__mutex__.synchronize { @singleton__instance__ ||= new }
|
||||
end
|
||||
|
||||
private
|
||||
|
||||
def inherited(sub_klass)
|
||||
super
|
||||
Singleton.__init__(sub_klass)
|
||||
sub_klass.include(Singleton)
|
||||
end
|
||||
end
|
||||
|
||||
class << Singleton # :nodoc:
|
||||
def __init__(klass) # :nodoc:
|
||||
klass.instance_eval {
|
||||
@singleton__instance__ = nil
|
||||
@singleton__mutex__ = Thread::Mutex.new
|
||||
}
|
||||
klass
|
||||
end
|
||||
|
||||
private
|
||||
|
||||
# extending an object with Singleton is a bad idea
|
||||
@ -156,14 +143,19 @@ module Singleton
|
||||
unless mod.instance_of?(Class)
|
||||
raise TypeError, "Inclusion of the OO-Singleton module in module #{mod}"
|
||||
end
|
||||
|
||||
super
|
||||
end
|
||||
|
||||
def included(klass)
|
||||
super
|
||||
|
||||
klass.private_class_method :new, :allocate
|
||||
klass.extend SingletonClassMethods
|
||||
Singleton.__init__(klass)
|
||||
klass.instance_eval {
|
||||
@singleton__instance__ = nil
|
||||
@singleton__mutex__ = Thread::Mutex.new
|
||||
}
|
||||
end
|
||||
end
|
||||
|
||||
|
@ -94,6 +94,13 @@ class TestSingleton < Test::Unit::TestCase
|
||||
assert_same a, b
|
||||
end
|
||||
|
||||
def test_inheritance_creates_separate_singleton
|
||||
a = SingletonTest.instance
|
||||
b = Class.new(SingletonTest).instance
|
||||
|
||||
assert_not_same a, b
|
||||
end
|
||||
|
||||
def test_class_level_cloning_preserves_singleton_behavior
|
||||
klass = SingletonTest.clone
|
||||
|
||||
@ -101,4 +108,8 @@ class TestSingleton < Test::Unit::TestCase
|
||||
b = klass.instance
|
||||
assert_same a, b
|
||||
end
|
||||
|
||||
def test_class_level_cloning_creates_separate_singleton
|
||||
assert_not_same SingletonTest.instance, SingletonTest.clone.instance
|
||||
end
|
||||
end
|
||||
|
Loading…
x
Reference in New Issue
Block a user