From fa28502a0d796a403ad68217f39ad4cf441e8819 Mon Sep 17 00:00:00 2001
From: rtldg <55846624+rtldg@users.noreply.github.com>
Date: Mon, 20 Jun 2022 11:02:05 +0000
Subject: [PATCH] buffer playtime disconnect queries to prevent map change
 query traffic jams

---
 addons/sourcemod/scripting/shavit-stats.sp | 59 +++++++++++++++++++---
 1 file changed, 51 insertions(+), 8 deletions(-)

diff --git a/addons/sourcemod/scripting/shavit-stats.sp b/addons/sourcemod/scripting/shavit-stats.sp
index 90b7619c2..10d84112f 100644
--- a/addons/sourcemod/scripting/shavit-stats.sp
+++ b/addons/sourcemod/scripting/shavit-stats.sp
@@ -60,6 +60,7 @@ int gI_TargetSteamID[MAXPLAYERS+1];
 char gS_TargetName[MAXPLAYERS+1][MAX_NAME_LENGTH];
 
 // playtime things
+Transaction gH_DisconnectPlaytimeQueries = null;
 float gF_PlaytimeStart[MAXPLAYERS+1];
 float gF_PlaytimeStyleStart[MAXPLAYERS+1];
 int gI_CurrentStyle[MAXPLAYERS+1];
@@ -143,6 +144,41 @@ public void OnPluginStart()
 	}
 
 	CreateTimer(2.5 * 60.0, Timer_SavePlaytime, 0, TIMER_REPEAT);
+	CreateTimer(3.0, Timer_SaveDisconnectPlaytime, 0, TIMER_REPEAT);
+}
+
+public void OnMapEnd()
+{
+	PrintToServer("OnMapEnd");
+	FlushDisconnectPlaytime();
+}
+
+public void OnPluginEnd()
+{
+	PrintToServer("OnPluginEnd");
+	FlushDisconnectPlaytime();
+}
+
+void FlushDisconnectPlaytime()
+{
+	float now = GetEngineTime();
+
+	for (int i = 1; i <= MaxClients; i++)
+	{
+		if (!IsValidClient(i))
+		{
+			continue;
+		}
+
+		SavePlaytime(i, now, gH_DisconnectPlaytimeQueries);
+	}
+
+	if (gH_DisconnectPlaytimeQueries != null)
+	{
+		PrintToServer("flushing...");
+		gH_SQL.Execute(gH_DisconnectPlaytimeQueries, Trans_SavePlaytime_Success, Trans_SavePlaytime_Failure);
+		gH_DisconnectPlaytimeQueries = null;
+	}
 }
 
 public void Shavit_OnDatabaseLoaded()
@@ -308,13 +344,8 @@ public void OnClientDisconnect(int client)
 		return;
 	}
 
-	Transaction trans = null;
-	SavePlaytime(client, GetEngineTime(), trans);
-
-	if (trans != null)
-	{
-		gH_SQL.Execute(trans, Trans_SavePlaytime_Success, Trans_SavePlaytime_Failure);
-	}
+	PrintToServer("OnClientDisconnect");
+	SavePlaytime(client, GetEngineTime(), gH_DisconnectPlaytimeQueries);
 }
 
 public void Shavit_OnStyleChanged(int client, int oldstyle, int newstyle, int track, bool manual)
@@ -444,7 +475,7 @@ public void Trans_SavePlaytime_Failure(Database db, any data, int numQueries, co
 	LogError("Timer (stats save playtime) SQL query %d/%d failed. Reason: %s", failIndex, numQueries, error);
 }
 
-void SavePlaytime(int client, float now, Transaction&trans)
+void SavePlaytime(int client, float now, Transaction& trans)
 {
 	int iSteamID = GetSteamAccountID(client);
 
@@ -500,6 +531,18 @@ public Action Timer_SavePlaytime(Handle timer, any data)
 	return Plugin_Continue;
 }
 
+public Action Timer_SaveDisconnectPlaytime(Handle timer, any data)
+{
+	if (gH_SQL == null || gH_DisconnectPlaytimeQueries == null)
+	{
+		return Plugin_Continue;
+	}
+
+	gH_SQL.Execute(gH_DisconnectPlaytimeQueries, Trans_SavePlaytime_Success, Trans_SavePlaytime_Failure);
+	gH_DisconnectPlaytimeQueries = null;
+	return Plugin_Continue;
+}
+
 public Action Command_Playtime(int client, int args)
 {
 	if (!IsValidClient(client))