def host_create():
data = get_json(request.body)
data = preset_merge(data, 'servers')
result = _host_create(data)
result['links'].extend([
base_link('service'),
base_link('get-releases'),
server_link('get-servers'),
server_link('add-server', self_rel=True),
replica_set_link('get-replica-sets'),
sharded_cluster_link('get-sharded-clusters')
])
return send_result(200, result)
def _sh_create(params):
cluster_id = ShardedClusters().create(params)
result = ShardedClusters().info(cluster_id)
result['links'] = all_sharded_cluster_links(cluster_id)
for router in result['routers']:
router['links'] = [
server_link('get-server-info', server_id=router['id'])
]
for cfg in result['configsvrs']:
cfg['links'] = [server_link('get-server-info', server_id=cfg['id'])]
for sh in result['shards']:
sh['links'] = [
sharded_cluster_link('get-shard-info', cluster_id, sh['id']),
_server_or_rs_link(sh)
]
return result
def routers(cluster_id):
logger.debug("routers({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
router_docs = []
for router_info in ShardedClusters().routers(cluster_id):
# Server id is the same as router id.
server_id = router_info['id']
links = [
sharded_cluster_link('delete-router',
cluster_id, router_id=server_id),
server_link('get-server-info', server_id)
]
router_info['links'] = links
router_docs.append(router_info)
result = {
'routers': router_docs,
'links': [
sharded_cluster_link('get-sharded-cluster-info', cluster_id),
sharded_cluster_link('get-shards', cluster_id),
sharded_cluster_link('get-configsvrs', cluster_id),
sharded_cluster_link('get-routers', cluster_id, self_rel=True)
]
}
return send_result(200, result)
def routers(cluster_id):
logger.debug("routers({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
router_docs = []
for router_info in ShardedClusters().routers(cluster_id):
# Server id is the same as router id.
server_id = router_info['id']
links = [
sharded_cluster_link('delete-router',
cluster_id,
router_id=server_id),
server_link('get-server-info', server_id)
]
router_info['links'] = links
router_docs.append(router_info)
result = {
'routers':
router_docs,
'links': [
sharded_cluster_link('get-sharded-cluster-info', cluster_id),
sharded_cluster_link('get-shards', cluster_id),
sharded_cluster_link('get-configsvrs', cluster_id),
sharded_cluster_link('get-routers', cluster_id, self_rel=True)
]
}
return send_result(200, result)
def host_info(host_id):
logger.debug("host_info({host_id})".format(**locals()))
if host_id not in Servers():
return send_result(404)
result = Servers().info(host_id)
result['links'] = all_server_links(host_id, rel_to='get-server-info')
result['links'].append(server_link('get-servers'))
return send_result(200, result)
def host_list():
logger.debug("host_list()")
servers = []
for server_id in Servers():
server_info = {'id': server_id}
server_info['links'] = all_server_links(
server_id, rel_to='get-servers')
servers.append(server_info)
response = {'links': [
base_link('service'),
base_link('get-releases'),
server_link('get-servers', self_rel=True),
server_link('add-server'),
replica_set_link('get-replica-sets'),
sharded_cluster_link('get-sharded-clusters')
]}
response['servers'] = servers
return send_result(200, response)
def _sh_create(params):
cluster_id = ShardedClusters().create(params)
result = ShardedClusters().info(cluster_id)
result['links'] = all_sharded_cluster_links(cluster_id)
for router in result['routers']:
router['links'] = [
server_link('get-server-info', server_id=router['id'])
]
for cfg in result['configsvrs']:
cfg['links'] = [
server_link('get-server-info', server_id=cfg['id'])
]
for sh in result['shards']:
sh['links'] = [
sharded_cluster_link('get-shard-info', cluster_id, sh['id']),
_server_or_rs_link(sh)
]
return result
def info(cluster_id):
logger.debug("info({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
result = ShardedClusters().info(cluster_id)
result['links'] = all_sharded_cluster_links(
cluster_id, rel_to='get-sharded-cluster-info')
for router in result['routers']:
router['links'] = [
server_link('get-server-info', server_id=router['id'])
]
for cfg in result['configsvrs']:
cfg['links'] = [server_link('get-server-info', server_id=cfg['id'])]
for sh in result['shards']:
sh['links'] = [
sharded_cluster_link('get-shard-info', cluster_id, sh['id']),
_server_or_rs_link(sh)
]
return send_result(200, result)
def _build_member_links(rs_id, member_doc):
server_id = member_doc['server_id']
member_id = member_doc['_id']
member_links = [
replica_set_link('get-replica-set-member-info', rs_id, member_id),
replica_set_link('delete-replica-set-member', rs_id, member_id),
replica_set_link('update-replica-set-member-config', rs_id, member_id),
server_link('get-server-info', server_id=server_id)
]
return member_links
def _build_member_links(rs_id, member_doc):
server_id = member_doc['server_id']
member_id = member_doc['_id']
member_links = [
replica_set_link('get-replica-set-member-info', rs_id, member_id),
replica_set_link('delete-replica-set-member', rs_id, member_id),
replica_set_link('update-replica-set-member-config',
rs_id, member_id),
server_link('get-server-info', server_id=server_id)
]
return member_links
def _rs_create(params):
rs_id = ReplicaSets().create(params)
result = ReplicaSets().info(rs_id)
result['links'] = all_replica_set_links(rs_id)
# Add GET link to corresponding Server resource.
for member in result['members']:
member['links'] = [
server_link('get-server-info', server_id=member['server_id']),
replica_set_link(
'get-replica-set-member-info', rs_id, member['_id'])
]
return result
def info(cluster_id):
logger.debug("info({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
result = ShardedClusters().info(cluster_id)
result['links'] = all_sharded_cluster_links(
cluster_id, rel_to='get-sharded-cluster-info')
for router in result['routers']:
router['links'] = [
server_link('get-server-info', server_id=router['id'])
]
for cfg in result['configsvrs']:
cfg['links'] = [
server_link('get-server-info', server_id=cfg['id'])
]
for sh in result['shards']:
sh['links'] = [
sharded_cluster_link('get-shard-info', cluster_id, sh['id']),
_server_or_rs_link(sh)
]
return send_result(200, result)
def _rs_create(params):
rs_id = ReplicaSets().create(params)
result = ReplicaSets().info(rs_id)
result['links'] = all_replica_set_links(rs_id)
# Add GET link to corresponding Server resource.
for member in result['members']:
member['links'] = [
server_link('get-server-info', server_id=member['server_id']),
replica_set_link('get-replica-set-member-info', rs_id,
member['_id'])
]
return result
def host_command(host_id):
logger.debug("host_command({host_id})".format(**locals()))
if host_id not in Servers():
return send_result(404)
command = get_json(request.body).get('action')
if command is None:
raise RequestError('Expected body with an {"action": ...}.')
result = {
'command_result': Servers().command(host_id, command),
'links': all_server_links(host_id, rel_to='server-command')
}
result['links'].append(server_link('get-servers'))
return send_result(200, result)
def rs_info(rs_id):
logger.debug("rs_info({rs_id})".format(**locals()))
if rs_id not in ReplicaSets():
return send_result(404)
result = ReplicaSets().info(rs_id)
result['links'] = all_replica_set_links(rs_id, 'get-replica-set-info')
for member in result['members']:
member['links'] = [
server_link('get-server-info', server_id=member['server_id']),
replica_set_link('get-replica-set-member-info', rs_id,
member['_id'])
]
return send_result(200, result)
def rs_info(rs_id):
logger.debug("rs_info({rs_id})".format(**locals()))
if rs_id not in ReplicaSets():
return send_result(404)
result = ReplicaSets().info(rs_id)
result['links'] = all_replica_set_links(rs_id, 'get-replica-set-info')
for member in result['members']:
member['links'] = [
server_link('get-server-info', server_id=member['server_id']),
replica_set_link(
'get-replica-set-member-info', rs_id, member['_id'])
]
return send_result(200, result)
def router_add(cluster_id):
logger.debug("router_add({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
data = get_json(request.body)
result = ShardedClusters().router_add(cluster_id, data)
router_id = result['id']
result['links'] = [
server_link('get-server-info', router_id),
sharded_cluster_link('add-router', cluster_id, self_rel=True),
sharded_cluster_link('delete-router', cluster_id, router_id=router_id),
sharded_cluster_link('get-sharded-cluster-info', cluster_id),
sharded_cluster_link('get-routers', cluster_id),
]
return send_result(200, result)
def router_add(cluster_id):
logger.debug("router_add({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
data = get_json(request.body)
result = ShardedClusters().router_add(cluster_id, data)
router_id = result['id']
result['links'] = [
server_link('get-server-info', router_id),
sharded_cluster_link('add-router', cluster_id, self_rel=True),
sharded_cluster_link('delete-router', cluster_id, router_id=router_id),
sharded_cluster_link('get-sharded-cluster-info', cluster_id),
sharded_cluster_link('get-routers', cluster_id),
]
return send_result(200, result)
def rs_create_by_id(rs_id):
logger.debug("rs_create_by_id()")
data = get_json(request.body)
data = preset_merge(data, 'replica_sets')
data['id'] = rs_id
result = _rs_create(data)
result['links'].extend([
base_link('service'),
base_link('get-releases'),
sharded_cluster_link('get-sharded-clusters'),
replica_set_link('get-replica-sets'),
replica_set_link('add-replica-set'),
replica_set_link('add-replica-set-by-id', rs_id, self_rel=True),
server_link('get-servers')
])
return send_result(200, result)
def rs_create_by_id(rs_id):
logger.debug("rs_create_by_id()")
data = get_json(request.body)
data = preset_merge(data, 'replica_sets')
data['id'] = rs_id
result = _rs_create(data)
result['links'].extend([
base_link('service'),
base_link('get-releases'),
sharded_cluster_link('get-sharded-clusters'),
replica_set_link('get-replica-sets'),
replica_set_link('add-replica-set'),
replica_set_link('add-replica-set-by-id', rs_id, self_rel=True),
server_link('get-servers')
])
return send_result(200, result)
def rs_list():
logger.debug("rs_list()")
replica_sets = []
for rs_id in ReplicaSets():
repl_info = {'id': rs_id}
repl_info['links'] = all_replica_set_links(rs_id, 'get-replica-sets')
replica_sets.append(repl_info)
response = {'links': [
base_link('service'),
base_link('get-releases'),
sharded_cluster_link('get-sharded-clusters'),
replica_set_link('get-replica-sets', self_rel=True),
replica_set_link('add-replica-set'),
server_link('get-servers')
]}
response['replica_sets'] = replica_sets
return send_result(200, response)
def rs_list():
logger.debug("rs_list()")
replica_sets = []
for rs_id in ReplicaSets():
repl_info = {'id': rs_id}
repl_info['links'] = all_replica_set_links(rs_id, 'get-replica-sets')
replica_sets.append(repl_info)
response = {'links': [
base_link('service'),
base_link('get-releases'),
sharded_cluster_link('get-sharded-clusters'),
replica_set_link('get-replica-sets', self_rel=True),
replica_set_link('add-replica-set'),
server_link('get-servers')
]}
response['replica_sets'] = replica_sets
return send_result(200, response)
def configsvrs(cluster_id):
logger.debug("configsvrs({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
config_docs = []
for config_info in ShardedClusters().configsvrs(cluster_id):
server_id = config_info['id']
config_info['links'] = [server_link('get-server-info', server_id)]
config_docs.append(config_info)
result = {
'configsvrs':
config_docs,
'links': [
sharded_cluster_link('get-sharded-cluster-info', cluster_id),
sharded_cluster_link('get-shards', cluster_id),
sharded_cluster_link('get-configsvrs', cluster_id, self_rel=True),
sharded_cluster_link('get-routers', cluster_id)
]
}
return send_result(200, result)
def configsvrs(cluster_id):
logger.debug("configsvrs({cluster_id})".format(**locals()))
if cluster_id not in ShardedClusters():
return send_result(404)
config_docs = []
for config_info in ShardedClusters().configsvrs(cluster_id):
server_id = config_info['id']
config_info['links'] = [
server_link('get-server-info', server_id)
]
config_docs.append(config_info)
result = {
'configsvrs': config_docs,
'links': [
sharded_cluster_link('get-sharded-cluster-info', cluster_id),
sharded_cluster_link('get-shards', cluster_id),
sharded_cluster_link('get-configsvrs', cluster_id, self_rel=True),
sharded_cluster_link('get-routers', cluster_id)
]
}
return send_result(200, result)
def _server_or_rs_link(shard_doc):
resource_id = shard_doc['_id']
if shard_doc.get('isReplicaSet'):
return replica_set_link('get-replica-set-info', resource_id)
return server_link('get-server-info', resource_id)
def _server_or_rs_link(shard_doc):
resource_id = shard_doc['_id']
if shard_doc.get('isReplicaSet'):
return replica_set_link('get-replica-set-info', resource_id)
return server_link('get-server-info', resource_id)
