mirror of
https://github.com/ansible/awx.git
synced 2026-02-12 07:04:45 -03:30
91dbc2de30
There is some history here. https://github.com/ansible/awx/pull/7190 <- This PR was an attempt at fixing a bug notting ran into where some jobs on k8s installs would get stuck in Waiting forever. The PR mentioned above introduced a bug where there are no instance groups on a fresh k8s-based install. This is because this process currently happens in the launch scripts, before the database is up. With this patch, queue / instance group registration happens in the heartbeat, right after auto-registering the instance.
140 lines
5.4 KiB
Python
140 lines
5.4 KiB
Python
# Copyright (c) 2017 Ansible Tower by Red Hat
|
|
# All Rights Reserved.
|
|
import sys
|
|
|
|
from awx.main.utils.pglock import advisory_lock
|
|
from awx.main.models import Instance, InstanceGroup
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.db import transaction
|
|
|
|
|
|
class InstanceNotFound(Exception):
|
|
def __init__(self, message, changed, *args, **kwargs):
|
|
self.message = message
|
|
self.changed = changed
|
|
super(InstanceNotFound, self).__init__(*args, **kwargs)
|
|
|
|
|
|
class RegisterQueue:
|
|
def __init__(self, queuename, controller, instance_percent, inst_min, hostname_list):
|
|
self.instance_not_found_err = None
|
|
self.queuename = queuename
|
|
self.controller = controller
|
|
self.instance_percent = instance_percent
|
|
self.instance_min = inst_min
|
|
self.hostname_list = hostname_list
|
|
|
|
def get_create_update_instance_group(self):
|
|
created = False
|
|
changed = False
|
|
(ig, created) = InstanceGroup.objects.get_or_create(name=self.queuename)
|
|
if ig.policy_instance_percentage != self.instance_percent:
|
|
ig.policy_instance_percentage = self.instance_percent
|
|
changed = True
|
|
if ig.policy_instance_minimum != self.instance_min:
|
|
ig.policy_instance_minimum = self.instance_min
|
|
changed = True
|
|
|
|
if changed:
|
|
ig.save()
|
|
|
|
return (ig, created, changed)
|
|
|
|
def update_instance_group_controller(self, ig):
|
|
changed = False
|
|
control_ig = None
|
|
|
|
if self.controller:
|
|
control_ig = InstanceGroup.objects.filter(name=self.controller).first()
|
|
|
|
if control_ig and ig.controller_id != control_ig.pk:
|
|
ig.controller = control_ig
|
|
ig.save()
|
|
changed = True
|
|
|
|
return (control_ig, changed)
|
|
|
|
def add_instances_to_group(self, ig):
|
|
changed = False
|
|
|
|
instance_list_unique = set([x.strip() for x in self.hostname_list if x])
|
|
instances = []
|
|
for inst_name in instance_list_unique:
|
|
instance = Instance.objects.filter(hostname=inst_name)
|
|
if instance.exists():
|
|
instances.append(instance[0])
|
|
else:
|
|
raise InstanceNotFound("Instance does not exist: {}".format(inst_name), changed)
|
|
|
|
ig.instances.add(*instances)
|
|
|
|
instance_list_before = ig.policy_instance_list
|
|
instance_list_after = instance_list_unique
|
|
new_instances = set(instance_list_after) - set(instance_list_before)
|
|
if new_instances:
|
|
changed = True
|
|
ig.policy_instance_list = ig.policy_instance_list + list(new_instances)
|
|
ig.save()
|
|
|
|
return (instances, changed)
|
|
|
|
def register(self):
|
|
with advisory_lock('cluster_policy_lock'):
|
|
with transaction.atomic():
|
|
changed2 = False
|
|
changed3 = False
|
|
(ig, created, changed1) = self.get_create_update_instance_group()
|
|
if created:
|
|
print("Creating instance group {}".format(ig.name))
|
|
elif not created:
|
|
print("Instance Group already registered {}".format(ig.name))
|
|
|
|
if self.controller:
|
|
(ig_ctrl, changed2) = self.update_instance_group_controller(ig)
|
|
if changed2:
|
|
print("Set controller group {} on {}.".format(self.controller, self.queuename))
|
|
|
|
try:
|
|
(instances, changed3) = self.add_instances_to_group(ig)
|
|
for i in instances:
|
|
print("Added instance {} to {}".format(i.hostname, ig.name))
|
|
except InstanceNotFound as e:
|
|
self.instance_not_found_err = e
|
|
|
|
if any([changed1, changed2, changed3]):
|
|
print('(changed: True)')
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--queuename', dest='queuename', type=str,
|
|
help='Queue to create/update')
|
|
parser.add_argument('--hostnames', dest='hostnames', type=str,
|
|
help='Comma-Delimited Hosts to add to the Queue (will not remove already assigned instances)')
|
|
parser.add_argument('--controller', dest='controller', type=str,
|
|
default='', help='The controlling group (makes this an isolated group)')
|
|
parser.add_argument('--instance_percent', dest='instance_percent', type=int, default=0,
|
|
help='The percentage of active instances that will be assigned to this group'),
|
|
parser.add_argument('--instance_minimum', dest='instance_minimum', type=int, default=0,
|
|
help='The minimum number of instance that will be retained for this group from available instances')
|
|
|
|
|
|
def handle(self, **options):
|
|
queuename = options.get('queuename')
|
|
if not queuename:
|
|
raise CommandError("Specify `--queuename` to use this command.")
|
|
ctrl = options.get('controller')
|
|
inst_per = options.get('instance_percent')
|
|
instance_min = options.get('instance_minimum')
|
|
hostname_list = []
|
|
if options.get('hostnames'):
|
|
hostname_list = options.get('hostnames').split(",")
|
|
|
|
rq = RegisterQueue(queuename, ctrl, inst_per, instance_min, hostname_list)
|
|
rq.register()
|
|
if rq.instance_not_found_err:
|
|
print(rq.instance_not_found_err.message)
|
|
sys.exit(1)
|