def get_disk_from_container(cid):
props = JBoxVol.dckr().inspect_container(cid)
vols = props['Volumes']
for _cpath, hpath in vols.iteritems():
if hpath.startswith(JBoxLoopbackVol.FS_LOC):
return JBoxLoopbackVol.get_disk_from_container(cid)
elif VolMgr.HAS_EBS and hpath.startswith(JBoxEBSVol.FS_LOC):
return JBoxEBSVol.get_disk_from_container(cid)
return None
def get_disk_from_container(cid):
props = JBoxVol.dckr().inspect_container(cid)
vols = props['Volumes']
for _cpath, hpath in vols.iteritems():
if hpath.startswith(JBoxLoopbackVol.FS_LOC):
return JBoxLoopbackVol.get_disk_from_container(cid)
elif VolMgr.HAS_EBS and hpath.startswith(JBoxEBSVol.FS_LOC):
return JBoxEBSVol.get_disk_from_container(cid)
return None
def do_task(plugin_type, data):
if plugin_type != JBPluginTask.JBP_CMD_ASYNC:
return
mode = data['action']
user_id = data['user_id']
sessname = data['sessname']
user = JBoxUserV2(user_id)
is_allowed = user.has_resource_profile(
JBoxUserV2.RES_PROF_DISK_EBS_10G)
if not is_allowed:
JBoxEBSVolAsyncTask.log_error(
"Data volume access not allowed for user")
return
cont = SessContainer.get_by_name(sessname)
if cont is None:
return
vol = JBoxEBSVol.get_disk_from_container(sessname)
disk_state = None
try:
disk_state = JBoxDiskState(cluster_id=CompEC2.INSTALL_ID,
region_id=CompEC2.REGION,
user_id=user_id)
except:
pass
JBoxEBSVolAsyncTask.log_debug("Data volume request %s for %s", mode,
cont.debug_str())
if mode == 'attach':
if vol is None:
vol = JBoxEBSVol.get_disk_for_user(user_id)
JBoxEBSVol.mount_host_device(vol.disk_path, cont.dockid,
JBoxVol.DATA_MOUNT_POINT)
disk_state = JBoxDiskState(cluster_id=CompEC2.INSTALL_ID,
region_id=CompEC2.REGION,
user_id=user_id)
if disk_state.get_state() != JBoxDiskState.STATE_ATTACHED:
disk_state.set_state(JBoxDiskState.STATE_ATTACHED)
disk_state.save()
elif mode == 'detach':
if cont is not None and cont.is_running():
if vol is not None:
# unmount from container first
JBoxEBSVol.unmount_host_device(vol.disk_path, cont.dockid)
elif disk_state is not None:
# no volume attached. ensure disk state is updated
if disk_state.get_state() != JBoxDiskState.STATE_DETACHED:
disk_state.set_state(JBoxDiskState.STATE_DETACHED)
disk_state.save()
if vol is not None:
vol.release(backup=True)
JBoxEBSVolAsyncTask.log_debug(
"Data volume request %s completed for %s", mode, cont.debug_str())
def get_disk_from_container(cid):
props = JBoxVol.dckr().inspect_container(cid)
try:
vols = props['Volumes']
for _cpath, hpath in vols.iteritems():
if hpath.startswith(JBoxLoopbackVol.FS_LOC):
return JBoxLoopbackVol.get_disk_from_container(cid)
elif VolMgr.HAS_EBS and hpath.startswith(JBoxEBSVol.FS_LOC):
return JBoxEBSVol.get_disk_from_container(cid)
except:
VolMgr.log_error("error finding disk ids used in " + cid)
return None
def get_disk_from_container(cid):
props = JBoxVol.dckr().inspect_container(cid)
try:
vols = props['Volumes']
for _cpath, hpath in vols.iteritems():
if hpath.startswith(JBoxLoopbackVol.FS_LOC):
return JBoxLoopbackVol.get_disk_from_container(cid)
elif VolMgr.HAS_EBS and hpath.startswith(JBoxEBSVol.FS_LOC):
return JBoxEBSVol.get_disk_from_container(cid)
except:
VolMgr.log_error("error finding disk ids used in " + cid)
return None
def do_task(plugin_type, data):
if plugin_type != JBPluginTask.JBP_CMD_ASYNC:
return
mode = data["action"]
user_id = data["user_id"]
sessname = data["sessname"]
user = JBoxUserV2(user_id)
is_allowed = user.has_resource_profile(JBoxUserV2.RES_PROF_DISK_EBS_10G)
if not is_allowed:
JBoxEBSVolAsyncTask.log_error("Data volume access not allowed for user")
return
cont = SessContainer.get_by_name(sessname)
if cont is None:
return
vol = JBoxEBSVol.get_disk_from_container(sessname)
disk_state = None
try:
disk_state = JBoxDiskState(cluster_id=CompEC2.INSTALL_ID, region_id=CompEC2.REGION, user_id=user_id)
except:
pass
JBoxEBSVolAsyncTask.log_debug("Data volume request %s for %s", mode, cont.debug_str())
if mode == "attach":
if vol is None:
vol = JBoxEBSVol.get_disk_for_user(user_id)
JBoxEBSVol.mount_host_device(vol.disk_path, cont.dockid, JBoxVol.DATA_MOUNT_POINT)
disk_state = JBoxDiskState(cluster_id=CompEC2.INSTALL_ID, region_id=CompEC2.REGION, user_id=user_id)
if disk_state.get_state() != JBoxDiskState.STATE_ATTACHED:
disk_state.set_state(JBoxDiskState.STATE_ATTACHED)
disk_state.save()
elif mode == "detach":
if cont is not None and cont.is_running():
if vol is not None:
# unmount from container first
JBoxEBSVol.unmount_host_device(vol.disk_path, cont.dockid)
elif disk_state is not None:
# no volume attached. ensure disk state is updated
if disk_state.get_state() != JBoxDiskState.STATE_DETACHED:
disk_state.set_state(JBoxDiskState.STATE_DETACHED)
disk_state.save()
if vol is not None:
vol.release(backup=True)
JBoxEBSVolAsyncTask.log_debug("Data volume request %s completed for %s", mode, cont.debug_str())
def _get_state(self, sessname, user_id):
vol = JBoxEBSVol.get_disk_from_container(sessname)
state_code = JBoxDiskState.STATE_DETACHED
try:
disk_state = JBoxDiskState(cluster_id=CompEC2.INSTALL_ID, region_id=CompEC2.REGION, user_id=user_id)
state_code = disk_state.get_state()
except:
pass
if ((state_code == JBoxDiskState.STATE_ATTACHED) and (vol is None)) or (
(state_code == JBoxDiskState.STATE_DETACHED) and (vol is not None)
):
state_code = -1
self.log_debug("EBS disk state: %r", state_code)
return {"disk_size": "10 GB", "state": state_code}
def _get_state(self, sessname, user_id):
vol = JBoxEBSVol.get_disk_from_container(sessname)
state_code = JBoxDiskState.STATE_DETACHED
try:
disk_state = JBoxDiskState(cluster_id=CompEC2.INSTALL_ID,
region_id=CompEC2.REGION,
user_id=user_id)
state_code = disk_state.get_state()
except:
pass
if ((state_code == JBoxDiskState.STATE_ATTACHED) and (vol is None)) or \
((state_code == JBoxDiskState.STATE_DETACHED) and (vol is not None)):
state_code = -1
self.log_debug("EBS disk state: %r", state_code)
return {'disk_size': '10 GB', 'state': state_code}
