websocket_celluloid_server.rb

require 'celluloid/io'
require 'websocket/driver'

class Server
  include Celluloid::Logger
  include Celluloid::IO

  def initialize
    @server = TCPServer.new('127.0.0.1', 5009)
    async.run
  end

  def run
    loop { async.handle_connection @server.accept }
  end

  def handle_connection(socket)
    driver = WebSocket::Driver.server(socket)

    driver.on(:close) do |e|
      debug "code: #{e.code}\nreason: #{e.reason}"
      socket.close
    end

    driver.on(:message) { |e| info e.data }

    driver.on(:connect) do
      if WebSocket::Driver.websocket?(driver.env)
        driver.start
      else
        driver.close
      end
    end

    loop { driver.parse socket.readpartial(16384) }

  rescue => e
    crash "#{self.class} crashed!", e
    socket.close
  end
end