51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
|
from owrx.config import PropertyManager
|
||
|
|
from owrx.metrics import Metrics, DirectMetric
|
||
|
|
import threading
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class TooManyClientsException(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class ClientRegistry(object):
|
||
|
|
sharedInstance = None
|
||
|
|
creationLock = threading.Lock()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def getSharedInstance():
|
||
|
|
with ClientRegistry.creationLock:
|
||
|
|
if ClientRegistry.sharedInstance is None:
|
||
|
|
ClientRegistry.sharedInstance = ClientRegistry()
|
||
|
|
return ClientRegistry.sharedInstance
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.clients = []
|
||
|
|
Metrics.getSharedInstance().addMetric("openwebrx.users", DirectMetric(self.clientCount))
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
def broadcast(self):
|
||
|
|
n = self.clientCount()
|
||
|
|
for c in self.clients:
|
||
|
|
c.write_clients(n)
|
||
|
|
|
||
|
|
def addClient(self, client):
|
||
|
|
pm = PropertyManager.getSharedInstance()
|
||
|
|
if len(self.clients) >= pm["max_clients"]:
|
||
|
|
raise TooManyClientsException()
|
||
|
|
self.clients.append(client)
|
||
|
|
self.broadcast()
|
||
|
|
|
||
|
|
def clientCount(self):
|
||
|
|
return len(self.clients)
|
||
|
|
|
||
|
|
def removeClient(self, client):
|
||
|
|
try:
|
||
|
|
self.clients.remove(client)
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
self.broadcast()
|