def _ensure_volume_available(vol_id, force_detach=False): conn = EBSVol._ec2() vol = EBSVol.get_volume(vol_id) if vol is None: raise Exception("Volume not found: " + vol_id) if CompEC2._state_check(vol, 'available'): return True # volume may be attached instance_id = CompEC2.get_instance_id() att_instance_id, att_device = EBSVol._get_volume_attach_info(vol_id) if (att_instance_id is None) or (att_instance_id == instance_id): return True if force_detach: EBSVol.log_warn("Forcing detach of volume %s", vol_id) conn.detach_volume(vol_id) CompEC2._wait_for_status(vol, 'available') if not CompEC2._state_check(vol, 'available'): raise Exception("Volume not available: " + vol_id + ", attached to: " + att_instance_id + ", state: " + vol.status)
def create_new_volume(snap_id, dev_id, tag=None, disk_sz_gb=1): EBSVol.configure_host_commands() EBSVol.log_info("Creating volume. Tag: %s, Snapshot: %s. Attached: %s", tag, snap_id, dev_id) conn = EBSVol._ec2() vol = conn.create_volume(disk_sz_gb, CompEC2._zone(), snapshot=snap_id, volume_type='gp2') # volume_type='io1', # iops=30*disk_sz_gb) CompEC2._wait_for_status(vol, 'available') vol_id = vol.id EBSVol.log_info("Created volume with id %s", vol_id) if tag is not None: conn.create_tags([vol_id], {"Name": tag}) EBSVol.log_info("Added tag %s to volume with id %s", tag, vol_id) device_path = EBSVol._attach_free_volume(vol_id, dev_id) return device_path, vol_id
def _attach_free_volume(vol_id, dev_id): conn = EBSVol._ec2() instance_id = CompEC2.get_instance_id() device = os.path.join('/dev', dev_id) vol = EBSVol.get_volume(vol_id) EBSVol.log_info("Attaching volume %s at %s", vol_id, device) t1 = time.time() conn.attach_volume(vol_id, instance_id, device) if not CompEC2._wait_for_status(vol, 'in-use'): EBSVol.log_error("Could not attach volume %s", vol_id) raise Exception("Volume could not be attached. Volume id: " + vol_id) if not EBSVol._wait_for_device(device): EBSVol.log_error("Could not attach volume %s to device %s", vol_id, device) raise Exception("Volume could not be attached. Volume id: " + vol_id + ", device: " + device) tdiff = int(time.time() - t1) CompEC2.publish_stats("EBSAttachTime", "Count", tdiff) return device