mirror of
https://github.com/BOSWatch/BW3-Core.git
synced 2026-04-05 06:15:31 +00:00
rework server class (added python queue support)
This commit is contained in:
parent
24eeef6ebb
commit
5c6039ef71
6 changed files with 69 additions and 76 deletions
|
|
@ -21,17 +21,13 @@ import time
|
|||
|
||||
logging.debug("- %s loaded", __name__)
|
||||
|
||||
# module wide global list for received data sets
|
||||
_dataPackets = []
|
||||
_lockDataPackets = threading.Lock()
|
||||
|
||||
# module wide global list for all currently connected clients
|
||||
_clients = {} # _clients[ThreadName] = {"address", "timestamp"}
|
||||
_lockClients = threading.Lock()
|
||||
|
||||
|
||||
class TCPHandler(socketserver.BaseRequestHandler):
|
||||
"""!RequestHandler class for our TCPServer class."""
|
||||
class ThreadedTCPRequestHandler(socketserver.ThreadingMixIn, socketserver.BaseRequestHandler):
|
||||
"""!ThreadedTCPRequestHandler class for our TCPServer class."""
|
||||
|
||||
def handle(self):
|
||||
"""!Handles the request from an single client in a own thread
|
||||
|
|
@ -51,10 +47,8 @@ class TCPHandler(socketserver.BaseRequestHandler):
|
|||
if data != "":
|
||||
logging.debug("%s recv: %s", req_name, data)
|
||||
|
||||
# add a new entry at first position (index 0) with client IP
|
||||
# and the decoded data dict as an string in utf-8 and an timestamp
|
||||
with _lockDataPackets:
|
||||
_dataPackets.insert(0, (self.client_address[0], data, time.time())) # time() to calc time in queue
|
||||
# add a new entry and the decoded data dict as an string in utf-8 and an timestamp
|
||||
self.server.alarmQueue.put_nowait((self.client_address[0], data, time.time()))
|
||||
logging.debug("Add data to queue")
|
||||
|
||||
logging.debug("%s send: [ack]", req_name)
|
||||
|
|
@ -71,14 +65,24 @@ class TCPHandler(socketserver.BaseRequestHandler):
|
|||
logging.info("Client disconnected: %s", self.client_address[0])
|
||||
|
||||
|
||||
class TCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
||||
class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
||||
"""!ThreadedTCPServer class for our TCPServer class."""
|
||||
pass
|
||||
|
||||
|
||||
class TCPServer:
|
||||
"""!TCP server class"""
|
||||
|
||||
def __init__(self, timeout=3):
|
||||
"""!Create a new instance"""
|
||||
def __init__(self, alarmQueue, timeout=3):
|
||||
"""!Create a new instance
|
||||
|
||||
@param alarmQueue: python queue instance
|
||||
@param timeout: server timeout in sec (3)
|
||||
"""
|
||||
self._server = None
|
||||
self._server_thread = None
|
||||
self._timeout = timeout
|
||||
self._alarmQueue = alarmQueue
|
||||
|
||||
def start(self, port=8080):
|
||||
"""!Start a threaded TCP socket server
|
||||
|
|
@ -91,10 +95,9 @@ class TCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
|||
|
||||
@return True or False"""
|
||||
try:
|
||||
self._server = socketserver.ThreadingTCPServer(("", port), TCPHandler)
|
||||
self._server = ThreadedTCPServer(("", port), ThreadedTCPRequestHandler)
|
||||
self._server.timeout = self._timeout
|
||||
|
||||
self.flushQueue()
|
||||
self._server.alarmQueue = self._alarmQueue
|
||||
|
||||
self._server_thread = threading.Thread(target=self._server.serve_forever)
|
||||
self._server_thread.name = "Thread-BWServer"
|
||||
|
|
@ -115,7 +118,9 @@ class TCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
|||
try:
|
||||
self._server.shutdown()
|
||||
self._server_thread.join()
|
||||
self._server_thread = None
|
||||
self._server.socket.close()
|
||||
self._server = None
|
||||
logging.debug("TCPServer stopped")
|
||||
return True
|
||||
except AttributeError:
|
||||
|
|
@ -135,33 +140,9 @@ class TCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
|||
|
||||
@staticmethod
|
||||
def getClientsConnected():
|
||||
# todo insert comment
|
||||
"""!List of currently connected Clients
|
||||
_clients[ThreadName] = {"address", "timestamp"}
|
||||
|
||||
@return List of connected clients"""
|
||||
# todo return full list or write a print/debug method?
|
||||
return _clients
|
||||
|
||||
@staticmethod
|
||||
def getDataFromQueue():
|
||||
"""!Function to get the data packages from server
|
||||
must be polled by main program
|
||||
|
||||
@return Next data packet.py from intern queue"""
|
||||
if _dataPackets:
|
||||
with _lockDataPackets:
|
||||
message = _dataPackets.pop()
|
||||
logging.debug("Get data from queue")
|
||||
return message
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def countPacketsInQueue():
|
||||
"""!Get packets waiting in queue
|
||||
|
||||
@return Packets in queue"""
|
||||
return len(_dataPackets) # no lock needed - only reading
|
||||
|
||||
@staticmethod
|
||||
def flushQueue():
|
||||
"""!To flush all existing data in queue"""
|
||||
logging.debug("Flush data queue")
|
||||
with _lockDataPackets:
|
||||
_dataPackets.clear()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue