lib/faye/engines/memory.rb in faye-0.8.8 vs lib/faye/engines/memory.rb in faye-0.8.9
- old
+ new
@@ -1,112 +1,112 @@
module Faye
module Engine
-
+
class Memory
include Timeouts
-
+
def self.create(server, options)
new(server, options)
end
-
+
def initialize(server, options)
@server = server
@options = options
@namespace = Namespace.new
@clients = {}
@channels = {}
@messages = {}
end
-
+
def create_client(&callback)
client_id = @namespace.generate
@server.debug 'Created new client ?', client_id
ping(client_id)
@server.trigger(:handshake, client_id)
callback.call(client_id)
end
-
+
def destroy_client(client_id, &callback)
return unless @namespace.exists?(client_id)
-
+
if @clients.has_key?(client_id)
@clients[client_id].each { |channel| unsubscribe(client_id, channel) }
end
-
+
remove_timeout(client_id)
@namespace.release(client_id)
@messages.delete(client_id)
@server.debug 'Destroyed client ?', client_id
@server.trigger(:disconnect, client_id)
callback.call if callback
end
-
+
def client_exists(client_id, &callback)
callback.call(@namespace.exists?(client_id))
end
-
+
def ping(client_id)
timeout = @server.timeout
return unless Numeric === timeout
@server.debug 'Ping ?, ?', client_id, timeout
remove_timeout(client_id)
add_timeout(client_id, 2 * timeout) { destroy_client(client_id) }
end
-
+
def subscribe(client_id, channel, &callback)
@clients[client_id] ||= Set.new
should_trigger = @clients[client_id].add?(channel)
-
+
@channels[channel] ||= Set.new
@channels[channel].add(client_id)
-
+
@server.debug 'Subscribed client ? to channel ?', client_id, channel
@server.trigger(:subscribe, client_id, channel) if should_trigger
callback.call(true) if callback
end
-
+
def unsubscribe(client_id, channel, &callback)
if @clients.has_key?(client_id)
should_trigger = @clients[client_id].delete?(channel)
@clients.delete(client_id) if @clients[client_id].empty?
end
-
+
if @channels.has_key?(channel)
@channels[channel].delete(client_id)
@channels.delete(channel) if @channels[channel].empty?
end
-
+
@server.debug 'Unsubscribed client ? from channel ?', client_id, channel
@server.trigger(:unsubscribe, client_id, channel) if should_trigger
callback.call(true) if callback
end
-
+
def publish(message, channels)
@server.debug 'Publishing message ?', message
-
+
clients = Set.new
-
+
channels.each do |channel|
next unless subs = @channels[channel]
subs.each(&clients.method(:add))
end
-
+
clients.each do |client_id|
@server.debug 'Queueing for client ?: ?', client_id, message
@messages[client_id] ||= []
@messages[client_id] << Faye.copy_object(message)
empty_queue(client_id)
end
@server.trigger(:publish, message['clientId'], message['channel'], message['data'])
end
-
+
def empty_queue(client_id)
return unless @server.has_connection?(client_id)
@server.deliver(client_id, @messages[client_id])
@messages.delete(client_id)
end
end
-
+
end
end