def __init__(self, db):
self._db = db
self._worker = BobbyWorker(self._db)
def __init__(self, db):
self._db = db
self._worker = BobbyWorker(self._db)
class Bobby(object):
"""Bobby app views."""
app = Klein()
def __init__(self, db):
self._db = db
self._worker = BobbyWorker(self._db)
@app.route('/<string:tenant_id>/groups', methods=['POST'])
@with_transaction_id()
def create_group(self, request, log, tenant_id):
"""Create a new group.
Receive application/json content for new group creation.
:param str tenant_id: A tenant id
"""
content = json.loads(request.content.read())
group_id = content.get('groupId')
d = self._worker.create_group(tenant_id, group_id)
def _serialize_object(group):
json_object = {
'groupId':
group['groupId'],
'links': [{
'href':
'{0}{1}'.format(request.URLPath().path, group['groupId']),
'rel':
'self'
}],
'notification':
group['notification'],
'notificationPlan':
group['notificationPlan'],
'tenantId':
group['tenantId']
}
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(201)
request.write(json.dumps(json_object))
request.finish()
return d.addCallback(_serialize_object)
@app.route('/<string:tenant_id>/groups/<string:group_id>',
methods=['DELETE'])
@with_transaction_id()
def delete_group(self, request, log, tenant_id, group_id):
"""Delete a group.
:param str tenant_id: A tenant id
:param str group_id: A groud id
"""
d = self._worker.delete_group(tenant_id, group_id)
def finish(_):
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(204)
request.finish()
return d.addCallback(finish)
@app.route('/<string:tenant_id>/groups/<string:group_id>/servers',
methods=['POST'])
@with_transaction_id()
def create_server(self, request, log, tenant_id, group_id):
"""Create a new server.
Receive application/json content for new server creation.
:param request: Twisted IRequest object.
:param log: A log object.
:param str tenant_id: A tenant id
:param str group_id: A group id
"""
# The server object is one provided via the nova API.
content = json.loads(request.content.read())
server = content.get('server')
d = self._worker.create_server(tenant_id, group_id, server)
def serialize(server):
json_object = {
'entityId':
server['entityId'],
'groupId':
server['groupId'],
'links': [{
'href':
'{0}{1}'.format(request.URLPath().path,
server['serverId']),
'rel':
'self'
}],
'serverId':
server['serverId']
}
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(201)
request.write(json.dumps(json_object))
request.finish()
return d.addCallback(serialize)
@app.route(
'/<string:tenant_id>/groups/<string:group_id>/servers/<string:server_id>',
methods=['DELETE'])
@with_transaction_id()
def delete_server(self, request, log, tenant_id, group_id, server_id):
"""Delete a server.
:param str tenant_id: A tenant id
:param str group_id: A groud id
:param str server_id: A server id
"""
d = self._worker.delete_server(tenant_id, group_id, server_id)
def finish(_):
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(204)
request.finish()
return d.addCallback(finish)
@app.route('/<string:tenant_id>/groups/<string:group_id>/policies',
methods=['POST'])
@with_transaction_id()
def create_policy(self, request, log, tenant_id, group_id):
"""Create a new policy.
Receive application/json content for new policy creation.
:param str tenant_id: A tenant id
:param str group_id: A group id
"""
content = json.loads(request.content.read())
alarm_template_id = content.get('alarmTemplate')
check_template_id = content.get('checkTemplate')
policy_id = content.get('policyId')
d = cass.create_policy(self._db, policy_id, group_id,
alarm_template_id, check_template_id)
# Trigger actions to create the alarm and checks on the MaaS side and set things up
def serialize(policy):
json_object = {
'alarmTemplate':
policy['alarmTemplate'],
'checkTemplate':
policy['checkTemplate'],
'groupId':
policy['groupId'],
'links': [{
'href':
'{0}{1}'.format(request.URLPath().path,
policy['policyId']),
'rel':
'self'
}],
'policyId':
policy['policyId']
}
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(201)
request.write(json.dumps(json_object))
request.finish()
return d.addCallback(serialize)
@app.route(
'/<string:tenant_id>/groups/<string:group_id>/policies/<string:policy_id>',
methods=['DELETE'])
@with_transaction_id()
def delete_policy(self, request, log, tenant_id, group_id, policy_id):
"""Delete a policy.
:param str tenant_id: A tenant id
:param str group_id: A groud id
:param str policy_id: A policy id
"""
d = cass.delete_policy(self._db, group_id, policy_id)
def finish(_):
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(204)
request.finish()
return d.addCallback(finish)
@app.route('/alarm', methods=['POST'])
@with_transaction_id()
def alarm(self, request, log):
"""Change the state of an alarm."""
_policy_id = []
content = json.loads(request.content.read())
alarm_id = content.get('alarm').get('id')
status = content.get('details').get('state')
d = cass.alter_alarm_state(self._db, alarm_id, status)
def check_quorum_health((policy_id, server_id)):
_policy_id.append(policy_id)
return cass.check_quorum_health(self._db, policy_id)
d.addCallback(check_quorum_health)
def maybe_execute_policy(health):
if health:
return self._worker.execute_policy(_policy_id[0])
else:
defer.succeed(None)
d.addCallback(maybe_execute_policy)
def finish(_):
request.setResponseCode(200)
request.finish()
return d.addCallback(finish)
class Bobby(object):
"""Bobby app views."""
app = Klein()
def __init__(self, db):
self._db = db
self._worker = BobbyWorker(self._db)
@app.route('/<string:tenant_id>/groups', methods=['POST'])
@with_transaction_id()
def create_group(self, request, log, tenant_id):
"""Create a new group.
Receive application/json content for new group creation.
:param str tenant_id: A tenant id
"""
content = json.loads(request.content.read())
group_id = content.get('groupId')
d = self._worker.create_group(tenant_id, group_id)
def _serialize_object(group):
json_object = {
'groupId': group['groupId'],
'links': [{
'href': '{0}{1}'.format(request.URLPath().path, group['groupId']),
'rel': 'self'
}],
'notification': group['notification'],
'notificationPlan': group['notificationPlan'],
'tenantId': group['tenantId']
}
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(201)
request.write(json.dumps(json_object))
request.finish()
return d.addCallback(_serialize_object)
@app.route('/<string:tenant_id>/groups/<string:group_id>', methods=['DELETE'])
@with_transaction_id()
def delete_group(self, request, log, tenant_id, group_id):
"""Delete a group.
:param str tenant_id: A tenant id
:param str group_id: A groud id
"""
d = self._worker.delete_group(tenant_id, group_id)
def finish(_):
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(204)
request.finish()
return d.addCallback(finish)
@app.route('/<string:tenant_id>/groups/<string:group_id>/servers', methods=['POST'])
@with_transaction_id()
def create_server(self, request, log, tenant_id, group_id):
"""Create a new server.
Receive application/json content for new server creation.
:param request: Twisted IRequest object.
:param log: A log object.
:param str tenant_id: A tenant id
:param str group_id: A group id
"""
# The server object is one provided via the nova API.
content = json.loads(request.content.read())
server = content.get('server')
d = self._worker.create_server(tenant_id, group_id, server)
def serialize(server):
json_object = {
'entityId': server['entityId'],
'groupId': server['groupId'],
'links': [
{
'href': '{0}{1}'.format(request.URLPath().path, server['serverId']),
'rel': 'self'
}
],
'serverId': server['serverId']
}
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(201)
request.write(json.dumps(json_object))
request.finish()
return d.addCallback(serialize)
@app.route('/<string:tenant_id>/groups/<string:group_id>/servers/<string:server_id>', methods=['DELETE'])
@with_transaction_id()
def delete_server(self, request, log, tenant_id, group_id, server_id):
"""Delete a server.
:param str tenant_id: A tenant id
:param str group_id: A groud id
:param str server_id: A server id
"""
d = self._worker.delete_server(tenant_id, group_id, server_id)
def finish(_):
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(204)
request.finish()
return d.addCallback(finish)
@app.route('/<string:tenant_id>/groups/<string:group_id>/policies', methods=['POST'])
@with_transaction_id()
def create_policy(self, request, log, tenant_id, group_id):
"""Create a new policy.
Receive application/json content for new policy creation.
:param str tenant_id: A tenant id
:param str group_id: A group id
"""
content = json.loads(request.content.read())
alarm_template_id = content.get('alarmTemplate')
check_template_id = content.get('checkTemplate')
policy_id = content.get('policyId')
d = cass.create_policy(self._db, policy_id, group_id, alarm_template_id, check_template_id)
# Trigger actions to create the alarm and checks on the MaaS side and set things up
def serialize(policy):
json_object = {
'alarmTemplate': policy['alarmTemplate'],
'checkTemplate': policy['checkTemplate'],
'groupId': policy['groupId'],
'links': [
{
'href': '{0}{1}'.format(request.URLPath().path, policy['policyId']),
'rel': 'self'
}
],
'policyId': policy['policyId']
}
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(201)
request.write(json.dumps(json_object))
request.finish()
return d.addCallback(serialize)
@app.route('/<string:tenant_id>/groups/<string:group_id>/policies/<string:policy_id>',
methods=['DELETE'])
@with_transaction_id()
def delete_policy(self, request, log, tenant_id, group_id, policy_id):
"""Delete a policy.
:param str tenant_id: A tenant id
:param str group_id: A groud id
:param str policy_id: A policy id
"""
d = cass.delete_policy(self._db, group_id, policy_id)
def finish(_):
request.setHeader('Content-Type', 'application/json')
request.setResponseCode(204)
request.finish()
return d.addCallback(finish)
@app.route('/alarm', methods=['POST'])
@with_transaction_id()
def alarm(self, request, log):
"""Change the state of an alarm."""
_policy_id = []
content = json.loads(request.content.read())
alarm_id = content.get('alarm').get('id')
status = content.get('details').get('state')
d = cass.alter_alarm_state(self._db, alarm_id, status)
def check_quorum_health((policy_id, server_id)):
_policy_id.append(policy_id)
return cass.check_quorum_health(self._db, policy_id)
d.addCallback(check_quorum_health)
def maybe_execute_policy(health):
if health:
return self._worker.execute_policy(_policy_id[0])
else:
defer.succeed(None)
d.addCallback(maybe_execute_policy)
def finish(_):
request.setResponseCode(200)
request.finish()
return d.addCallback(finish)
