MiNode/minode/listener.py

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

from . import shared
from .connection import Connection


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: %s:%i', *addr[:2])
                with shared.connections_lock:
                    if len(shared.connections) > shared.connection_limit:
                        conn.close()
                    else:
                        c = Connection(*addr[:2], conn, server=True)
                        c.start()
                        shared.connections.add(c)
            except socket.timeout:
                pass