def update(container_id):
status_in_db = itf_database.get_status(container_id)
assert status_in_db != None
status_remote = get_container_status(status_in_db.StatusPath)
if status_remote == None:
print '[Container Monitor] Container', container_id, 'not available.'
# TODO: update database and reschedule
else:
assert status_remote.ContainerId == container_id
# Avoid overwriting the reservable size (container doesn't know this)
status_remote.StorageReserved = status_in_db.StorageReserved
itf_database.update_container(status_remote)
# call QoS Monitor
print '[Container Monitor] Call QoS Monitor for', container_id
monitor_qos.monitor_container(container_id)
def insert(status_path):
status_remote = get_container_status(status_path)
if status_remote is not None:
container_id = status_remote.ContainerId
status_remote.StatusPath = status_path
# Initial reserved size = used size
status_remote.StorageReserved = status_remote.StorageUsed
added = itf_database.update_container(status_remote, init=True)
else:
print "[Container Monitor] Status path " + status_path + \
" is not available"
def process_request(request):
cmd = request.cmd
data = request.data
print '[QoS Manager] Incoming request: ' + cmd + ' from ' + \
('cmdline' if request.from_cmdline else 'socket')
if data != '':
print 'Data:\n--------------------'
print data + '\n--------------------'
if cmd == '-schedule':
spec = itf_spec.QosSpec()
err = spec.parse_string(data)
if not err:
spec_db = itf_database.get_spec(spec.SpecId)
if spec_db == None: # spec does not exist
print '[QoS Manager] Schedule for new spec {' + spec.SpecId + '}'
scheduled_containers, data = qos_scheduler.schedule(spec, task='new')
if len(scheduled_containers) > 0:
itf_database.add_scheduled_spec(spec, scheduled_containers, init=True)
print '[QoS Manager] Successfully scheduled {' + spec.SpecId + '}'
else:
print '[QoS Manager] Fail to schedule new spec {' \
+ spec.SpecId + '} {' + data + '}'
exit(-1)
else: # reschedule for updated spec
print '[QoS Manager] User specification already exists'
print '[QoS Manager] Reschedule for spec {' + spec.SpecId + '}'
scheduled_containers, data = qos_scheduler.schedule(spec, task='update')
if len(scheduled_containers) > 0:
itf_database.add_scheduled_spec(spec, scheduled_containers)
print '[QoS Manager] Successfully rescheduled {' + spec.SpecId + '}'
else:
print '[QoS Manager] Fail to reschedule spec {' \
+ spec.SpecId + '} {' + data + '}'
exit(-1)
else:
print '[QoS Manager] Error: Invalid QoS spec.'
elif cmd == '-rm_spec':
spec_id = data
print '[QoS Manager] Remove spec {' + spec_id + '} from database'
itf_database.remove_spec(spec_id)
elif cmd == '-add_container':
# data is the path to the container status file
print '[QoS Manager] Add a new container with status: ' + data
# this request is from system admin when creating new container
monitor_container.insert(data) # insert status to db
status = itf_database.get_status(data)
print '[QoS Manager] Successfully adds new container ' + data
elif cmd == '-rm_container':
container_id = data
print '[QoS Manager] Remove container id: ' + container_id
# Set container availability to 0 for triggering reschedule
status = itf_database.get_status(container_id)
if status is not None:
old_avail = status.ContainerAvailability
status.ContainerAvailability = -1
itf_database.update_container(status)
# Reschedule every related spec - finally no spec will be on container
related_specs = itf_database.get_spec_ids_on_container(container_id)
for spec_id in related_specs:
print '[QoS Manager] Reschedule for spec {' + spec_id + '}'
spec = itf_database.get_spec(spec_id)
scheduled_containers, data = qos_scheduler.schedule(spec, task='new')
if len(scheduled_containers) > 0:
itf_database.add_scheduled_spec(spec, scheduled_containers)
print '[QoS Manager] Successfully rescheduled {' + spec_id + '}'
else:
print '[QoS Manager] Fail to reschedule spec {' \
+ spec_id + '} {' + data + '}'
status.ContainerAvailability = old_avail
itf_database.update_container(status)
exit(-1)
# Delete container in database
# NOTE: Must after finishing the file copy
succ = itf_database.remove_container(container_id)
if succ:
print '[QoS Manager] Container', container_id, 'is removed from database'
else:
print '[QoS Manager] Container', container_id, 'is not in database'
elif cmd == '-show_db':
print '[QoS Manager] QoS database summary:'
itf_database.summary(verbose=False)
elif cmd == '-show_db_verbose':
print '[QoS Manager] QoS database summary:'
itf_database.summary(verbose=True)
elif cmd == '-destroy_db':
print '[QoS Manager] Clearing all contents in the QoS database?'
prompt = raw_input('[y/n]:')
if prompt == 'y' or prompt == 'Y':
itf_database.init()
elif cmd == '-start_server':
if request.from_cmdline:
qos_server_main()
elif cmd == '-stop_server':
if request.from_cmdline:
# send a stop request to server
request = QosRequest()
request.cmd = '-stop_server'
client_send(request)
else:
return 1 # use this value for stop the server main loop
else:
print '[QoS Manager] Unsupported QoS request command: ', cmd
return -1
return 0
