Esempio n. 1
0
    def _internal_setup (self):
        self.signal = None
        signal.signal(signal.SIGTERM, self.sig_receiver)
        signal.signal(signal.SIGINT,  self.sig_receiver)
        signal.signal(signal.SIGQUIT, self.sig_receiver)
        signal.signal(signal.SIGHUP,  self.sig_receiver)
        signal.signal(signal.SIGCHLD, self.sig_receiver)
        signal.signal(signal.SIGUSR2, self.sig_receiver)

        self.reactor = get_reactor()
        self.reactor.call_later(self.housekeeping_period, self.housekeeping)
Esempio n. 2
0
def schedule_ewma_decay ():
    get_reactor().call_later(EWMA_DECAY_INTERVAL, ewma_decay)
Esempio n. 3
0
def __test ():

    from squib.core.async import get_reactor, LineOrientedProtocolMixin, TimeoutMixin

    class ChatProtocol (TCPReactable, LineOrientedProtocolMixin, TimeoutMixin):

        channels = dict()
        idletime = 10

        def __init__ (self, address, sock, reactor=None):
            super(ChatProtocol, self).__init__(address, sock, reactor)
            ChatProtocol.channels[self] = 1
            self.nick = None
            self.write_data('nickname: ')
            self.set_timeout(self.idletime)

        def on_closed (self):
            del ChatProtocol.channels[self]

        def on_timeout (self):
            self.write_line("Connection timed out. Goodbye.")
            self.handle_talk("[quit - timed out]")
            self.close_when_done()

        def on_message_received (self, line):
            self.reset_timeout()
            if self.nick is None:
                try:
                    self.nick = line.split()[0]
                except IndexError:
                    self.nick = None
                if not self.nick:
                    self.write_line("Huh?")
                    self.write_data('nickname: ')
                else:
                    # Greet
                    self.write_line("Hello, %s" % self.nick)
                    self.handle_talk("[joined]")
                    self.cmd_callers(None)
            else:
                if not line: pass
                elif line[0] != '/':
                    self.handle_talk(line)
                else:
                    self.handle_command(line)

        def handle_talk (self, line):
            for channel in ChatProtocol.channels.keys():
                if channel is not self:
                    channel.write_line("%s: %s" % (self.nick, line))

        def handle_command (self, line):
            command = line.split()
            name = 'cmd_%s' % command[0][1:]
            if hasattr(self, name):
                method = getattr(self, name)
                if callable(method):
                    method(command[1:])
                    return
            self.write_line('unknown command: %s' % command[0])

        def cmd_quit (self, args):
            if args:
                self.handle_talk('[quit] (%s)' % ' '.join(args))
            else:
                self.handle_talk('[quit]')
            self.write_line('goodbye.')
            self.close_when_done()

        cmd_q = cmd_quit

        def cmd_callers (self, args):
            num_channels = len(ChatProtocol.channels)
            if num_channels == 1:
                self.write_line("[You're the only caller]")
            else:
                self.write_line("[There are %d callers]" % (num_channels))
                nicks = [ x.nick or '<unknown>' for x in ChatProtocol.channels.keys() ]
                self.write_data(' ' + '\r\n '.join(nicks) + '\r\n')

    class ChatServer (TCPListener):
        def on_accept (self, sock, addr):
            ChatProtocol(addr, sock)

    ChatServer(address=('', 8518)).activate()
    get_reactor().start()
Esempio n. 4
0
 def __init__ (self, config, metrics_recorder):
     super(SelfStatistics, self).__init__()
     self.config = config
     self.metrics_recorder = metrics_recorder
     self.reactor = get_reactor()
     self.setup()