79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import logging
|
|
import threading
|
|
|
|
from django import db
|
|
|
|
from orchestra.utils.python import import_class
|
|
|
|
from . import settings
|
|
from .helpers import send_report
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def as_task(execute):
|
|
def wrapper(*args, **kwargs):
|
|
with db.transaction.commit_manually():
|
|
log = execute(*args, **kwargs)
|
|
db.transaction.commit()
|
|
if log.state != log.SUCCESS:
|
|
send_report(execute, args, log)
|
|
return log
|
|
return wrapper
|
|
|
|
|
|
def close_connection(execute):
|
|
""" Threads have their own connection pool, closing it when finishing """
|
|
# TODO rewrite as context manager
|
|
def wrapper(*args, **kwargs):
|
|
log = execute(*args, **kwargs)
|
|
db.connection.close()
|
|
# Using the wrapper function as threader messenger for the execute output
|
|
wrapper.log = log
|
|
return wrapper
|
|
|
|
|
|
def execute(operations):
|
|
""" generates and executes the operations on the servers """
|
|
router = import_class(settings.ORCHESTRATION_ROUTER)
|
|
# Generate scripts per server+backend
|
|
scripts = {}
|
|
cache = {}
|
|
for operation in operations:
|
|
logger.debug("Queued %s" % str(operation))
|
|
servers = router.get_servers(operation, cache=cache)
|
|
for server in servers:
|
|
key = (server, operation.backend)
|
|
if key not in scripts:
|
|
scripts[key] = (operation.backend(), [operation])
|
|
scripts[key][0].prepare()
|
|
else:
|
|
scripts[key][1].append(operation)
|
|
method = getattr(scripts[key][0], operation.action)
|
|
method(operation.instance)
|
|
# Execute scripts on each server
|
|
threads = []
|
|
executions = []
|
|
for key, value in scripts.iteritems():
|
|
server, __ = key
|
|
backend, operations = value
|
|
backend.commit()
|
|
execute = as_task(backend.execute)
|
|
execute = close_connection(execute)
|
|
thread = threading.Thread(target=execute, args=(server,))
|
|
thread.start()
|
|
threads.append(thread)
|
|
executions.append((execute, operations))
|
|
[ thread.join() for thread in threads ]
|
|
logs = []
|
|
for execution, operations in executions:
|
|
for operation in operations:
|
|
logger.info("Executed %s" % str(operation))
|
|
operation.log = execution.log
|
|
operation.save()
|
|
logger.debug(execution.log.stdout)
|
|
logger.debug(execution.log.stderr)
|
|
logs.append(execution.log)
|
|
return logs
|