From bbcfbacb4fd1facbc18d4106842da93a1cf8bee0 Mon Sep 17 00:00:00 2001
From: Austin Hellyer <hello@austinhellyer.me>
Date: Tue, 29 Nov 2016 07:33:38 -0800
Subject: Retrieve a new URL on reconnect

---
 src/client/gateway/shard.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/client/gateway/shard.rs b/src/client/gateway/shard.rs
index e64d6fe..f7e1bde 100644
--- a/src/client/gateway/shard.rs
+++ b/src/client/gateway/shard.rs
@@ -6,7 +6,7 @@ use std::thread::{self, Builder as ThreadBuilder};
 use std::time::Duration as StdDuration;
 use std::mem;
 use super::super::login_type::LoginType;
-use super::super::Client;
+use super::super::{Client, rest};
 use super::{GatewayError, GatewayStatus, prep};
 use websocket::client::{Client as WsClient, Sender, Receiver};
 use websocket::message::Message as WsMessage;
@@ -435,7 +435,9 @@ impl Shard {
         // Take a few attempts at reconnecting; otherwise fall back to
         // re-instantiating the connection.
         for _ in 0..3 {
-            let shard = Shard::new(&self.ws_url,
+            let gateway_url = try!(rest::get_gateway()).url;
+
+            let shard = Shard::new(&gateway_url,
                                    &self.token,
                                    self.shard_info,
                                    self.login_type);
-- 
cgit v1.2.3