for line in f: # Ignore comment lines if not line.startswith('#'): #edge_exists = False # nodes hold the 2 end-nodes of the edge nodes = [int(x) for x in line.split()] #print "(%d,%d)" % (nodes[0], nodes[1]) added_replicas1=0 added_replicas2=0 added_replicas3=0 user1 = node.node(nodes[0], functions.getServer(col_fam_master,nodes[0]), functions.get_server_rep(col_fam_replica, nodes[0])) user2 = node.node(nodes[1], functions.getServer(col_fam_master,nodes[1]), functions.get_server_rep(col_fam_replica, nodes[1])) if nodes[0] < nodes[1]: col_fam_edge.insert(functions.get_a_Uuid(), {'u':nodes[0],'v':nodes[1]}) edge_count += 1 else: col_fam_edge.insert(functions.get_a_Uuid(), {'u':nodes[1],'v':nodes[0]}) edge_count +=1 #if edge_exists == False: #if not edge_exists: if True: # Check if masters/replicas of both nodes are co-located flag_col = functions.colocated(user1, user2, col_fam_mme, col_fam_mse) #print "%s --> %d" % (line, flag_col) #print "nodes[%d,%d] in servers[%d,%d]" % (user1.getId(), user2.getId(), user1.getMaster(), user2.getMaster()) if flag_col == 1: #print "--> Co-located!", user1.getId(), user2.getId() cocounter+=1
server_master_list = [0]*total_servers if len(sys.argv) != 2: raise sys.exit("Number of arguments should be 2. Give a dataset with nodes. \nExiting...") # Read the nodes file and store the nodes file_name = str(sys.argv[1]) nodes_array=[] try: with open(file_name, 'r') as f: for line in f: # Ignore comment lines if not line.startswith('#'): val = int(line) node_exists = functions.node_exists(col_fam_master, val) # If node does not exist, ADD node in the tables of masters and replicas if not node_exists: col_fam_master.insert(str(val),{'server':server_id}) server_master_list[server_id] += 1 server_id = (server_id+1) % total_servers replica_id = server_id for value1 in range(total_replicas): col_fam_replica.insert(functions.get_a_Uuid(),{'node_id':val,'server':replica_id}) #num_replica += 1 replica_id = (replica_id+1) % total_servers except: print "Unexpected error in insert_nodes.py:", sys.exc_info()[0] print "Number of masters in each server",server_master_list
# nodes hold the 2 end-nodes of the edge nodes = [int(x) for x in line.split()] #print "(%d,%d)" % (nodes[0], nodes[1]) added_replicas1 = 0 added_replicas2 = 0 added_replicas3 = 0 user1 = node.node( nodes[0], functions.getServer(col_fam_master, nodes[0]), functions.get_server_rep(col_fam_replica, nodes[0])) user2 = node.node( nodes[1], functions.getServer(col_fam_master, nodes[1]), functions.get_server_rep(col_fam_replica, nodes[1])) if nodes[0] < nodes[1]: col_fam_edge.insert(functions.get_a_Uuid(), { 'u': nodes[0], 'v': nodes[1] }) edge_count += 1 else: col_fam_edge.insert(functions.get_a_Uuid(), { 'u': nodes[1], 'v': nodes[0] }) edge_count += 1 #if edge_exists == False: #if not edge_exists: if True: # Check if masters/replicas of both nodes are co-located