def on_delete(self, req, resp, address):
"""
Handles the Deletion of a Host.
:param req: Request instance that will be passed through.
:type req: falcon.Request
:param resp: Response instance that will be passed through.
:type resp: falcon.Response
:param address: The address of the Host being requested.
:type address: str
"""
resp.body = '{}'
try:
self.store.delete(util.etcd_host_key(address))
resp.status = falcon.HTTP_200
except etcd.EtcdKeyNotFound:
resp.status = falcon.HTTP_404
# Also remove the host from all clusters.
# Note: We've done all we need to for the host deletion,
# so if an error occurs from here just log it and
# return.
try:
clusters_dir = self.store.get('/commissaire/clusters')
self.logger.debug('Etcd Response: {0}'.format(clusters_dir))
except etcd.EtcdKeyNotFound:
self.logger.warn('Etcd does not have any clusters')
return
if len(clusters_dir._children):
self.logger.info(
'There are clusters associated with {0}...'.format(address))
for etcd_resp in clusters_dir.leaves:
cluster = Cluster(**json.loads(etcd_resp.value))
if address in cluster.hostset:
cluster_name = etcd_resp.key.split('/')[-1]
self.logger.info('Removing {0} from cluster {1}'.format(
address, cluster_name))
cluster.hostset.remove(address)
self.store.set(etcd_resp.key, cluster.to_json(secure=True))
self.logger.info(
'{0} has been removed from cluster {1}'.format(
address, cluster_name))
def on_delete(self, req, resp, address):
"""
Handles the Deletion of a Host.
:param req: Request instance that will be passed through.
:type req: falcon.Request
:param resp: Response instance that will be passed through.
:type resp: falcon.Response
:param address: The address of the Host being requested.
:type address: str
"""
resp.body = '{}'
try:
self.store.delete(util.etcd_host_key(address))
resp.status = falcon.HTTP_200
except etcd.EtcdKeyNotFound:
resp.status = falcon.HTTP_404
# Also remove the host from all clusters.
# Note: We've done all we need to for the host deletion,
# so if an error occurs from here just log it and
# return.
try:
clusters_dir = self.store.get('/commissaire/clusters')
self.logger.debug('Etcd Response: {0}'.format(clusters_dir))
except etcd.EtcdKeyNotFound:
self.logger.warn('Etcd does not have any clusters')
return
if len(clusters_dir._children):
self.logger.info(
'There are clusters associated with {0}...'.format(address))
for etcd_resp in clusters_dir.leaves:
cluster = Cluster(**json.loads(etcd_resp.value))
if address in cluster.hostset:
cluster_name = etcd_resp.key.split('/')[-1]
self.logger.info('Removing {0} from cluster {1}'.format(
address, cluster_name))
cluster.hostset.remove(address)
self.store.set(etcd_resp.key, cluster.to_json(secure=True))
self.logger.info(
'{0} has been removed from cluster {1}'.format(
address, cluster_name))
def on_get(self, req, resp, address):
"""
Handles retrieval of an existing Host.
:param req: Request instance that will be passed through.
:type req: falcon.Request
:param resp: Response instance that will be passed through.
:type resp: falcon.Response
:param address: The address of the Host being requested.
:type address: str
"""
# TODO: Verify input
try:
etcd_resp = self.store.get(util.etcd_host_key(address))
self.logger.debug('Etcd Response: {0}'.format(etcd_resp))
except etcd.EtcdKeyNotFound:
resp.status = falcon.HTTP_404
return
resp.status = falcon.HTTP_200
req.context['model'] = Host(**json.loads(etcd_resp.value))
def on_get(self, req, resp, address):
"""
Handles retrieval of an existing Host.
:param req: Request instance that will be passed through.
:type req: falcon.Request
:param resp: Response instance that will be passed through.
:type resp: falcon.Response
:param address: The address of the Host being requested.
:type address: str
"""
# TODO: Verify input
etcd_resp, error = cherrypy.engine.publish(
'store-get', util.etcd_host_key(address))[0]
self.logger.debug('Etcd Response: {0}'.format(etcd_resp))
if error:
resp.status = falcon.HTTP_404
return
resp.status = falcon.HTTP_200
req.context['model'] = Host(**json.loads(etcd_resp.value))
def on_get(self, req, resp, address):
"""
Handles retrieval of an existing Host.
:param req: Request instance that will be passed through.
:type req: falcon.Request
:param resp: Response instance that will be passed through.
:type resp: falcon.Response
:param address: The address of the Host being requested.
:type address: str
"""
# TODO: Verify input
etcd_resp, error = cherrypy.engine.publish(
'store-get', util.etcd_host_key(address))[0]
self.logger.debug('Etcd Response: {0}'.format(etcd_resp))
if error:
resp.status = falcon.HTTP_404
return
resp.status = falcon.HTTP_200
req.context['model'] = Host(**json.loads(etcd_resp.value))
def on_put(self, req, resp, address):
"""
Handles the creation of a new Host.
:param req: Request instance that will be passed through.
:type req: falcon.Request
:param resp: Response instance that will be passed through.
:type resp: falcon.Response
:param address: The address of the Host being requested.
:type address: str
"""
try:
# Extract what we need from the input data.
# Don't treat it as a skeletal host record.
req_data = req.stream.read()
req_body = json.loads(req_data.decode())
ssh_priv_key = req_body['ssh_priv_key']
# Cluster member is optional.
cluster_name = req_body.get('cluster', None)
except (KeyError, ValueError):
self.logger.info(
'Bad client PUT request for host {0}: {1}'.
format(address, req_data))
resp.status = falcon.HTTP_400
return
key = util.etcd_host_key(address)
try:
etcd_resp = self.store.get(key)
self.logger.debug('Etcd Response: {0}'.format(etcd_resp))
# Check if the request conflicts with the existing host.
existing_host = Host(**json.loads(etcd_resp.value))
if existing_host.ssh_priv_key != ssh_priv_key:
resp.status = falcon.HTTP_409
return
if cluster_name:
try:
assert util.etcd_cluster_has_host(
self.store, cluster_name, address)
except (AssertionError, KeyError):
resp.status = falcon.HTTP_409
return
# Request is compatible with the existing host, so
# we're done. (Not using HTTP_201 since we didn't
# actually create anything.)
resp.status = falcon.HTTP_200
req.context['model'] = existing_host
return
except etcd.EtcdKeyNotFound:
pass
host_creation = {
'address': address,
'ssh_priv_key': ssh_priv_key,
'os': '',
'status': 'investigating',
'cpus': -1,
'memory': -1,
'space': -1,
'last_check': None
}
# Verify the cluster exists, if given. Do it now
# so we can fail before writing anything to etcd.
if cluster_name:
if not util.etcd_cluster_exists(self.store, cluster_name):
resp.status = falcon.HTTP_409
return
host = Host(**host_creation)
new_host = self.store.set(key, host.to_json(secure=True))
INVESTIGATE_QUEUE.put((host_creation, ssh_priv_key))
# Add host to the requested cluster.
if cluster_name:
util.etcd_cluster_add_host(self.store, cluster_name, address)
resp.status = falcon.HTTP_201
req.context['model'] = Host(**json.loads(new_host.value))