class TrafficApp(object):
def __init__(self, config, record_queue=None):
self.log = logging.getLogger(__name__)
self._conf = config
self._cs_db = ConnectedStateProcessor(DBConnectedState())
namespaces = NamespaceManager().get_all_namespaces()
record_queue = Queue(
RECORD_QUEUE_SIZE) if record_queue is None else record_queue
if self._conf.NAMESPACE_MODE:
if not namespaces:
self.log.warning(
"No namespace is found but NAMESPACE_MODE "
"is set to True on config file. Create namespace(s) "
"and run rediscover_namespaces")
self.namespace_mode = True
self._server_agent = AxonNameSpaceServerAgent()
self._client_agent = AxonNameSpaceClientAgent(record_queue)
else:
self.namespace_mode = False
self._server_agent = AxonRootNamespaceServerAgent()
self._client_agent = AxonRootNamespaceClientAgent(record_queue)
self._start_db_pool_manager(record_queue)
def _start_db_pool_manager(self, queue):
manager = DBPoolManager(queue)
thread = threading.Thread(target=manager.run)
thread.daemon = True
thread.start()
def add_server(self, protocol, port, endpoint, namespace=None):
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self.log.info("Add %s Server on port %s started" % (protocol, port))
self._server_agent.add_server(port, protocol, endpoint, namespace)
def delete_traffic_rules(self, endpoint=None):
# If endpoint is None , traffic rules for all endpoints will be
# deleted
endpoint_name = endpoint if endpoint else 'all'
self.log.info("Deleting traffic config for %s" % endpoint_name)
self._cs_db.delete_connected_state(endpoint)
def register_traffic(self, traffic_configs):
self.log.info("Register traffic called with config %s" %
traffic_configs)
for config in traffic_configs:
self._cs_db.create_or_update_connected_state(
config['endpoint'], config['servers'], config['clients'])
def unregister_traffic(self, traffic_configs):
self.log.info("Un-Register traffic called with config %s" %
traffic_configs)
for config in traffic_configs:
self._cs_db.delete_connected_state(config.get('endpoint'),
config.get('servers', []),
config.get('clients', []))
def get_traffic_rules(self, endpoint=None):
return self._cs_db.get_connected_state(endpoint)
def list_servers(self):
return self._server_agent.list_servers()
def get_server(self, protocol, port):
return self._server_agent.get_server(protocol, port)
def stop_servers(self, namespace=None):
self.log.info("=====Stop servers called=====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._server_agent.stop_servers(namespace)
def start_servers(self, namespace=None):
self.log.info("=====Start servers called=====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._server_agent.start_servers(namespace)
def stop_server(self, protocol, port, namespace=None, endpoint=None):
self.log.info("stop %s server called on port %s" % (protocol, port))
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._server_agent.stop_server(protocol, port, namespace, endpoint)
def stop_client(self, namespace=None, endpoint=None):
self.log.info("====stop client initiated====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._client_agent.stop_client(namespace, endpoint)
def stop_clients(self, namespace=None):
self.log.info("====stop clients initiated====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._client_agent.stop_clients(namespace=namespace)
def start_clients(self):
self.log.info("====start clients initiated====")
self._client_agent.start_clients()
def rediscover_namespaces(self):
self.log.info("====rediscover namespaces initiated====")
self._server_agent.rediscover_namespaces()
self._client_agent.rediscover_namespaces()
class TrafficApp(object):
def __init__(self, config):
self.log = logging.getLogger(__name__)
self._conf = config
self._cs_db = ConnectedStateProcessor(DBConnectedState())
namespaces = NamespaceManager().get_all_namespaces()
if namespaces and self._conf.NAMESPACE_MODE:
self.namespace_mode = True
self._server_agent = AxonNameSpaceServerAgent()
self._client_agent = AxonNameSpaceClientAgent()
else:
self.namespace_mode = False
self._server_agent = AxonRootNamespaceServerAgent()
self._client_agent = AxonRootNamespaceClientAgent()
def add_server(self, protocol, port, endpoint, namespace=None):
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self.log.info("Add %s Server on port %s started" % (protocol, port))
self._server_agent.add_server(port, protocol, endpoint, namespace)
def register_traffic(self, traffic_configs):
self.log.info("Register traffic called with config %s" %
traffic_configs)
for config in traffic_configs:
self._cs_db.create_or_update_connected_state(
config['endpoint'], config['servers'], config['clients'])
def unregister_traffic(self, traffic_configs):
self.log.info("Un-Register traffic called with config %s" %
traffic_configs)
for config in traffic_configs:
self._cs_db.delete_connected_state(config.get('endpoint'),
config.get('servers', []),
config.get('clients', []))
def get_traffic_config(self, endpoint=None):
return self._cs_db.get_connected_state(endpoint)
def list_servers(self):
return self._server_agent.list_servers()
def get_server(self, protocol, port):
return self._server_agent.get_server(protocol, port)
def stop_servers(self, namespace=None):
self.log.info("=====Stop servers called=====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._server_agent.stop_servers(namespace)
def start_servers(self, namespace=None):
self.log.info("=====Start servers called=====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._server_agent.start_servers(namespace)
def stop_server(self, protocol, port, namespace=None):
self.log.info("stop %s server called on port %s" % (protocol, port))
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._server_agent.stop_server(protocol, port, namespace)
def stop_client(self, namespace=None):
self.log.info("====stop client initiated====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._client_agent.stop_client(namespace)
def stop_clients(self, namespace=None):
self.log.info("====stop clients initiated====")
if not self.namespace_mode and namespace:
raise ValueError("namespace parameter must not be provided,"
"as the Axon is running in non namespace mode")
self._client_agent.stop_clients(namespace=namespace)
def start_clients(self):
self.log.info("====start clients initiated====")
self._client_agent.start_clients()