X-Git-Url: https://plomlompom.com/repos/foo.html?a=blobdiff_plain;ds=sidebyside;f=server_%2Fio.py;h=c8f6114514b460ed0ccf3101ce09dd7185f02649;hb=131e3c6e9a1808cef19a9085f060fe845a7b123d;hp=501399f7dda4ea7920c3302ae26d254829f6f47a;hpb=52463b31fbb7f121a52f72e863a779560a6de531;p=plomrogue2-experiments
diff --git a/server_/io.py b/server_/io.py
index 501399f..c8f6114 100644
--- a/server_/io.py
+++ b/server_/io.py
@@ -45,45 +45,39 @@ class IO_Handler(socketserver.BaseRequestHandler):
instructions.
"""
- import plom_socket_io
- def caught_send(socket, message):
- """Send message by socket, catch broken socket connection error."""
- try:
- plom_socket_io.send(socket, message)
- except plom_socket_io.BrokenSocketConnection:
- pass
-
- def send_queue_messages(socket, queue_in, thread_alive):
+ def send_queue_messages(plom_socket, queue_in, thread_alive):
"""Send messages via socket from queue_in while thread_alive[0]."""
while thread_alive[0]:
try:
msg = queue_in.get(timeout=1)
except queue.Empty:
continue
- caught_send(socket, msg)
+ plom_socket.send(msg, True)
import uuid
+ import plom_socket
+ plom_socket = plom_socket.PlomSocket(self.request)
print('CONNECTION FROM:', str(self.client_address))
connection_id = uuid.uuid4()
queue_in = queue.Queue()
self.server.queue_out.put(('ADD_QUEUE', connection_id, queue_in))
thread_alive = [True]
t = threading.Thread(target=send_queue_messages,
- args=(self.request, queue_in, thread_alive))
+ args=(plom_socket, queue_in, thread_alive))
t.start()
- for message in plom_socket_io.recv(self.request):
+ for message in plom_socket.recv():
if message is None:
- caught_send(self.request, 'BAD MESSAGE')
+ plom_socket.send('BAD MESSAGE', True)
elif 'QUIT' == message:
- caught_send(self.request, 'BYE')
+ plom_socket.send('BYE', True)
break
else:
self.server.queue_out.put(('COMMAND', connection_id, message))
self.server.queue_out.put(('KILL_QUEUE', connection_id))
thread_alive[0] = False
print('CONNECTION CLOSED FROM:', str(self.client_address))
- self.request.close()
+ plom_socket.socket.close()
class GameIO():