Added async functionality on backends
This commit is contained in:
parent
31ac035c27
commit
39bf68caad
14
TODO.md
14
TODO.md
|
@ -334,7 +334,7 @@ pip3 install https://github.com/APSL/django-mailer-2/archive/master.zip
|
||||||
# all signals + accouns.register() services.register() on apps.py
|
# all signals + accouns.register() services.register() on apps.py
|
||||||
|
|
||||||
# if backend.async: don't join.
|
# if backend.async: don't join.
|
||||||
# RELATED: domains.sync to ns3 make it async
|
# RELATED: domains.sync to ns3 make it async backend rather than cron based ?
|
||||||
|
|
||||||
from orchestra.contrib.tasks import task
|
from orchestra.contrib.tasks import task
|
||||||
import time, sys
|
import time, sys
|
||||||
|
@ -358,11 +358,8 @@ TODO mount the filesystem with "nosuid" option
|
||||||
# autoiscover modules on app.ready()
|
# autoiscover modules on app.ready()
|
||||||
# uwse uwsgi cron: decorator or config cron = 59 2 -1 -1 -1 %(virtualenv)/bin/python manage.py runmyfunnytask
|
# uwse uwsgi cron: decorator or config cron = 59 2 -1 -1 -1 %(virtualenv)/bin/python manage.py runmyfunnytask
|
||||||
|
|
||||||
# SecondaryMailServerBackend and check_origin signal
|
|
||||||
try: import uwsgi to know its running uwsgi
|
|
||||||
# avoid cron email errors when failing hard
|
# avoid cron email errors when failing hard
|
||||||
|
|
||||||
|
|
||||||
# mailboxes.address settings multiple local domains, not only one?
|
# mailboxes.address settings multiple local domains, not only one?
|
||||||
# backend.context = self.get_context() or save(obj, context=None)
|
# backend.context = self.get_context() or save(obj, context=None)
|
||||||
|
|
||||||
|
@ -372,20 +369,11 @@ try: import uwsgi to know its running uwsgi
|
||||||
|
|
||||||
# MERGE beats and inspect INSTALLED_APPS and get IS_ENABLED
|
# MERGE beats and inspect INSTALLED_APPS and get IS_ENABLED
|
||||||
|
|
||||||
# make exceptions fot check origin shit
|
|
||||||
|
|
||||||
# rename virtual_maps to virtual_alias_maps and remove virtual_alias_domains ?
|
# rename virtual_maps to virtual_alias_maps and remove virtual_alias_domains ?
|
||||||
# virtdomains file is not ideal, prevent fake/error on domains there! and make sure this file is required!
|
# virtdomains file is not ideal, prevent fake/error on domains there! and make sure this file is required!
|
||||||
|
|
||||||
|
|
||||||
# Message last_retry auto_now doesn't work!
|
# Message last_retry auto_now doesn't work!
|
||||||
|
|
||||||
|
|
||||||
# LOCK beat lockfile, all tasks (monitors, billing (or unique number will fuck the transaction?))
|
|
||||||
|
|
||||||
# don't block on beat, and --report periodic tasks
|
# don't block on beat, and --report periodic tasks
|
||||||
|
|
||||||
# Deprecate restart/start/stop services (do touch wsgi.py and fuck celery)
|
# Deprecate restart/start/stop services (do touch wsgi.py and fuck celery)
|
||||||
|
|
||||||
|
|
||||||
# high perf apache sync,
|
|
||||||
|
|
|
@ -25,10 +25,10 @@ STATE_COLORS = {
|
||||||
|
|
||||||
class RouteAdmin(admin.ModelAdmin):
|
class RouteAdmin(admin.ModelAdmin):
|
||||||
list_display = (
|
list_display = (
|
||||||
'backend', 'host', 'match', 'display_model', 'display_actions', 'is_active'
|
'backend', 'host', 'match', 'display_model', 'display_actions', 'async', 'is_active'
|
||||||
)
|
)
|
||||||
list_editable = ('host', 'match', 'is_active')
|
list_editable = ('host', 'match', 'async', 'is_active')
|
||||||
list_filter = ('host', 'is_active', 'backend')
|
list_filter = ('host', 'is_active', 'async', 'backend')
|
||||||
ordering = ('backend',)
|
ordering = ('backend',)
|
||||||
|
|
||||||
BACKEND_HELP_TEXT = helpers.get_backends_help_text(ServiceBackend.get_backends())
|
BACKEND_HELP_TEXT = helpers.get_backends_help_text(ServiceBackend.get_backends())
|
||||||
|
|
|
@ -40,6 +40,7 @@ class ServiceBackend(plugins.Plugin, metaclass=ServiceMount):
|
||||||
script_executable = '/bin/bash'
|
script_executable = '/bin/bash'
|
||||||
function_method = methods.Python
|
function_method = methods.Python
|
||||||
type = 'task' # 'sync'
|
type = 'task' # 'sync'
|
||||||
|
# Don't wait for the backend to finish before continuing with request/response
|
||||||
ignore_fields = []
|
ignore_fields = []
|
||||||
actions = []
|
actions = []
|
||||||
default_route_match = 'True'
|
default_route_match = 'True'
|
||||||
|
@ -160,15 +161,22 @@ class ServiceBackend(plugins.Plugin, metaclass=ServiceMount):
|
||||||
time = timezone.now().strftime("%h %d, %Y %I:%M:%S %Z")
|
time = timezone.now().strftime("%h %d, %Y %I:%M:%S %Z")
|
||||||
return "Generated by Orchestra at %s" % time
|
return "Generated by Orchestra at %s" % time
|
||||||
|
|
||||||
def execute(self, server, async=False):
|
def create_log(self, server, **kwargs):
|
||||||
from .models import BackendLog
|
from .models import BackendLog
|
||||||
scripts = self.scripts
|
state = BackendLog.RECEIVED
|
||||||
state = BackendLog.STARTED
|
run = bool(self.scripts) or (self.force_empty_action_execution or bool(self.content))
|
||||||
run = bool(scripts) or (self.force_empty_action_execution or bool(self.content))
|
|
||||||
if not run:
|
if not run:
|
||||||
state = BackendLog.NOTHING
|
state = BackendLog.NOTHING
|
||||||
log = BackendLog.objects.create(backend=self.get_name(), state=state, server=server)
|
log = BackendLog.objects.create(backend=self.get_name(), state=state, server=server)
|
||||||
|
return log
|
||||||
|
|
||||||
|
def execute(self, server, async=False, log=None):
|
||||||
|
from .models import BackendLog
|
||||||
|
if log is None:
|
||||||
|
log = self.create_log(server)
|
||||||
|
run = log.state != BackendLog.NOTHING
|
||||||
if run:
|
if run:
|
||||||
|
scripts = self.scripts
|
||||||
for method, commands in scripts:
|
for method, commands in scripts:
|
||||||
method(log, server, commands, async)
|
method(log, server, commands, async)
|
||||||
if log.state != BackendLog.SUCCESS:
|
if log.state != BackendLog.SUCCESS:
|
||||||
|
|
|
@ -70,9 +70,18 @@ def send_report(method, args, log):
|
||||||
mail_admins(subject, message, html_message=html_message)
|
mail_admins(subject, message, html_message=html_message)
|
||||||
|
|
||||||
|
|
||||||
|
def get_backend_url(ids):
|
||||||
|
if len(ids) == 1:
|
||||||
|
return reverse('admin:orchestration_backendlog_change', args=ids)
|
||||||
|
elif len(ids) > 1:
|
||||||
|
url = reverse('admin:orchestration_backendlog_changelist')
|
||||||
|
return url + '?id__in=%s' % ','.join(map(str, ids))
|
||||||
|
return ''
|
||||||
|
|
||||||
def message_user(request, logs):
|
def message_user(request, logs):
|
||||||
total, successes = 0, 0
|
total, successes, async = 0, 0, 0
|
||||||
ids = []
|
ids = []
|
||||||
|
async_ids = []
|
||||||
for log in logs:
|
for log in logs:
|
||||||
total += 1
|
total += 1
|
||||||
if log.state != log.EXCEPTION:
|
if log.state != log.EXCEPTION:
|
||||||
|
@ -80,25 +89,42 @@ def message_user(request, logs):
|
||||||
ids.append(log.pk)
|
ids.append(log.pk)
|
||||||
if log.state in (log.SUCCESS, log.NOTHING):
|
if log.state in (log.SUCCESS, log.NOTHING):
|
||||||
successes += 1
|
successes += 1
|
||||||
errors = total-successes
|
elif log.state in (log.RECEIVED, log.STARTED):
|
||||||
if len(ids) == 1:
|
async += 1
|
||||||
url = reverse('admin:orchestration_backendlog_change', args=ids)
|
async_ids.append(log.id)
|
||||||
href = '<a href="{}">backends</a>'.format(url)
|
errors = total-successes-async
|
||||||
elif len(ids) > 1:
|
url = get_backend_url(ids)
|
||||||
url = reverse('admin:orchestration_backendlog_changelist')
|
async_url = get_backend_url(async_ids)
|
||||||
url += '?id__in=%s' % ','.join(map(str, ids))
|
async_msg = ''
|
||||||
href = '<a href="{}">backends</a>'.format(url)
|
if async:
|
||||||
else:
|
async_msg = ungettext(
|
||||||
href = ''
|
_('<a href="{async_url}">{async} backend</a> is running on the background'),
|
||||||
|
_('<a href="{async_url}">{async} backends</a> are running on the background'),
|
||||||
|
async)
|
||||||
if errors:
|
if errors:
|
||||||
msg = ungettext(
|
msg = ungettext(
|
||||||
_('{errors} out of {total} {href} has fail to execute.'),
|
_('<a href="{url}">{errors} out of {total} backend</a> has fail to execute'),
|
||||||
_('{errors} out of {total} {href} have fail to execute.'),
|
_('<a href="{url}">{errors} out of {total} backends</a> have fail to execute'),
|
||||||
errors)
|
errors)
|
||||||
messages.error(request, mark_safe(msg.format(errors=errors, total=total, href=href)))
|
if async_msg:
|
||||||
|
msg += ', ' + str(async_msg)
|
||||||
|
msg = msg.format(errors=errors, async=async, async_url=async_url, total=total, url=url)
|
||||||
|
messages.error(request, mark_safe(msg + '.'))
|
||||||
|
elif successes:
|
||||||
|
if async_msg:
|
||||||
|
msg = ungettext(
|
||||||
|
_('<a href="{url}">{successes} out of {total} backend</a> has been executed'),
|
||||||
|
_('<a href="{url}">{successes} out of {total} backends</a> have been executed'),
|
||||||
|
successes)
|
||||||
|
msg += ', ' + str(async_msg)
|
||||||
|
else:
|
||||||
|
msg = ungettext(
|
||||||
|
_('<a href="{url}">{total} backend</a> has been executed'),
|
||||||
|
_('<a href="{url}">{total} backends</a> have been executed'),
|
||||||
|
total)
|
||||||
|
msg = msg.format(total=total, url=url, async_url=async_url, async=async, successes=successes)
|
||||||
|
messages.success(request, mark_safe(msg + '.'))
|
||||||
else:
|
else:
|
||||||
msg = ungettext(
|
msg = async_msg.format(url=url, async_url=async_url, async=async)
|
||||||
_('{total} {href} has been executed.'),
|
messages.success(request, mark_safe(msg + '.'))
|
||||||
_('{total} {href} have been executed.'),
|
|
||||||
total)
|
|
||||||
messages.success(request, mark_safe(msg.format(total=total, href=href)))
|
|
||||||
|
|
|
@ -2,6 +2,7 @@ import logging
|
||||||
import threading
|
import threading
|
||||||
import traceback
|
import traceback
|
||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
|
from functools import partial
|
||||||
|
|
||||||
from django.core.mail import mail_admins
|
from django.core.mail import mail_admins
|
||||||
|
|
||||||
|
@ -19,7 +20,7 @@ logger = logging.getLogger(__name__)
|
||||||
router = import_class(settings.ORCHESTRATION_ROUTER)
|
router = import_class(settings.ORCHESTRATION_ROUTER)
|
||||||
|
|
||||||
|
|
||||||
def as_task(execute):
|
def as_task(execute, log, operations):
|
||||||
def wrapper(*args, **kwargs):
|
def wrapper(*args, **kwargs):
|
||||||
""" send report """
|
""" send report """
|
||||||
# Tasks run on a separate transaction pool (thread), no need to temper with the transaction
|
# Tasks run on a separate transaction pool (thread), no need to temper with the transaction
|
||||||
|
@ -33,12 +34,21 @@ def as_task(execute):
|
||||||
logger.error(subject)
|
logger.error(subject)
|
||||||
logger.error(message)
|
logger.error(message)
|
||||||
mail_admins(subject, message)
|
mail_admins(subject, message)
|
||||||
|
log.state = BackendLog.EXCEPTION
|
||||||
|
log.stderr = traceback.format_exc()
|
||||||
|
log.save(update_fields=('state', 'stderr'))
|
||||||
# We don't propagate the exception further to avoid transaction rollback
|
# We don't propagate the exception further to avoid transaction rollback
|
||||||
else:
|
finally:
|
||||||
# Using the wrapper function as threader messenger for the execute output
|
# Store the operation
|
||||||
# Absense of it will indicate a failure at this stage
|
for operation in operations:
|
||||||
wrapper.log = log
|
logger.info("Executed %s" % str(operation))
|
||||||
return log
|
if operation.instance.pk:
|
||||||
|
# Not all backends are called with objects saved on the database
|
||||||
|
operation.store(log)
|
||||||
|
stdout = log.stdout.strip()
|
||||||
|
stdout and logger.debug('STDOUT %s', stdout)
|
||||||
|
stderr = log.stderr.strip()
|
||||||
|
stderr and logger.debug('STDERR %s', stderr)
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
|
@ -92,41 +102,31 @@ def execute(scripts, block=False, async=False):
|
||||||
logger.info('Orchestration execution is dissabled by ORCHESTRATION_DISABLE_EXECUTION settings.')
|
logger.info('Orchestration execution is dissabled by ORCHESTRATION_DISABLE_EXECUTION settings.')
|
||||||
return []
|
return []
|
||||||
# Execute scripts on each server
|
# Execute scripts on each server
|
||||||
threads = []
|
|
||||||
executions = []
|
executions = []
|
||||||
|
threads_to_join = []
|
||||||
|
logs = []
|
||||||
for key, value in scripts.items():
|
for key, value in scripts.items():
|
||||||
server, __ = key
|
server, __ = key
|
||||||
backend, operations = value
|
backend, operations = value
|
||||||
execute = as_task(backend.execute)
|
args = (server,)
|
||||||
|
kwargs = {
|
||||||
|
'async': async or server.async
|
||||||
|
}
|
||||||
|
log = backend.create_log(*args, **kwargs)
|
||||||
|
kwargs['log'] = log
|
||||||
|
task = as_task(backend.execute, log, operations)
|
||||||
logger.debug('%s is going to be executed on %s' % (backend, server))
|
logger.debug('%s is going to be executed on %s' % (backend, server))
|
||||||
if block:
|
if block:
|
||||||
# Execute one backend at a time, no need for threads
|
# Execute one backend at a time, no need for threads
|
||||||
execute(server, async=async)
|
task(*args, **kwargs)
|
||||||
else:
|
else:
|
||||||
execute = close_connection(execute)
|
task = close_connection(task)
|
||||||
thread = threading.Thread(target=execute, args=(server,), kwargs={'async': async})
|
thread = threading.Thread(target=task, args=args, kwargs=kwargs)
|
||||||
thread.start()
|
thread.start()
|
||||||
threads.append(thread)
|
if not server.async:
|
||||||
executions.append((execute, operations))
|
threads_to_join.append(thread)
|
||||||
[ thread.join() for thread in threads ]
|
logs.append(log)
|
||||||
logs = []
|
[ thread.join() for thread in threads_to_join ]
|
||||||
# collect results
|
|
||||||
for execution, operations in executions:
|
|
||||||
# There is no log if an exception has been rised at the very end of the execution
|
|
||||||
if hasattr(execution, 'log'):
|
|
||||||
for operation in operations:
|
|
||||||
logger.info("Executed %s" % str(operation))
|
|
||||||
if operation.instance.pk:
|
|
||||||
# Not all backends are called with objects saved on the database
|
|
||||||
operation.store(execution.log)
|
|
||||||
stdout = execution.log.stdout.strip()
|
|
||||||
stdout and logger.debug('STDOUT %s', stdout)
|
|
||||||
stderr = execution.log.stderr.strip()
|
|
||||||
stderr and logger.debug('STDERR %s', stderr)
|
|
||||||
logs.append(execution.log)
|
|
||||||
else:
|
|
||||||
mocked_log = BackendLog(state=BackendLog.EXCEPTION)
|
|
||||||
logs.append(mocked_log)
|
|
||||||
return logs
|
return logs
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -33,8 +33,9 @@ def SSH(backend, log, server, cmds, async=False):
|
||||||
digest = hashlib.md5(bscript).hexdigest()
|
digest = hashlib.md5(bscript).hexdigest()
|
||||||
path = os.path.join(settings.ORCHESTRATION_TEMP_SCRIPT_DIR, digest)
|
path = os.path.join(settings.ORCHESTRATION_TEMP_SCRIPT_DIR, digest)
|
||||||
remote_path = "%s.remote" % path
|
remote_path = "%s.remote" % path
|
||||||
|
log.state = log.STARTED
|
||||||
log.script = '# %s\n%s' % (remote_path, script)
|
log.script = '# %s\n%s' % (remote_path, script)
|
||||||
log.save(update_fields=['script'])
|
log.save(update_fields=('script', 'state'))
|
||||||
if not cmds:
|
if not cmds:
|
||||||
return
|
return
|
||||||
channel = None
|
channel = None
|
||||||
|
@ -133,8 +134,9 @@ def Python(backend, log, server, cmds, async=False):
|
||||||
# TODO collect stdout?
|
# TODO collect stdout?
|
||||||
script = [ str(cmd.func.__name__) + str(cmd.args) for cmd in cmds ]
|
script = [ str(cmd.func.__name__) + str(cmd.args) for cmd in cmds ]
|
||||||
script = json.dumps(script, indent=4).replace('"', '')
|
script = json.dumps(script, indent=4).replace('"', '')
|
||||||
|
log.state = log.STARTED
|
||||||
log.script = '\n'.join([log.script, script])
|
log.script = '\n'.join([log.script, script])
|
||||||
log.save(update_fields=['script'])
|
log.save(update_fields=('script', 'state'))
|
||||||
try:
|
try:
|
||||||
for cmd in cmds:
|
for cmd in cmds:
|
||||||
with CaptureStdout() as stdout:
|
with CaptureStdout() as stdout:
|
||||||
|
|
|
@ -0,0 +1,34 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
|
from django.db import models, migrations
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('orchestration', '0001_initial'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='route',
|
||||||
|
name='async',
|
||||||
|
field=models.BooleanField(help_text='Whether or not block the request/response cycle waitting this backend to finish its execution.', default=False),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='backendlog',
|
||||||
|
name='state',
|
||||||
|
field=models.CharField(verbose_name='state', choices=[('RECEIVED', 'RECEIVED'), ('TIMEOUT', 'TIMEOUT'), ('STARTED', 'STARTED'), ('SUCCESS', 'SUCCESS'), ('FAILURE', 'FAILURE'), ('ERROR', 'ERROR'), ('ABORTED', 'ABORTED'), ('REVOKED', 'REVOKED'), ('NOTHING', 'NOTHING')], default='RECEIVED', max_length=16),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='backendlog',
|
||||||
|
name='stderr',
|
||||||
|
field=models.TextField(verbose_name='stderr'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='route',
|
||||||
|
name='backend',
|
||||||
|
field=models.CharField(choices=[('Apache2Traffic', '[M] Apache 2 Traffic'), ('DovecotMaildirDisk', '[M] Dovecot Maildir size'), ('Exim4Traffic', '[M] Exim4 traffic'), ('MailmanSubscribers', '[M] Mailman subscribers'), ('MailmanTraffic', '[M] Mailman traffic'), ('MysqlDisk', '[M] MySQL disk'), ('OpenVZTraffic', '[M] OpenVZTraffic'), ('PostfixMailscannerTraffic', '[M] Postfix-Mailscanner traffic'), ('UNIXUserDisk', '[M] UNIX user disk'), ('VsFTPdTraffic', '[M] VsFTPd traffic'), ('Apache2Backend', '[S] Apache 2'), ('BSCWBackend', '[S] BSCW SaaS'), ('Bind9MasterDomainBackend', '[S] Bind9 master domain'), ('Bind9SlaveDomainBackend', '[S] Bind9 slave domain'), ('DokuWikiMuBackend', '[S] DokuWiki multisite'), ('DovecotPostfixPasswdVirtualUserBackend', '[S] Dovecot-Postfix virtualuser'), ('DrupalMuBackend', '[S] Drupal multisite'), ('GitLabSaaSBackend', '[S] GitLab SaaS'), ('AutoresponseBackend', '[S] Mail autoresponse'), ('MailmanBackend', '[S] Mailman'), ('MailmanVirtualDomainBackend', '[S] Mailman virtdomain-only'), ('MySQLBackend', '[S] MySQL database'), ('MySQLUserBackend', '[S] MySQL user'), ('PHPBackend', '[S] PHP FPM/FCGID'), ('PostfixAddressBackend', '[S] Postfix address'), ('PostfixAddressVirtualDomainBackend', '[S] Postfix address virtdomain-only'), ('uWSGIPythonBackend', '[S] Python uWSGI'), ('StaticBackend', '[S] Static'), ('SymbolicLinkBackend', '[S] Symbolic link webapp'), ('SyncBind9MasterDomainBackend', '[S] Sync Bind9 master domain'), ('SyncBind9SlaveDomainBackend', '[S] Sync Bind9 slave domain'), ('UNIXUserMaildirBackend', '[S] UNIX maildir user'), ('UNIXUserBackend', '[S] UNIX user'), ('WebalizerAppBackend', '[S] Webalizer App'), ('WebalizerBackend', '[S] Webalizer Content'), ('WordPressBackend', '[S] Wordpress'), ('WordpressMuBackend', '[S] Wordpress multisite'), ('PhpListSaaSBackend', '[S] phpList SaaS')], verbose_name='backend', max_length=256),
|
||||||
|
),
|
||||||
|
]
|
|
@ -133,7 +133,9 @@ class Route(models.Model):
|
||||||
match = models.CharField(_("match"), max_length=256, blank=True, default='True',
|
match = models.CharField(_("match"), max_length=256, blank=True, default='True',
|
||||||
help_text=_("Python expression used for selecting the targe host, "
|
help_text=_("Python expression used for selecting the targe host, "
|
||||||
"<em>instance</em> referes to the current object."))
|
"<em>instance</em> referes to the current object."))
|
||||||
# async = models.BooleanField(default=False)
|
async = models.BooleanField(default=False,
|
||||||
|
help_text=_("Whether or not block the request/response cycle waitting this backend to "
|
||||||
|
"finish its execution."))
|
||||||
# method = models.CharField(_("method"), max_lenght=32, choices=method_choices,
|
# method = models.CharField(_("method"), max_lenght=32, choices=method_choices,
|
||||||
# default=MethodBackend.get_default())
|
# default=MethodBackend.get_default())
|
||||||
is_active = models.BooleanField(_("active"), default=True)
|
is_active = models.BooleanField(_("active"), default=True)
|
||||||
|
@ -148,6 +150,7 @@ class Route(models.Model):
|
||||||
def backend_class(self):
|
def backend_class(self):
|
||||||
return ServiceBackend.get_backend(self.backend)
|
return ServiceBackend.get_backend(self.backend)
|
||||||
|
|
||||||
|
# TODO rename to get_hosts
|
||||||
@classmethod
|
@classmethod
|
||||||
def get_servers(cls, operation, **kwargs):
|
def get_servers(cls, operation, **kwargs):
|
||||||
cache = kwargs.get('cache', {})
|
cache = kwargs.get('cache', {})
|
||||||
|
@ -169,6 +172,7 @@ class Route(models.Model):
|
||||||
else:
|
else:
|
||||||
for route in routes:
|
for route in routes:
|
||||||
if route.matches(operation.instance):
|
if route.matches(operation.instance):
|
||||||
|
route.host.async = route.async
|
||||||
servers.append(route.host)
|
servers.append(route.host)
|
||||||
return servers
|
return servers
|
||||||
|
|
||||||
|
|
|
@ -8,7 +8,7 @@ from .backends import ServiceMonitor
|
||||||
|
|
||||||
@task(name='resources.Monitor')
|
@task(name='resources.Monitor')
|
||||||
def monitor(resource_id, ids=None, async=True):
|
def monitor(resource_id, ids=None, async=True):
|
||||||
with LockFile('/dev/shm/resources.monitor.lock', expire=60*60):
|
with LockFile('/dev/shm/resources.monitor-%i.lock' % resource_id, expire=60*60):
|
||||||
from .models import ResourceData, Resource
|
from .models import ResourceData, Resource
|
||||||
resource = Resource.objects.get(pk=resource_id)
|
resource = Resource.objects.get(pk=resource_id)
|
||||||
resource_model = resource.content_type.model_class()
|
resource_model = resource.content_type.model_class()
|
||||||
|
|
|
@ -28,9 +28,15 @@ class WebAppServiceMixin(object):
|
||||||
def set_under_construction(self, context):
|
def set_under_construction(self, context):
|
||||||
if context['under_construction_path']:
|
if context['under_construction_path']:
|
||||||
self.append(textwrap.dedent("""\
|
self.append(textwrap.dedent("""\
|
||||||
if [[ $CREATED == 1 ]]; then
|
if [[ $CREATED == 1 && ! $(ls -A %(app_path)s) ]]; then
|
||||||
cp -r %(under_construction_path)s %(app_path)s
|
{
|
||||||
chown -R %(user)s:%(group)s %(app_path)s
|
# Wait for other backends to do their thing or cp under construction
|
||||||
|
sleep 1
|
||||||
|
if [[ ! $(ls -A %(app_path)s) ]]; then
|
||||||
|
cp -r %(under_construction_path)s %(app_path)s
|
||||||
|
chown -R %(user)s:%(group)s %(app_path)s
|
||||||
|
fi
|
||||||
|
} &
|
||||||
fi""") % context
|
fi""") % context
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -34,13 +34,13 @@ class PHPBackend(WebAppServiceMixin, ServiceController):
|
||||||
def save(self, webapp):
|
def save(self, webapp):
|
||||||
context = self.get_context(webapp)
|
context = self.get_context(webapp)
|
||||||
self.create_webapp_dir(context)
|
self.create_webapp_dir(context)
|
||||||
self.set_under_construction(context)
|
|
||||||
if webapp.type_instance.is_fpm:
|
if webapp.type_instance.is_fpm:
|
||||||
self.save_fpm(webapp, context)
|
self.save_fpm(webapp, context)
|
||||||
self.delete_fcgid(webapp, context)
|
self.delete_fcgid(webapp, context)
|
||||||
elif webapp.type_instance.is_fcgid:
|
elif webapp.type_instance.is_fcgid:
|
||||||
self.save_fcgid(webapp, context)
|
self.save_fcgid(webapp, context)
|
||||||
self.delete_fpm(webapp, context)
|
self.delete_fpm(webapp, context)
|
||||||
|
self.set_under_construction(context)
|
||||||
|
|
||||||
def save_fpm(self, webapp, context):
|
def save_fpm(self, webapp, context):
|
||||||
self.append(textwrap.dedent("""\
|
self.append(textwrap.dedent("""\
|
||||||
|
@ -108,36 +108,38 @@ class PHPBackend(WebAppServiceMixin, ServiceController):
|
||||||
def prepare(self):
|
def prepare(self):
|
||||||
super(PHPBackend, self).prepare()
|
super(PHPBackend, self).prepare()
|
||||||
# Coordinate apache restart with php backend in order not to overdo it
|
# Coordinate apache restart with php backend in order not to overdo it
|
||||||
self.append('echo "PHPBackend" >> /dev/shm/restart.apache2')
|
self.append(textwrap.dedent("""\
|
||||||
|
backend="PHPBackend"
|
||||||
|
echo "$backend" >> /dev/shm/restart.apache2
|
||||||
|
""")
|
||||||
|
)
|
||||||
|
|
||||||
def commit(self):
|
def commit(self):
|
||||||
self.append(textwrap.dedent("""
|
self.append(textwrap.dedent("""
|
||||||
if [[ $UPDATED_FPM -eq 1 ]]; then
|
if [[ $UPDATED_FPM -eq 1 ]]; then
|
||||||
service php5-fpm reload
|
service php5-fpm reload
|
||||||
fi
|
fi
|
||||||
# Coordinate apache restart with Apache2Backend
|
|
||||||
restart=0
|
# Coordinate Apache restart with other concurrent backends (i.e. Apache2Backend)
|
||||||
backend='PHPBackend'
|
is_last=0
|
||||||
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked || {
|
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked || {
|
||||||
sleep 0.1
|
sleep 0.2
|
||||||
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked
|
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked
|
||||||
}
|
}
|
||||||
state="$(grep -v $backend /dev/shm/restart.apache2.locked)" || restart=1
|
state="$(grep -v "$backend" /dev/shm/restart.apache2.locked)" || is_last=1
|
||||||
echo -n "$state" > /dev/shm/restart.apache2.locked
|
if [[ $is_last -eq 1 ]]; then
|
||||||
if [[ $UPDATED_APACHE -eq 1 ]]; then
|
if [[ $UPDATED_APACHE -eq 1 || "$state" =~ .*RESTART$ ]]; then
|
||||||
if [[ $restart == 1 ]]; then
|
|
||||||
service apache2 status && service apache2 reload || service apache2 start
|
service apache2 status && service apache2 reload || service apache2 start
|
||||||
rm /dev/shm/restart.apache2.locked
|
|
||||||
else
|
|
||||||
echo "$backend RESTART" >> /dev/shm/restart.apache2.locked
|
|
||||||
mv /dev/shm/restart.apache2.locked /dev/shm/restart.apache2
|
|
||||||
fi
|
fi
|
||||||
elif [[ "$state" =~ .*RESTART$ ]]; then
|
|
||||||
rm /dev/shm/restart.apache2.locked
|
rm /dev/shm/restart.apache2.locked
|
||||||
service apache2 status && service apache2 reload || service apache2 start
|
|
||||||
else
|
else
|
||||||
|
echo -n "$state" > /dev/shm/restart.apache2.locked
|
||||||
|
if [[ $UPDATED_APACHE -eq 1 ]]; then
|
||||||
|
echo "$backend RESTART" >> /dev/shm/restart.apache2.locked
|
||||||
|
fi
|
||||||
mv /dev/shm/restart.apache2.locked /dev/shm/restart.apache2
|
mv /dev/shm/restart.apache2.locked /dev/shm/restart.apache2
|
||||||
fi
|
fi
|
||||||
|
# End of coordination
|
||||||
""")
|
""")
|
||||||
)
|
)
|
||||||
super(PHPBackend, self).commit()
|
super(PHPBackend, self).commit()
|
||||||
|
|
|
@ -132,34 +132,36 @@ class Apache2Backend(ServiceController):
|
||||||
def prepare(self):
|
def prepare(self):
|
||||||
super(Apache2Backend, self).prepare()
|
super(Apache2Backend, self).prepare()
|
||||||
# Coordinate apache restart with php backend in order not to overdo it
|
# Coordinate apache restart with php backend in order not to overdo it
|
||||||
self.append('echo "Apache2Backend" >> /dev/shm/restart.apache2')
|
self.append(textwrap.dedent("""\
|
||||||
|
backend="Apache2Backend"
|
||||||
|
echo "$backend" >> /dev/shm/restart.apache2
|
||||||
|
""")
|
||||||
|
)
|
||||||
|
|
||||||
def commit(self):
|
def commit(self):
|
||||||
""" reload Apache2 if necessary """
|
""" reload Apache2 if necessary """
|
||||||
self.append(textwrap.dedent("""\
|
self.append(textwrap.dedent("""
|
||||||
# Coordinate apache restart with Apache2Backend
|
# Coordinate Apache restart with other concurrent backends (i.e. Apache2Backend)
|
||||||
restart=0
|
is_last=0
|
||||||
backend='Apache2Backend'
|
|
||||||
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked || {
|
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked || {
|
||||||
sleep 0.1
|
sleep 0.2
|
||||||
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked
|
mv /dev/shm/restart.apache2 /dev/shm/restart.apache2.locked
|
||||||
}
|
}
|
||||||
state="$(grep -v $backend /dev/shm/restart.apache2.locked)" || restart=1
|
state="$(grep -v "$backend" /dev/shm/restart.apache2.locked)" || is_last=1
|
||||||
echo -n "$state" > /dev/shm/restart.apache2.locked
|
if [[ $is_last -eq 1 ]]; then
|
||||||
if [[ $UPDATED_APACHE -eq 1 ]]; then
|
if [[ $UPDATED_APACHE -eq 1 || "$state" =~ .*RESTART$ ]]; then
|
||||||
if [[ $restart == 1 ]]; then
|
|
||||||
service apache2 status && service apache2 reload || service apache2 start
|
service apache2 status && service apache2 reload || service apache2 start
|
||||||
rm /dev/shm/restart.apache2.locked
|
|
||||||
else
|
|
||||||
echo "$backend RESTART" >> /dev/shm/restart.apache2.locked
|
|
||||||
mv /dev/shm/restart.apache2.locked /dev/shm/restart.apache2
|
|
||||||
fi
|
fi
|
||||||
elif [[ "$state" =~ .*RESTART$ ]]; then
|
|
||||||
rm /dev/shm/restart.apache2.locked
|
rm /dev/shm/restart.apache2.locked
|
||||||
service apache2 status && service apache2 reload || service apache2 start
|
|
||||||
else
|
else
|
||||||
|
echo -n "$state" > /dev/shm/restart.apache2.locked
|
||||||
|
if [[ $UPDATED_APACHE -eq 1 ]]; then
|
||||||
|
echo "$backend RESTART" >> /dev/shm/restart.apache2.locked
|
||||||
|
fi
|
||||||
mv /dev/shm/restart.apache2.locked /dev/shm/restart.apache2
|
mv /dev/shm/restart.apache2.locked /dev/shm/restart.apache2
|
||||||
fi""")
|
fi
|
||||||
|
# End of coordination
|
||||||
|
""")
|
||||||
)
|
)
|
||||||
super(Apache2Backend, self).commit()
|
super(Apache2Backend, self).commit()
|
||||||
|
|
||||||
|
|
|
@ -19,8 +19,8 @@ class Command(BaseCommand):
|
||||||
content = run('crontab -l || true').stdout.decode('utf8')
|
content = run('crontab -l || true').stdout.decode('utf8')
|
||||||
if 'orchestra-beat' not in content:
|
if 'orchestra-beat' not in content:
|
||||||
if context['venv']:
|
if context['venv']:
|
||||||
content += "* * * * * . %(venv)s/bin/activate && %(orchestra_beat)s %(site_dir)s/manage.py; deactivate" % context
|
content += "\n* * * * * . %(venv)s/bin/activate && %(orchestra_beat)s %(site_dir)s/manage.py; deactivate" % context
|
||||||
else:
|
else:
|
||||||
content += "* * * * * %(orchestra_beat)s %(site_dir)s/manage.py" % context
|
content += "\n* * * * * %(orchestra_beat)s %(site_dir)s/manage.py" % context
|
||||||
context['content'] = content
|
context['content'] = content
|
||||||
run("echo '%(content)s' | crontab" % context, display=True)
|
run("cat << EOF | crontab\n%(content)s\nEOF" % context, display=True)
|
||||||
|
|
|
@ -7,7 +7,7 @@ def close_connection(execute):
|
||||||
try:
|
try:
|
||||||
log = execute(*args, **kwargs)
|
log = execute(*args, **kwargs)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
pass
|
raise
|
||||||
else:
|
else:
|
||||||
wrapper.log = log
|
wrapper.log = log
|
||||||
finally:
|
finally:
|
||||||
|
|
|
@ -124,7 +124,7 @@ def join(iterator, display=False, silent=False, error_codes=[0]):
|
||||||
if display:
|
if display:
|
||||||
sys.stderr.write("\n\033[1;31mCommandError: %s %s\033[m\n" % (msg, err))
|
sys.stderr.write("\n\033[1;31mCommandError: %s %s\033[m\n" % (msg, err))
|
||||||
if not silent:
|
if not silent:
|
||||||
raise CommandError("%s %s %s" % (msg, err, out))
|
raise CommandError("%s %s" % (msg, err))
|
||||||
|
|
||||||
out.succeeded = not out.failed
|
out.succeeded = not out.failed
|
||||||
return out
|
return out
|
||||||
|
@ -158,6 +158,13 @@ def get_default_celeryd_username():
|
||||||
return user
|
return user
|
||||||
|
|
||||||
|
|
||||||
|
def touch(fname, mode=0o666, dir_fd=None, **kwargs):
|
||||||
|
flags = os.O_CREAT | os.O_APPEND
|
||||||
|
with os.fdopen(os.open(fname, flags=flags, mode=mode, dir_fd=dir_fd)) as f:
|
||||||
|
os.utime(f.fileno() if os.utime in os.supports_fd else fname,
|
||||||
|
dir_fd=None if os.supports_fd else dir_fd, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
class LockFile(object):
|
class LockFile(object):
|
||||||
""" File-based lock mechanism used for preventing concurrency problems """
|
""" File-based lock mechanism used for preventing concurrency problems """
|
||||||
def __init__(self, lockfile, expire=5*60, unlocked=False):
|
def __init__(self, lockfile, expire=5*60, unlocked=False):
|
||||||
|
|
Loading…
Reference in New Issue