require 'rubygems'
require 'bundler/setup'
require 'reel'
require 'celluloid/autostart'
class TimeServer
include Celluloid
include Celluloid::Notifications
def initialize
async.run
end
def run
now = Time.now.to_f
sleep now.ceil - now + 0.001
every(1) { publish 'time_change', Time.now }
end
end
class TimeClient
include Celluloid
include Celluloid::Notifications
include Celluloid::Logger
def initialize(websocket)
info "Streaming time changes to client"
@socket = websocket
subscribe('time_change', :notify_time_change)
end
def notify_time_change(topic, new_time)
@socket << new_time.inspect
rescue Reel::SocketError
info "Time client disconnected"
terminate
end
end
class WebServer < Reel::Server::HTTP
include Celluloid::Logger
def initialize(host = "127.0.0.1", port = 1234)
info "Time server example starting on #{host}:#{port}"
super(host, port, &method(:on_connection))
end
def on_connection(connection)
while request = connection.request
if request.websocket?
info "Received a WebSocket connection"
# We're going to hand off this connection to another actor (TimeClient)
# However, initially Reel::Connections are "attached" to the
# Reel::Server::HTTP actor, meaning that the server manages the connection
# lifecycle (e.g. error handling) for us.
#
# If we want to hand this connection off to another actor, we first
# need to detach it from the Reel::Server (in this case, Reel::Server::HTTP)
connection.detach
route_websocket request.websocket
return
else
route_request connection, request
end
end
end
def route_request(connection, request)
if request.url == "/"
return render_index(connection)
end
info "404 Not Found: #{request.path}"
connection.respond :not_found, "Not found"
end
def route_websocket(socket)
if socket.url == "/timeinfo"
TimeClient.new(socket)
else
info "Received invalid WebSocket request for: #{socket.url}"
socket.close
end
end
def render_index(connection)
info "200 OK: /"
connection.respond :ok, <<-HTML
Reel WebSockets time server example
Time Server Example
The time is now: ...
HTML
end
end
TimeServer.supervise_as :time_server
WebServer.supervise_as :reel
sleep