示例#1
0
def keepAlive():
	server = config.master
	while True:
		try:
			# Wait keepAlive period
			time.sleep(config.timeout/5)
		except:
			break

		# Simulate dying
		#if (randint(0, 10) == 0):
		#	return

		try:
			# Connect to master and ask for status
			print("{time:s} Send KeepAlive".format(time=Utilities.current_time_formatted()))
			conn = messages.Connection(Utilities.connect((server["address"], server["port"])))
			connections_manager.add(conn)
			sendStatusMessage(conn)
			for msg in pending_messages.getAll():
				print("Sending result: {:s}".format(str(msg)))
				conn.send(msg)
			conn.socket.shutdown(socket.SHUT_WR)

			# Read response
			for msg in conn:
				# TM, CN: Backup servers info
				if type(msg) == messages.NoOperation:
					messageNoOperation(msg)
				# TM: Divide problem
				elif type(msg) == messages.DivideProblem:
					messageDivideProblem(msg)
				# CN: Compute partial problems solutions
				elif type(msg) == messages.SolvePartialProblems:
					messageSolvePartialProblems(msg)
				# TM: Merge partial solutions
				elif type(msg) == messages.Solutions:
					messageSolutions(msg)
				else:
					print("? {:s}".format(str(msg)))

			conn.socket.close()
			connections_manager.remove(conn)

		except OSError as msg:
			print("KeepAlive connection error: {:s}".format(str(msg)))
			if config.backups:
				index = config.backups.index(server) if server in config.backups else -1
				server = config.backups[index + 1] if len(config.backups) > index + 1 else None
				if server == None:
					print("No further backup servers available, exiting...")
					break
				print("Backup list: ", config.backups)
				print("Connecting to next backup: {:s}:{:d}".format(server["address"], server["port"]))
			else:
				break
示例#2
0
def registerBackup(address, port, assumeId=None):
	print("{time:s} Connecting to {address:s}:{port:d}...".format(time=Utilities.current_time_formatted(), address=address, port=port))
	while True:
		try:
			conn = messages.Connection(connect((address, port), ('', args.port)))
			connections_manager.add(conn)
			break
		except Exception as msg:
			print("Trying to bind...")
			time.sleep(1)

	print("Send Register message")
	request = messages.Register(Type="CommunicationServer", SolvableProblems=[], ParallelThreads=0, Id=assumeId)
	conn.send(request)
	conn.socket.shutdown(socket.SHUT_WR)

	for msg in conn:
		if type(msg) == messages.RegisterResponse:
			# Has BackupCommunicationServers field -> connect to next backup in list
			if msg.BackupCommunicationServers:
				backup = msg.BackupCommunicationServers[0]
				print("Sent to next BCS at {address:s}:{port:d}".format(address=backup['address'], port=backup['port']))
				conn.socket.close()
				connections_manager.remove(conn)
				registerBackup(backup['address'], backup['port'], msg.Id)
				return
			# Registered at last communication server in list
			else:
				print("Registered successfully")
				config.id = msg.Id
				config.timeout = msg.Timeout
				config.parent_address = address
				config.parent_port = port
		# Messages with state, sent from higher communication server
		else:
			parseMessageBCS(msg)

	conn.socket.close()
	connections_manager.remove(conn)