mirror of
https://github.com/craigerl/aprsd.git
synced 2025-06-24 21:15:18 -04:00
Hemna
a62e490353
This adds a layer between the client object and the actual client instance, so we can reset the actual client object instance upon failure of connection.
105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
import abc
|
|
import datetime
|
|
import logging
|
|
import threading
|
|
from typing import List
|
|
|
|
import wrapt
|
|
|
|
|
|
LOG = logging.getLogger("APRSD")
|
|
|
|
|
|
class APRSDThread(threading.Thread, metaclass=abc.ABCMeta):
|
|
"""Base class for all threads in APRSD."""
|
|
|
|
loop_interval = 1
|
|
|
|
def __init__(self, name):
|
|
super().__init__(name=name)
|
|
self.thread_stop = False
|
|
APRSDThreadList().add(self)
|
|
self._last_loop = datetime.datetime.now()
|
|
|
|
def _should_quit(self):
|
|
""" see if we have a quit message from the global queue."""
|
|
if self.thread_stop:
|
|
return True
|
|
|
|
def stop(self):
|
|
self.thread_stop = True
|
|
|
|
@abc.abstractmethod
|
|
def loop(self):
|
|
pass
|
|
|
|
def _cleanup(self):
|
|
"""Add code to subclass to do any cleanup"""
|
|
|
|
def __str__(self):
|
|
out = f"Thread <{self.__class__.__name__}({self.name}) Alive? {self.is_alive()}>"
|
|
return out
|
|
|
|
def loop_age(self):
|
|
"""How old is the last loop call?"""
|
|
return datetime.datetime.now() - self._last_loop
|
|
|
|
def run(self):
|
|
LOG.debug("Starting")
|
|
while not self._should_quit():
|
|
can_loop = self.loop()
|
|
self.loop_interval += 1
|
|
self._last_loop = datetime.datetime.now()
|
|
if not can_loop:
|
|
self.stop()
|
|
self._cleanup()
|
|
APRSDThreadList().remove(self)
|
|
LOG.debug("Exiting")
|
|
|
|
|
|
class APRSDThreadList:
|
|
"""Singleton class that keeps track of application wide threads."""
|
|
|
|
_instance = None
|
|
|
|
threads_list: List[APRSDThread] = []
|
|
lock = threading.Lock()
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
if cls._instance is None:
|
|
cls._instance = super().__new__(cls)
|
|
cls.threads_list = []
|
|
return cls._instance
|
|
|
|
@wrapt.synchronized(lock)
|
|
def add(self, thread_obj):
|
|
self.threads_list.append(thread_obj)
|
|
|
|
@wrapt.synchronized(lock)
|
|
def remove(self, thread_obj):
|
|
self.threads_list.remove(thread_obj)
|
|
|
|
@wrapt.synchronized(lock)
|
|
def stop_all(self):
|
|
"""Iterate over all threads and call stop on them."""
|
|
for th in self.threads_list:
|
|
LOG.info(f"Stopping Thread {th.name}")
|
|
if hasattr(th, "packet"):
|
|
LOG.info(F"{th.name} packet {th.packet}")
|
|
th.stop()
|
|
|
|
@wrapt.synchronized(lock)
|
|
def info(self):
|
|
"""Go through all the threads and collect info about each."""
|
|
info = {}
|
|
for thread in self.threads_list:
|
|
alive = thread.is_alive()
|
|
age = thread.loop_age()
|
|
key = thread.__class__.__name__
|
|
info[key] = {"alive": True if alive else False, "age": age, "name": thread.name}
|
|
return info
|
|
|
|
@wrapt.synchronized(lock)
|
|
def __len__(self):
|
|
return len(self.threads_list)
|