Synnefo » snf-ganeti

  def _UnlockedAddDisk(self, disk):

    """Add a disk to the config.

    This function is for internal use, when the config lock is already held.

    The `_WriteConfig' function has to be called afterwards (so we can

    add many disks at once).

    @type disk: L{objects.Disk}

    @param disk: the disk object

    """

    if not isinstance(disk, objects.Disk):

      raise errors.ProgrammerError("Invalid type passed to _UnlockedAddDisk")

    logging.info("Adding disk %s to configuration", disk.uuid)

    self._CheckUniqueUUID(disk, include_temporary=False)

    disk.serial_no = 1

    disk.ctime = disk.mtime = time.time()

    self._config_data.disks[disk.uuid] = disk

    self._config_data.cluster.serial_no += 1

  # pylint: disable=R0201

  def _UpdateIvNames(self, base_index, disks):

    """Update the C{iv_name} attribute of disks.

    @type disks: list of L{objects.Disk}

    """

    for (idx, disk) in enumerate(disks):

      disk.iv_name = "disk/%s" % (base_index + idx)

  def _UnlockedAttachInstDisk(self, instance, disk, idx=None):

    """Attach a disk to an instance.

    This function is for internal use, when the config lock is already held.

    The `_WriteConfig' function has to be called afterwards (so we can

    add a disk to config and attach it to an instance at once).

    @type instance: L{objects.Instance}

    @param instance: the instance object

    @type disk: L{objects.Disk}

    @param disk: the disk object

    @type idx: int

    @param idx: the index of the newly attached disk

    """

    if not isinstance(instance, objects.Instance) or \

        not isinstance(disk, objects.Disk):

      raise errors.ProgrammerError(

        "Invalid type passed to _UnlockedAttachInstDisk")

    if idx is None:

      idx = len(instance.disks)

    else:

      if idx < 0:

        raise IndexError("Not accepting negative indices other than -1")

      elif idx > len(instance.disks):

        raise IndexError("Got disk index %s, but there are only %s" %

                         (idx, len(instance.disks)))

    # Disk must not be attached anywhere else

    if disk.instance:

      raise errors.ReservationError("Disk %s already attached to instance %s"

                                    % (disk.uuid, instance.name))

    disk.instance = instance.uuid

    disk.serial_no += 1

    disk.mtime = time.time()

    instance.disks.insert(idx, disk.uuid)

    inst_disks = self._UnlockedGetInstanceDisks(instance)

    self._UpdateIvNames(idx, inst_disks[idx:])

    instance.serial_no += 1

    instance.mtime = time.time()

  @locking.ssynchronized(_config_lock)

  def AddInstDisk(self, instance, disk, idx=None):

    """Add a disk to the config and attach it to instance.

    @type instance: L{objects.Instance}

    @param instance: the instance object

    @type disk: L{objects.Disk}

    @param disk: the disk object

    @type idx: int

    @param idx: the index of the newly attached disk

    """

    self._UnlockedAddDisk(disk)

    self._UnlockedAttachInstDisk(instance, disk, idx)

    self._WriteConfig()

    elif isinstance(target, objects.Disk):

      test = target in self._config_data.disks.values()