Synnefo

# Copyright 2011-2012 GRNET S.A. All rights reserved.

import datetime

from django.core.management.base import BaseCommand, CommandError

from synnefo.db.models import (Backend, VirtualMachine, Network,

                               pooled_rapi_client)

from synnefo.logic import reconciliation, utils

from synnefo.logic import backend as backend_mod

from synnefo.util.mac2eui64 import mac2eui64

    output_transaction = True  # The management command runs inside

                               # an SQL transaction

        make_option('--detect-stale', action='store_true', dest='detect_stale',

                    default=False, help='Detect stale VM entries in DB'),

        make_option('--detect-orphans', action='store_true',

                    dest='detect_orphans',

                    default=False, help='Detect orphan instances in Ganeti'),

        make_option('--detect-unsynced', action='store_true',

                    dest='detect_unsynced',

                    default=False, help='Detect unsynced operstate between ' +

                                        'DB and Ganeti'),

        make_option('--detect-build-errors', action='store_true',

                    dest='detect_build_errors', default=False,

                    help='Detect instances with build error'),

        make_option('--detect-unsynced-nics', action='store_true',

                    dest='detect_unsynced_nics', default=False,

                    help='Detect unsynced nics between DB and Ganeti'),

        make_option('--detect-unsynced-flavors', action='store_true',

                    dest='detect_unsynced_flavors', default=False,

                    help='Detect unsynced flavors between DB and Ganeti'),

        make_option('--detect-all', action='store_true',

                    dest='detect_all',

                    default=False, help='Enable all --detect-* arguments'),

        make_option('--fix-build-errors', action='store_true',

                    dest='fix_build_errors', default=False,

                    help='Fix (remove) instances with build errors'),

        make_option('--backend-id', default=None, dest='backend-id',

                    help='Reconcilie VMs only for this backend'),

        keys_detect = [k for k in options.keys() if k.startswith('detect_')]

        if not reduce(lambda x, y: x or y,

                      map(lambda x: options[x], keys_detect)):

            options['detect_all'] = True

        if options['detect_all']:

            for kd in keys_detect:

                options[kd] = True

        for kf in keys_fix:

            kd = kf.replace('fix_', 'detect_', 1)

            if (options[kf] and not options[kd]):

                raise CommandError("Cannot use --%s without corresponding "

                                   "--%s argument" % (kf, kd))

        verbosity = int(options['verbosity'])

        self._process_args(options)

            backends = Backend.objects.filter(offline=False)

        with_nics = options["detect_unsynced_nics"]

        DBVMs = reconciliation.get_servers_from_db(backends, with_nics)

        GanetiVMs = reconciliation.get_instances_from_ganeti(backends)

        #

        # Detect problems

        #

        if options['detect_stale']:

            stale = reconciliation.stale_servers_in_db(DBVMs, GanetiVMs)

            if len(stale) > 0:

                print >> sys.stderr, "Found the following stale server IDs: "

                print "    " + "\n    ".join(

                    [str(x) for x in stale])

            elif verbosity == 2:

                print >> sys.stderr, "Found no stale server IDs in DB."

        if options['detect_orphans']:

            orphans = reconciliation.orphan_instances_in_ganeti(DBVMs,

                                                                GanetiVMs)

            if len(orphans) > 0:

                print >> sys.stderr, "Found orphan Ganeti instances with IDs: "

                print "    " + "\n    ".join(

                    [str(x) for x in orphans])

            elif verbosity == 2:

                print >> sys.stderr, "Found no orphan Ganeti instances."

        if options['detect_unsynced']:

            unsynced = reconciliation.unsynced_operstate(DBVMs, GanetiVMs)

            if len(unsynced) > 0:

                print >> sys.stderr, "The operstate of the following server" \

                                     " IDs is out-of-sync:"

                print "    " + "\n    ".join(

                    ["%d is %s in DB, %s in Ganeti" %

                     (x[0], x[1], ('UP' if x[2] else 'DOWN'))

                     for x in unsynced])

            elif verbosity == 2:

                print >> sys.stderr, "The operstate of all servers is in sync."

        if options['detect_build_errors']:

            build_errors = reconciliation.\

                instances_with_build_errors(DBVMs, GanetiVMs)

            if len(build_errors) > 0:

                msg = "The os for the following server IDs was not build"\

                      " successfully:"

                print >> sys.stderr, msg

                print "    " + "\n    ".join(

                    ["%d" % x for x in build_errors])

            elif verbosity == 2:

                print >> sys.stderr, "Found no instances with build errors."

        if options['detect_unsynced_nics']:

            def pretty_print_nics(nics):

                if not nics:

                    print ''.ljust(18) + 'None'

                for index, info in nics.items():

                    print ''.ljust(18) + 'nic/' + str(index) +\

                          ': MAC: %s, IP: %s, Network: %s' % \

                          (info['mac'], info['ipv4'], info['network'])

            unsynced_nics = reconciliation.unsynced_nics(DBVMs, GanetiVMs)

            if len(unsynced_nics) > 0:

                msg = "The NICs of the servers with the following IDs are"\

                      " unsynced:"

                print >> sys.stderr, msg

                for id, nics in unsynced_nics.items():

                    print ''.ljust(2) + '%6d:' % id

                    print ''.ljust(8) + '%8s:' % 'DB'

                    pretty_print_nics(nics[0])

                    print ''.ljust(8) + '%8s:' % 'Ganeti'

                    pretty_print_nics(nics[1])

            elif verbosity == 2:

                print >> sys.stderr, "All instance nics are synced."

        if options["detect_unsynced_flavors"]:

            unsynced_flavors = reconciliation.unsynced_flavors(DBVMs,

                                                               GanetiVMs)

            if len(unsynced_flavors) > 0:

                print >> sys.stderr, "The flavor of the following server" \

                                     " IDs is out-of-sync:"

                print "    " + "\n    ".join(

                    ["%d is %s in DB, %s in Ganeti" %

                     (x[0], x[1], x[2])

                     for x in unsynced_flavors])

            elif verbosity == 2:

                print >> sys.stderr, "All instance flavors are synced."

        #

        # Then fix them

        #

        if options['fix_stale'] and len(stale) > 0:

            print >> sys.stderr, \

                "Simulating successful Ganeti removal for %d " \

                "servers in the DB:" % len(stale)

            for vm in VirtualMachine.objects.filter(pk__in=stale):

                event_time = datetime.datetime.now()

                backend_mod.process_op_status(

                    vm=vm,

                    etime=event_time,

                    jobid=-0,

                    opcode='OP_INSTANCE_REMOVE', status='success',

                    logmsg='Reconciliation: simulated Ganeti event')

            print >> sys.stderr, "    ...done"

        if options['fix_orphans'] and len(orphans) > 0:

            print >> sys.stderr, \

                "Issuing OP_INSTANCE_REMOVE for %d Ganeti instances:" % \

                len(orphans)

            for id in orphans:

                try:

                    vm = VirtualMachine.objects.get(pk=id)

                    with pooled_rapi_client(vm) as client:

                        client.DeleteInstance(utils.id_to_instance_name(id))

                except VirtualMachine.DoesNotExist:

                    print >> sys.stderr, "No entry for VM %d in DB !!" % id

            print >> sys.stderr, "    ...done"

        if options['fix_unsynced'] and len(unsynced) > 0:

            print >> sys.stderr, "Setting the state of %d out-of-sync VMs:" % \

                len(unsynced)

            for id, db_state, ganeti_up in unsynced:

                vm = VirtualMachine.objects.get(pk=id)

                opcode = "OP_INSTANCE_REBOOT" if ganeti_up \

                         else "OP_INSTANCE_SHUTDOWN"

                event_time = datetime.datetime.now()

                backend_mod.process_op_status(

                    vm=vm, etime=event_time, jobid=-0,

                    opcode=opcode, status='success',

                    logmsg='Reconciliation: simulated Ganeti event')

            print >> sys.stderr, "    ...done"

        if options['fix_build_errors'] and len(build_errors) > 0:

            print >> sys.stderr, "Setting the state of %d build-errors VMs:" %\

                                 len(build_errors)

            for id in build_errors:

                vm = VirtualMachine.objects.get(pk=id)

                event_time = datetime.datetime.now()

                backend_mod.process_op_status(

                    vm=vm, etime=event_time, jobid=-0,

                    opcode="OP_INSTANCE_CREATE", status='error',

                    logmsg='Reconciliation: simulated Ganeti event')

            print >> sys.stderr, "    ...done"

        if options['fix_unsynced_nics'] and len(unsynced_nics) > 0:

            print >> sys.stderr, "Setting the nics of %d out-of-sync VMs:" % \

                                 len(unsynced_nics)

            for id, nics in unsynced_nics.items():

                vm = VirtualMachine.objects.get(pk=id)

                nics = nics[1]  # Ganeti nics

                if nics == {}:  # No nics

                    vm.nics.all.delete()

                    continue

                for index, nic in nics.items():

                    net_id = utils.id_from_network_name(nic['network'])

                    subnet6 = Network.objects.get(id=net_id).subnet6

                    # Produce ipv6

                    ipv6 = subnet6 and mac2eui64(nic['mac'], subnet6) or None

                    nic['ipv6'] = ipv6

                    # Rename ipv4 to ip

                    nic['ip'] = nic['ipv4']

                # Dict to sorted list

                final_nics = []

                nics_keys = nics.keys()

                nics_keys.sort()

                for i in nics_keys:

                    if nics[i]['network']:

                        final_nics.append(nics[i])

                    else:

                        print 'Network of nic %d of vm %s is None. ' \

                              'Can not reconcile' % (i, vm.backend_vm_id)

                event_time = datetime.datetime.now()

                backend_mod.process_net_status(vm=vm, etime=event_time,

                                               nics=final_nics)

            print >> sys.stderr, "    ...done"

        if options["fix_unsynced_flavors"] and len(unsynced_flavors) > 0:

            print >> sys.stderr, "Setting the flavor of %d unsynced VMs:" % \

                len(unsynced_flavors)

            for id, db_flavor, gnt_flavor in unsynced_flavors:

                vm = VirtualMachine.objects.get(pk=id)

                old_state = vm.operstate

                opcode = "OP_INSTANCE_SET_PARAMS"

                beparams = {"vcpus": gnt_flavor.cpu,

                            "minmem": gnt_flavor.ram,

                            "maxmem": gnt_flavor.ram}

                event_time = datetime.datetime.now()

                backend_mod.process_op_status(

                    vm=vm, etime=event_time, jobid=-0,

                    opcode=opcode, status='success',

                    beparams=beparams,

                    logmsg='Reconciliation: simulated Ganeti event')

                # process_op_status with beparams will set the vmstate to

                # shutdown. Fix this be returning it to old state

                vm = VirtualMachine.objects.get(pk=id)

                vm.operstate = old_state

                vm.save()

            print >> sys.stderr, "    ...done"