Initial commit

app/src/server.py

@@ -0,0 +1,80 @@
+import logging, asyncio, signal, functools, os
+#from logging.handlers import TimedRotatingFileHandler
+from logging import config
+from async_stream import AsyncStream
+from proxy import Proxy
+from config import Config
+from mqtt import Mqtt
+
+    
+async def handle_client(reader, writer):
+    '''Handles a new incoming connection and starts an async loop'''
+
+    addr = writer.get_extra_info('peername')
+    await Proxy(reader, writer, addr).server_loop(addr) 
+
+
+def handle_SIGTERM(loop):
+    '''Close all TCP connections and stop the event loop'''
+
+    logging.info('Shutdown due to SIGTERM')
+
+    #
+    # first, close all open TCP connections
+    #
+    for stream in AsyncStream:
+        stream.close()
+
+    #
+    # at last, we stop the loop
+    #
+    loop.stop()
+
+    logging.info('Shutdown complete')
+    
+
+    
+
+
+
+if __name__ == "__main__":
+    #
+    # Setup our daily, rotating logger
+    #
+    serv_name = os.getenv('SERVICE_NAME', 'proxy')
+
+    logging.config.fileConfig('logging.ini')
+    logging.info(f'Server "{serv_name}" will be started')
+    
+    # read config file
+    Config.read()    
+
+    loop = asyncio.get_event_loop()
+
+    # call Mqtt singleton to establisch the connection to the mqtt broker
+    mqtt = Mqtt()
+    
+    #
+    # Register some UNIX Signal handler for a gracefully server shutdown on Docker restart and stop
+    #  
+    for signame in ('SIGINT','SIGTERM'):
+        loop.add_signal_handler(getattr(signal, signame), functools.partial(handle_SIGTERM, loop))
+
+    #
+    # Create a task for our listening server. This must be a task! If we call start_server directly out
+    # of our main task, the eventloop will be blocked and we can't receive and handle the UNIX signals!
+    #   
+    loop.create_task(asyncio.start_server(handle_client, '0.0.0.0', 5005))
+       
+    try:
+        loop.run_forever()
+    except KeyboardInterrupt:
+        pass
+    finally:
+        logging.info ('Close MQTT Task')        
+        loop.run_until_complete(mqtt.close())
+        mqtt = None   # release the last reference to the singleton
+        logging.info ('Close event loop')        
+        loop.close()
+        logging.info (f'Finally, exit Server "{serv_name}"')
+