Stop heart beating while reconnecting

yourWaifu · Aug 29, 2018 · 6ffa037 · 6ffa037
1 parent 88c7604
commit 6ffa037
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 6 deletions.
diff --git a/examples/hello/hello.vcxproj b/examples/hello/hello.vcxproj
@@ -89,7 +89,7 @@
     <Link>
       <EnableCOMDATFolding>true</EnableCOMDATFolding>
       <OptimizeReferences>true</OptimizeReferences>
-      <AdditionalDependencies>sleepy_discord.lib;libcurl_a.lib;libsslMT.lib;libcryptoMT.lib;crypt32.lib;%(AdditionalDependencies)</AdditionalDependencies>
+      <AdditionalDependencies>sleepy_discord.lib;libcurl_a.lib;libcryptoMT.lib;libsslMT.lib;crypt32.lib;Wldap32.lib;Advapi32.lib;Normaliz.lib;User32.lib</AdditionalDependencies>
     </Link>
   </ItemDefinitionGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
@@ -100,7 +100,7 @@
       <PreprocessorDefinitions>_SCL_SECURE_NO_WARNINGS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
     </ClCompile>
     <Link>
-      <AdditionalDependencies>sleepy_discord.lib;libcurl_a.lib;libsslMT.lib;libcryptoMT.lib;crypt32.lib;Wldap32.lib;Normaliz.lib;%(AdditionalDependencies)</AdditionalDependencies>
+      <AdditionalDependencies>sleepy_discord.lib;libcurl_a.lib;libcryptoMT.lib;libsslMT.lib;crypt32.lib;Wldap32.lib;Advapi32.lib;Normaliz.lib;User32.lib</AdditionalDependencies>
     </Link>
   </ItemDefinitionGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">

diff --git a/include/sleepy_discord/client.h b/include/sleepy_discord/client.h
@@ -322,6 +322,7 @@ namespace SleepyDiscord {
 		void processCloseCode(const int16_t code);
 		void heartbeat();
 		void sendHeartbeat();
+		void resetHeartbeatValues();
 		inline std::string getToken() { return *token.get(); }
 		void start(const std::string _token, const char maxNumOfThreads = 2, int _shardID = 0, int _shardCount = 0);
 		virtual bool connect(const std::string & uri) { return false; }

diff --git a/sleepy_discord/client.cpp b/sleepy_discord/client.cpp
@@ -427,7 +427,7 @@ namespace SleepyDiscord {
 		case HELLO:
 			heartbeatInterval = std::stoi(json::getValue(d->c_str(), "heartbeat_interval"));
 			heartbeat();
-			if (!ready) sendIdentity();
+			if (sessionID.empty()) sendIdentity();
 			else sendResume();
 			break;
 		case RECONNECT:
@@ -438,7 +438,7 @@ namespace SleepyDiscord {
 				sleep(2500);
 				sendResume();
 			} else {
-				sessionID = "";
+				sessionID = {};
 				sendIdentity();
 			}
 			break;
@@ -460,15 +460,20 @@ namespace SleepyDiscord {
 		case DECODE_ERROR:
 		case NOT_AUTHENTICATED:
 		case ALREADY_AUTHENTICATED:
-		case INVALID_SEQ:
 		case RATE_LIMITED:
 		case SESSION_TIMEOUT:
 		default:
 			break;
 
+		case SESSION_NO_LONGER_VALID:
+		case INVALID_SEQ:
 		case 1000:
-			if (!isQuiting())
+			if (!isQuiting()) {
+				//restart with new session
+				sessionID = {};
+				lastSReceived = 0;
 				break;
+			}
 			//else fall through
 
 		//Might be Unrecoveralbe
@@ -479,6 +484,7 @@ namespace SleepyDiscord {
 			return quit(false, true);
 			break;
 		}
+		resetHeartbeatValues();
 		reconnect(1001);
 	}