MiNode/src/listener.py

# -*- coding: utf-8 -*-
import logging
import socket
import threading

from connection import Connection
import shared


class Listener(threading.Thread):
    def __init__(self, host, port, family=socket.AF_INET):
        super().__init__(name='Listener')
        self.host = host
        self.port = port
        self.family = family
        self.s = socket.socket(self.family, socket.SOCK_STREAM)
        self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.s.bind((self.host, self.port))
        self.s.listen(1)
        self.s.settimeout(1)

    def run(self):
        while True:
            if shared.shutting_down:
                logging.debug('Shutting down Listener')
                break
            try:
                conn, addr = self.s.accept()
                logging.info('Incoming connection from: {}:{}'.format(addr[0], addr[1]))
                with shared.connections_lock:
                    if len(shared.connections) > shared.connection_limit:
                        conn.close()
                    else:
                        c = Connection(addr[0], addr[1], conn)
                        c.start()
                        shared.connections.add(c)
            except socket.timeout:
                pass