This commit is contained in:
@@ -0,0 +1,56 @@
|
||||
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
|
||||
Reference in New Issue
Block a user