def run(self):
database = SQLDB(self.sql_config.driver, self.sql_config.host, self.sql_config.port, self.sql_config.database,
self.sql_config.username, self.sql_config.password, self.sql_config.pool_size)
database.connect()
with database.session() as session:
Base.metadata.create_all(bind=session.get_bind())
ldap = None
if self.ldap_config is not None:
ldap = LDAP(self.ldap_config.host, self.ldap_config.domain, self.ldap_config.base_dn,
self.ldap_config.bind_username, self.ldap_config.bind_password)
hosts = []
for host in self.rabbitmq_config.hosts:
(ip, port) = host.split(":")
hosts.append((ip, int(port)))
self.rabbitmq = RabbitMQ(hosts, self.rabbitmq_config.username, self.rabbitmq_config.password,
self.rabbitmq_config.virtual_host)
self.rabbitmq.setup_database()
if 'driver' not in self.base_config.identity:
self.logger.error("Identity Config does not have a driver")
return False
identity_module = self.base_config.identity['driver'].split(".")
module = __import__(self.base_config.identity['driver'])
identity_module.pop(0)
for m in identity_module:
module = getattr(module, m)
identity = getattr(module, 'IdentityDriver')
identity = identity()
if not identity.validate_config(self.base_config.identity):
self.logger.error("Error validating identity config")
return False
identity.db_connections(database=database, rabbitmq=self.rabbitmq, ldap=ldap)
if 'driver' not in self.base_config.assignment:
self.logger.error("Assignment Config does not have a driver set")
return
assignment_modules = self.base_config.assignment['driver'].split(".")
module = __import__(self.base_config.assignment['driver'])
assignment_modules.pop(0)
for m in assignment_modules:
module = getattr(module, m)
assignment = getattr(module, 'AssignmentDriver')
assignment = assignment()
if not assignment.validate_config(self.base_config.assignment):
self.logger.error("Error validating assignment config")
return False
assignment.db_connections(database=database, rabbitmq=self.rabbitmq, ldap=ldap)
RegisterFrameworkSubscriber(self.rabbitmq).start()
TaskStatusSubscriber(self.rabbitmq, database).start()
TaskLaunchSubscriber(self.rabbitmq, database).start()
WorkerRegister(self.rabbitmq, database).start()
WorkerReload(self.rabbitmq, database).start()
WorkerGet(self.rabbitmq, database).start()
Validate(self.rabbitmq, assignment).start()
hqmanager.api.setup(self.rabbitmq, database, identity, assignment)
return True
