57 lines
1.4 KiB
Ruby
57 lines
1.4 KiB
Ruby
|
module EventMachine
|
||
|
|
module WebSocket
|
||
|
|
class << self
|
||
|
|
attr_accessor :max_frame_size
|
||
|
|
attr_accessor :close_timeout
|
||
|
|
end
|
||
|
|
@max_frame_size = 10 * 1024 * 1024 # 10MB
|
||
|
|
# Connections are given 60s to close after being sent a close handshake
|
||
|
|
@close_timeout = 60
|
||
|
|
|
||
|
|
# All errors raised by em-websocket should descend from this class
|
||
|
|
class WebSocketError < RuntimeError; end
|
||
|
|
|
||
|
|
# Used for errors that occur during WebSocket handshake
|
||
|
|
class HandshakeError < WebSocketError; end
|
||
|
|
|
||
|
|
# Used for errors which should cause the connection to close.
|
||
|
|
# See RFC6455 §7.4.1 for a full description of meanings
|
||
|
|
class WSProtocolError < WebSocketError
|
||
|
|
def code; 1002; end
|
||
|
|
end
|
||
|
|
|
||
|
|
class InvalidDataError < WSProtocolError
|
||
|
|
def code; 1007; end
|
||
|
|
end
|
||
|
|
|
||
|
|
# 1009: Message too big to process
|
||
|
|
class WSMessageTooBigError < WSProtocolError
|
||
|
|
def code; 1009; end
|
||
|
|
end
|
||
|
|
|
||
|
|
# Start WebSocket server, including starting eventmachine run loop
|
||
|
|
def self.start(options, &blk)
|
||
|
|
EM.epoll
|
||
|
|
EM.run {
|
||
|
|
trap("TERM") { stop }
|
||
|
|
trap("INT") { stop }
|
||
|
|
|
||
|
|
run(options, &blk)
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
# Start WebSocket server inside eventmachine run loop
|
||
|
|
def self.run(options)
|
||
|
|
host, port = options.values_at(:host, :port)
|
||
|
|
EM.start_server(host, port, Connection, options) do |c|
|
||
|
|
yield c
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.stop
|
||
|
|
puts "Terminating WebSocket Server"
|
||
|
|
EM.stop
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|