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 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 _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 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 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)