from flask_restful import Resource
from flask import request
import json
+from copy import deepcopy
logging.basicConfig(level=logging.INFO)
+CORS_HEADER = {'Access-Control-Allow-Origin': '*'}
+
dcs = {}
global dcs
def put(self, dc_label, compute_name, resource=None, value=None):
- # check if resource update
- if resource and value:
- c = self._update_resource(dc_label, compute_name, resource, value)
- return c.getStatus(), 200
# deploy new container
# check if json data is a dict
c = dcs.get(dc_label).startCompute(
compute_name, image=image, command=command, network=nw_list)
# return docker inspect dict
- return c.getStatus(), 200
+ return c.getStatus(), 200, CORS_HEADER
except Exception as ex:
logging.exception("API error.")
- return ex.message, 500
-
- def _update_resource(self, dc_label, compute_name, resource, value):
- #check if container exists
- d = dcs.get(dc_label).net.getNodeByName(compute_name)
- if resource == 'cpu':
- cpu_period = int(dcs.get(dc_label).net.cpu_period)
- cpu_quota = int(cpu_period * float(value))
- #put default values back
- if float(value) <= 0:
- cpu_period = 100000
- cpu_quota = -1
- d.updateCpuLimit(cpu_period=cpu_period, cpu_quota=cpu_quota)
- return d
-
+ return ex.message, 500, CORS_HEADER
def get(self, dc_label, compute_name):
logging.debug("API CALL: compute status")
try:
- return dcs.get(dc_label).containers.get(compute_name).getStatus(), 200
+ return dcs.get(dc_label).containers.get(compute_name).getStatus(), 200, CORS_HEADER
except Exception as ex:
logging.exception("API error.")
- return ex.message, 500
+ return ex.message, 500, CORS_HEADER
def delete(self, dc_label, compute_name):
logging.debug("API CALL: compute stop")
try:
- return dcs.get(dc_label).stopCompute(compute_name), 200
+ return dcs.get(dc_label).stopCompute(compute_name), 200, CORS_HEADER
except Exception as ex:
logging.exception("API error.")
- return ex.message, 500
+ return ex.message, 500, CORS_HEADER
def _parse_network(self, network_str):
'''
all_containers = []
for dc in dcs.itervalues():
all_containers += dc.listCompute()
- return [(c.name, c.getStatus()) for c in all_containers], 200
+ return [(c.name, c.getStatus()) for c in all_containers], 200, CORS_HEADER
else:
# return list of compute nodes for specified DC
return [(c.name, c.getStatus())
- for c in dcs.get(dc_label).listCompute()], 200
+ for c in dcs.get(dc_label).listCompute()], 200, CORS_HEADER
except Exception as ex:
logging.exception("API error.")
- return ex.message, 500
+ return ex.message, 500, CORS_HEADER
+
+class ComputeResources(Resource):
+ """
+ Update the container's resources using the docker.update function
+ re-using the same parameters:
+ url params:
+ blkio_weight
+ cpu_period, cpu_quota, cpu_shares
+ cpuset_cpus
+ cpuset_mems
+ mem_limit
+ mem_reservation
+ memswap_limit
+ kernel_memory
+ restart_policy
+ see https://docs.docker.com/engine/reference/commandline/update/
+ or API docs: https://docker-py.readthedocs.io/en/stable/api.html#module-docker.api.container
+ :param dc_label: name of the DC
+ :param compute_name: compute container name
+
+ :return: docker inspect dict of deployed docker
+ """
+ global dcs
+ def put(self, dc_label, compute_name):
+ logging.debug("REST CALL: update container resources")
+
+ try:
+ c = self._update_resources(dc_label, compute_name)
+ return c.getStatus(), 200, CORS_HEADER
+ except Exception as ex:
+ logging.exception("API error.")
+ return ex.message, 500, CORS_HEADER
+
+ def _update_resources(self, dc_label, compute_name):
+
+ # get URL parameters
+ params = request.args
+ # then no data
+ if params is None:
+ params = {}
+ logging.debug("REST CALL: update container resources {0}".format(params))
+ #check if container exists
+ d = dcs.get(dc_label).net.getNodeByName(compute_name)
+
+ # general request of cpu percentage
+ # create a mutable copy
+ params = params.to_dict()
+ if 'cpu_bw' in params:
+ cpu_period = int(dcs.get(dc_label).net.cpu_period)
+ value = params.get('cpu_bw')
+ cpu_quota = int(cpu_period * float(value))
+ #put default values back
+ if float(value) <= 0:
+ cpu_period = 100000
+ cpu_quota = -1
+ params['cpu_period'] = cpu_period
+ params['cpu_quota'] = cpu_quota
+ #d.updateCpuLimit(cpu_period=cpu_period, cpu_quota=cpu_quota)
+
+ # only pass allowed keys to docker
+ allowed_keys = ['blkio_weight', 'cpu_period', 'cpu_quota', 'cpu_shares', 'cpuset_cpus',
+ 'cpuset_mems', 'mem_limit', 'mem_reservation', 'memswap_limit',
+ 'kernel_memory', 'restart_policy']
+ filtered_params = {key:params[key] for key in allowed_keys if key in params}
+
+ d.update_resources(**filtered_params)
+
+ return d
class DatacenterList(Resource):
global dcs
def get(self):
logging.debug("API CALL: datacenter list")
try:
- return [d.getStatus() for d in dcs.itervalues()], 200
+ return [d.getStatus() for d in dcs.itervalues()], 200, CORS_HEADER
except Exception as ex:
logging.exception("API error.")
- return ex.message, 500
+ return ex.message, 500, CORS_HEADER
class DatacenterStatus(Resource):
def get(self, dc_label):
logging.debug("API CALL: datacenter status")
try:
- return dcs.get(dc_label).getStatus(), 200
+ return dcs.get(dc_label).getStatus(), 200, CORS_HEADER
except Exception as ex:
logging.exception("API error.")
- return ex.message, 500
+ return ex.message, 500, CORS_HEADER