You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.5 KiB
56 lines
1.5 KiB
2 years ago
|
import asyncio
|
||
|
|
from functools import wraps
|
||
|
|
import socketio
|
||
|
|
import socket
|
||
|
|
|
||
|
|
def get_ip_address():
|
||
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
try:
|
||
|
|
# This IP address doesn't need to be reachable, as we're only using it to find the local IP address
|
||
|
|
s.connect(("10.255.255.255", 1))
|
||
|
|
ip = s.getsockname()[0]
|
||
|
|
except Exception:
|
||
|
|
ip = "127.0.0.1"
|
||
|
|
finally:
|
||
|
|
s.close()
|
||
|
|
return ip
|
||
|
|
|
||
|
|
# Update these with the correct values for your host and server
|
||
|
|
HOST_SERVER_IP = "192.168.1.113"
|
||
|
|
HOST_SERVER_PORT = 4999
|
||
|
|
SERVER_NAME = "server_1"
|
||
|
|
SERVER_IP = get_ip_address()
|
||
|
|
SERVER_PORT = 8000
|
||
|
|
|
||
|
|
sio = socketio.AsyncClient()
|
||
|
|
|
||
|
|
async def announce_server():
|
||
|
|
await sio.connect(f'http://{HOST_SERVER_IP}:{HOST_SERVER_PORT}')
|
||
|
|
await sio.emit('register', {'name': SERVER_NAME, 'ip': SERVER_IP, 'port': SERVER_PORT})
|
||
|
|
|
||
|
|
@sio.on("heartbeat")
|
||
|
|
async def on_heartbeat():
|
||
|
|
print("Received heartbeat from host")
|
||
|
|
|
||
|
|
@sio.event
|
||
|
|
async def disconnect():
|
||
|
|
print("Disconnected from host")
|
||
|
|
|
||
|
|
def announce_server_decorator(host_block_function):
|
||
|
|
@wraps(host_block_function)
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
loop = asyncio.get_event_loop()
|
||
|
|
|
||
|
|
# Start the server announcement task
|
||
|
|
announce_task = loop.create_task(announce_server())
|
||
|
|
|
||
|
|
# Run the original host_block function
|
||
|
|
result = host_block_function(*args, **kwargs)
|
||
|
|
|
||
|
|
# Cancel the announcement task after the host_block function is done
|
||
|
|
announce_task.cancel()
|
||
|
|
|
||
|
|
return result
|
||
|
|
|
||
|
|
return wrapper
|