@@ -27,6 +27,15 @@ module ProxyServer
include LineBufferedConnection, Deferrable
extend Synchronizable
+ @@servers = (1..$options[:count]).inject([]) do |servers, i|
+ servers << Server.new($options[:host], $options[:port] + i)
+ end
+ @@balancer = $options[:balancer].new(@@servers)
+
+ def self.forward(data)
+ @@balancer.forward(data)
+ end
+
def receive_line(line)
defer do
$stats.transaction do
@@ -37,27 +46,6 @@ module ProxyServer
end
end
end
-
- def self.forward(data)
- balancer.forward(data)
- end
-
- private
- def self.servers
- synchronize(:servers) do
- @servers ||= begin
- (1..$options[:count]).inject([]) do |servers, i|
- servers << Server.new($options[:host], $options[:port] + i)
- end
- end
- end
- end
-
- def self.balancer
- synchronize(:balancer) do
- @balancer ||= $options[:balancer].new(servers)
- end
- end
end
EM.run do
proxy.rb
@@ -13,9 +13,7 @@ class LeastConnections < Balancer
def next_server
server = nil
synchronize(:next_server) do
- server = servers.min do |s1, s2|
- s1.connections <=> s2.connections
- end
+ server = servers.min
server.reserve
end
proxy/balancers/least_connections.rb
@@ -5,18 +5,18 @@ class Server
def initialize(host, port)
@host, @port = host, port
- synchronize(:connections) { @connections = 0 }
+ @connections = 0
end
def reserve
synchronize(:connections) do
- self.connections += 1
+ @connections += 1
end
end
def release
synchronize(:connections) do
- self.connections -= 1
+ @connections -= 1
end
end
@@ -26,17 +26,13 @@ class Server
socket.readline
end
end
+
+ def <=>(other)
+ connections <=> other.connections
+ end
- private
- # Whilst I'm not normally a fan of closing apis or gratuitously spreading
- # private around, this code is subject to thread safety concerns, so
- # assignment should be ensure local, or the value should not be accessible,
- # instead it should have a thread safe incr! and decr!.
+ protected
def connections
@connections
end
-
- def connections=(connections)
- @connections = connections
- end
end
\ No newline at end of file
proxy/server.rb
@@ -1,19 +1,16 @@
module Synchronizable
@@mutex = Mutex.new
-
- def mutex
- # Yup, this is nasty, but it's the only way without patching
- # initialize/new and having included and extended callbacks.
+ @@mutexes = Hash.new do |h,k|
+ h[k] = Mutex.new
+ end
+
+ def mutex(mutex)
@@mutex.synchronize do
- @mutex ||= Hash.new do |h,k|
- @@mutex.synchronize do
- h[k] = Mutex.new
- end
- end
+ @@mutexes[mutex]
end
end
def synchronize(name, &block)
- mutex[name].synchronize(&block)
+ mutex(name).synchronize(&block)
end
end
\ No newline at end of file
util/synchronizable.rb
e946183a140ba251d30ab734362910231e0bf32e
Nick Kallen
nkallen@yahoo.com
http://github.com/nkallen/gogaruco/commit/6abbb13f2b6491d1993909e012ed6e05fd3845e7
6abbb13f2b6491d1993909e012ed6e05fd3845e7
2009-04-20T20:02:53-07:00
2009-04-20T20:02:53-07:00
fixes some race conditions, keeps the code terse
a59f445403d0f99a6fcf7d1da608acc93e1d266a
Nick Kallen
nkallen@yahoo.com