Sha256: 9fae563afe75bd377f10a2ded036d410c2b87c66e7755bd235182031cff06db3
Contents?: true
Size: 1.83 KB
Versions: 10
Compression:
Stored size: 1.83 KB
Contents
module PatronusFati
DisconnectError = Class.new(StandardError)
class Connection
attr_reader :port, :read_queue, :server, :write_queue
def initialize(server, port)
@server = server
@port = port
self.read_queue = Queue.new
self.write_queue = Queue.new
end
def connect
establish_connection
start_read_thread
start_write_thread
end
def connected?
!socket.nil?
end
def disconnect
return unless connected?
Thread.kill(read_thread)
Thread.kill(write_thread)
socket.close unless socket.closed?
self.socket = nil
end
def read
read_queue.pop
end
def write(msg)
write_queue.push(msg)
end
protected
attr_accessor :read_thread, :socket, :write_thread
attr_writer :read_queue, :write_queue
def establish_connection
return if connected?
@socket = TCPSocket.new(server, port)
end
def start_read_thread
self.read_thread = Thread.new do
begin
while (line = socket.readline)
read_queue << line
end
rescue IOError, EOFError => e
raise DisconnectError
rescue => e
$stderr.puts format('Error in read thread: %s %s', e.class.to_s, e.message)
$stderr.puts e.backtrace
ensure
socket.close if socket
end
end
end
def start_write_thread
self.write_thread = Thread.new do
begin
count = 0
while (msg = write_queue.pop)
socket.write("!%i %s\r\n" % [count, msg])
count += 1
end
rescue => e
$stderr.puts format('Error in write thread: %s %s', e.class.to_s, e.message)
$stderr.puts e.backtrace
ensure
socket.close if socket
end
end
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems