Multiple Goliath WebSocket endpoints

gistfile1.rb

require "rubygems"
require "bundler/setup"
require "goliath"
require "goliath/websocket"

class User < Goliath::WebSocket

  def on_open(env)
    env.logger.info("WS OPEN")
  end

  def on_message(env, msg)
    env.logger.info("WS MESSAGE: #{msg}")
  end

  def on_close(env)
    # this never get's called
    env.logger.info("WS CLOSED from User")
  end

  def on_error(env, error)
    env.logger.error error
  end

  def response(env)
    super(env)
  end
end

class Server < Goliath::WebSocket
  def response(env)
    case env['PATH_INFO']
    when '/user'
      User.new.response(env)
    when '/admin'
      Admin.new.reponse(env)
    end
  end

  def on_close(env)
    # This get's triggered instead of User.on_close
    env.logger.info "WS CLOSED from Server"
  end
end