class BaseConnector(BrokerMessageReceiver):
""" A base class for both channels and outgoing connectors.
"""
def __init__(self, repo_location, def_id):
self.repo_location = repo_location
self.def_id = def_id
self.odb = None
self.odb_config = None
self.sql_pool_store = None
def _close(self):
""" Close the process, don't forget about the ODB connection if it exists.
"""
if self.odb:
self.odb.close()
p = psutil.Process(os.getpid())
p.terminate()
def _setup_odb(self):
# First let's see if the server we're running on top of exists in the ODB.
self.server = self.odb.fetch_server(self.odb_config)
if not self.server:
raise Exception('Server does not exist in the ODB')
def _init(self):
""" Initializes all the basic run-time data structures and connects
to the Zato broker.
"""
fs_server_config = get_config(self.repo_location, 'server.conf')
app_context = get_app_context(fs_server_config)
crypto_manager = get_crypto_manager(self.repo_location, app_context, fs_server_config)
config_odb = fs_server_config.odb
self.odb = app_context.get_object('odb_manager')
self.odb.crypto_manager = crypto_manager
self.odb.token = fs_server_config.main.token
# Key-value DB
self.kvdb = KVDB()
self.kvdb.config = fs_server_config.kvdb
self.kvdb.decrypt_func = self.odb.crypto_manager.decrypt
self.kvdb.init()
# Broker client
self.broker_client = BrokerClient(self.kvdb, self.broker_client_id, self.broker_callbacks)
self.broker_client.start()
# ODB
self.odb_config = Bunch()
self.odb_config.db_name = config_odb.db_name
self.odb_config.is_active = True
self.odb_config.engine = config_odb.engine
self.odb_config.extra = config_odb.extra
self.odb_config.host = config_odb.host
self.odb_config.password = self.odb.crypto_manager.decrypt(config_odb.password)
self.odb_config.pool_size = config_odb.pool_size
self.odb_config.username = config_odb.username
self.odb_config.is_odb = True
self.sql_pool_store = app_context.get_object('sql_pool_store')
self.sql_pool_store[ZATO_ODB_POOL_NAME] = self.odb_config
self.odb.pool = self.sql_pool_store[ZATO_ODB_POOL_NAME].pool
self._setup_odb()
class BaseConnector(BrokerMessageReceiver):
""" A base class for both channels and outgoing connectors.
"""
def __init__(self, repo_location, def_id):
self.repo_location = repo_location
self.def_id = def_id
self.odb = None
self.odb_config = None
self.sql_pool_store = None
def _close(self):
""" Close the process, don't forget about the ODB connection if it exists.
"""
if self.odb:
self.odb.close()
p = psutil.Process(os.getpid())
p.terminate()
def _setup_odb(self):
# First let's see if the server we're running on top of exists in the ODB.
self.server = self.odb.fetch_server(self.odb_config)
if not self.server:
raise Exception('Server does not exist in the ODB')
def _init(self):
""" Initializes all the basic run-time data structures and connects
to the Zato broker.
"""
fs_server_config = get_config(self.repo_location, 'server.conf')
app_context = get_app_context(fs_server_config)
crypto_manager = get_crypto_manager(self.repo_location, app_context,
fs_server_config)
config_odb = fs_server_config.odb
self.odb = app_context.get_object('odb_manager')
self.odb.crypto_manager = crypto_manager
self.odb.token = fs_server_config.main.token
# Key-value DB
self.kvdb = KVDB()
self.kvdb.config = fs_server_config.kvdb
self.kvdb.decrypt_func = self.odb.crypto_manager.decrypt
self.kvdb.init()
# Broker client
self.broker_client = BrokerClient(self.kvdb, self.broker_client_id,
self.broker_callbacks)
self.broker_client.start()
# ODB
self.odb_config = Bunch()
self.odb_config.db_name = config_odb.db_name
self.odb_config.is_active = True
self.odb_config.engine = config_odb.engine
self.odb_config.extra = config_odb.extra
self.odb_config.host = config_odb.host
self.odb_config.password = self.odb.crypto_manager.decrypt(
config_odb.password)
self.odb_config.pool_size = config_odb.pool_size
self.odb_config.username = config_odb.username
self.odb_config.is_odb = True
self.sql_pool_store = app_context.get_object('sql_pool_store')
self.sql_pool_store[ZATO_ODB_POOL_NAME] = self.odb_config
self.odb.pool = self.sql_pool_store[ZATO_ODB_POOL_NAME].pool
self._setup_odb()
