timeout of 60 seconds to connect to lavalink

This commit is contained in:
Mylloon 2021-08-19 09:59:47 +02:00
parent 82a682f03a
commit 1266b63e75

View file

@ -354,7 +354,7 @@ class Music(commands.Cog, wavelink.WavelinkMixin):
await node.destroy() await node.destroy()
nodes = {'MAIN': {'host': f'{self.url}', nodes = {'MAIN': {'host': f'{self.url}',
'port': 2333, 'port': 2334,
'rest_uri': f'http://{self.url}:2333', 'rest_uri': f'http://{self.url}:2333',
'password': 'youshallnotpass', 'password': 'youshallnotpass',
'identifier': 'MAIN', 'identifier': 'MAIN',
@ -363,6 +363,7 @@ class Music(commands.Cog, wavelink.WavelinkMixin):
nodeValues = list(nodes.values()) nodeValues = list(nodes.values())
n = 0 n = 0
timeout = 60 # en secondes
while n < len(nodeValues): while n < len(nodeValues):
location = (nodeValues[n]["host"], nodeValues[n]["port"]) location = (nodeValues[n]["host"], nodeValues[n]["port"])
a_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) a_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@ -371,8 +372,12 @@ class Music(commands.Cog, wavelink.WavelinkMixin):
await self.bot.wavelink.initiate_node(**nodeValues[n]) # Port isn't open -> a service is behind, connecting... await self.bot.wavelink.initiate_node(**nodeValues[n]) # Port isn't open -> a service is behind, connecting...
n += 1 n += 1
else: else:
timeout -= 1
sleep(1) # Port is open -> no service behind, retrying in 1 second... sleep(1) # Port is open -> no service behind, retrying in 1 second...
if timeout <= 0:
raise TimeoutError
@wavelink.WavelinkMixin.listener() @wavelink.WavelinkMixin.listener()
async def on_node_ready(self, node: wavelink.Node): async def on_node_ready(self, node: wavelink.Node):
print(f'Node Lavalink \"{node.identifier}\" prête !') print(f'Node Lavalink \"{node.identifier}\" prête !')