def keepAlive():
while True:
try:
# Wait keepAlive period
time.sleep(config.timeout / 2)
except:
break
# Connect to master and ask for status
print("{time:s} Send KeepAlive".format(time=Utilities.current_time_formatted()))
try:
conn = messages.Connection(socket.create_connection((config.parent_address, config.parent_port)))
connections_manager.add(conn)
msg = messages.Status(config.id, [{ "State": "Idle" }])
conn.send(msg)
# Inform parent about lower backup registrations and deregistrations
for msg in backup_servers.getParentMessages():
conn.send(msg)
conn.socket.shutdown(socket.SHUT_WR)
# Read all messages from client separated by 0x17 until EOC (write stream closed)
for msg in conn:
parseMessageBCS(msg)
conn.socket.close()
connections_manager.remove(conn)
except OSError as msg:
print("KeepAlive connection error: {:s}".format(str(msg)))
print("Assuming main communication server role")
# Refresh components timeout, else when assume main cs role all would be thought dead
for component in components.active():
component.touch()
config.is_backup = False
break
def handleConnection(conn):
print("{time:s} Connection from {server:s}:{port:d}".format(time=strftime("%H:%M:%S"), server=conn.socket.getpeername()[0], port=conn.socket.getpeername()[1]))
try:
parseMessages(conn)
except AssertionError as e:
print("Assertion error: {0}".format(e))
finally:
# Close connection
conn.socket.shutdown(socket.SHUT_WR)
conn.socket.close()
connections_manager.remove(conn)
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)
