GarpBot/cs2-WeaponPaints
1
0
Fork 0
mirror of https://github.com/Nereziel/cs2-WeaponPaints.git synced 2026-03-11 08:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

using transaction to sync killstreaks

Browse Source
This commit is contained in:
originalaidn
2024-10-18 23:25:37 +02:00
parent 7172dc344f
commit c43f4bde80
2 changed files with 56 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
Events.cs
View File

@@ -5,6 +5,7 @@ using CounterStrikeSharp.API.Modules.Entities;
using CounterStrikeSharp.API.Modules.Memory; using CounterStrikeSharp.API.Modules.Memory;
using CounterStrikeSharp.API.Modules.Memory.DynamicFunctions; using CounterStrikeSharp.API.Modules.Memory.DynamicFunctions;
using System.Runtime.InteropServices; using System.Runtime.InteropServices;
using System.Collections.Concurrent;
namespace WeaponPaints namespace WeaponPaints
{ {
@@ -68,6 +69,21 @@ namespace WeaponPaints
if (player is null || !player.IsValid || player.IsBot) return HookResult.Continue; if (player is null || !player.IsValid || player.IsBot) return HookResult.Continue;
var playerInfo = new PlayerInfo
{
UserId = player.UserId,
Slot = player.Slot,
Index = (int)player.Index,
SteamId = player.SteamID.ToString(),
Name = player.PlayerName,
IpAddress = player.IpAddress?.Split(":")[0]
};
if (!GPlayerWeaponsInfo.TryGetValue(player.Slot, out var weaponInfos))
return HookResult.Continue;
_ = Task.Run(async () => await SyncStatTrakForPlayer(playerInfo, weaponInfos));
if (Config.Additional.SkinEnabled) if (Config.Additional.SkinEnabled)
{ {
GPlayerWeaponsInfo.TryRemove(player.Slot, out _); GPlayerWeaponsInfo.TryRemove(player.Slot, out _);
@@ -97,37 +113,29 @@ namespace WeaponPaints
CommandsCooldown.Remove(player.Slot); CommandsCooldown.Remove(player.Slot);
var playerInfo = new PlayerInfo
{
UserId = player.UserId,
Slot = player.Slot,
Index = (int)player.Index,
SteamId = player.SteamID.ToString(),
Name = player.PlayerName,
IpAddress = player.IpAddress?.Split(":")[0]
};
if (!GPlayerWeaponsInfo.TryGetValue(player.Slot, out var weaponInfos))
return HookResult.Continue;
foreach (var weapon in weaponInfos)
{
var weaponDefIndex = weapon.Key;
var weaponInfo = weapon.Value;
if (weaponInfo.StatTrak)
{
if (WeaponSync != null)
{
_ = Task.Run(async () => await WeaponSync.SyncStatTrakToDatabase(playerInfo, weaponInfo.StatTrakCount, weaponDefIndex));
}
}
}
return HookResult.Continue; return HookResult.Continue;
} }
public async Task SyncStatTrakForPlayer(PlayerInfo playerInfo, ConcurrentDictionary<int, WeaponInfo> weaponInfos)
{
if (WeaponSync == null || weaponInfos == null || weaponInfos.Count == 0) return;
var statTrakWeapons = weaponInfos
.Where(w => w.Value.StatTrak)
.ToDictionary(w => w.Key, w => w.Value.StatTrakCount);
if (statTrakWeapons.Count == 0) return;
try
{
await WeaponSync.SyncStatTrakToDatabase(playerInfo, statTrakWeapons);
}
catch (Exception ex)
{
Utility.Log($"Error syncing StatTrak for player {playerInfo.SteamId}: {ex.Message}");
}
}
private void OnMapStart(string mapName) private void OnMapStart(string mapName)
{ {
if (Config.Additional is { KnifeEnabled: false, SkinEnabled: false, GloveEnabled: false }) return; if (Config.Additional is { KnifeEnabled: false, SkinEnabled: false, GloveEnabled: false }) return;

37
WeaponSynchronization.cs
View File

@@ -373,34 +373,37 @@ namespace WeaponPaints
} }
} }
internal async Task SyncStatTrakToDatabase(PlayerInfo player, int StatTrakCount, int defindex) internal async Task SyncStatTrakToDatabase(PlayerInfo player, Dictionary<int, int> weaponStatTrakCounts)
{ {
if (string.IsNullOrEmpty(player.SteamId) || !WeaponPaints.GPlayerWeaponsInfo.TryGetValue(player.Slot, out var weaponsInfo)) if (string.IsNullOrEmpty(player.SteamId) || weaponStatTrakCounts == null || weaponStatTrakCounts.Count == 0)
return; return;
try try
{ {
await using var connection = await _database.GetConnectionAsync(); await using var connection = await _database.GetConnectionAsync();
await using var transaction = await connection.BeginTransactionAsync();
const string queryCheckExistence = "SELECT COUNT(*) FROM `wp_player_skins` WHERE `steamid` = @steamid AND `weapon_defindex` = @weaponDefIndex"; foreach (var weapon in weaponStatTrakCounts)
{
int defindex = weapon.Key;
int statTrakCount = weapon.Value;
var existingRecordCount = await connection.ExecuteScalarAsync<int>(queryCheckExistence, new { steamid = player.SteamId, weaponDefIndex = defindex }); const string query = @"
INSERT INTO `wp_player_skins` (`steamid`, `weapon_defindex`, `weapon_stattrak_count`)
VALUES (@steamid, @weaponDefIndex, @StatTrakCount)
ON DUPLICATE KEY UPDATE `weapon_stattrak_count` = @StatTrakCount";
string query = string.Empty; var parameters = new
object? parameters = null;
if (existingRecordCount > 0)
{ {
query = "UPDATE `wp_player_skins` SET `weapon_stattrak_count` = @StatTrakCount WHERE `steamid` = @steamid AND `weapon_defindex` = @weaponDefIndex"; steamid = player.SteamId,
parameters = new { steamid = player.SteamId, weaponDefIndex = defindex, StatTrakCount }; weaponDefIndex = defindex,
} StatTrakCount = statTrakCount
else };
{
query = "INSERT INTO `wp_player_skins` (`steamid`, `weapon_defindex`, `weapon_stattrak_count`) VALUES (@steamid, @weaponDefIndex, @StatTrakCount)";
parameters = new { steamid = player.SteamId, weaponDefIndex = defindex, StatTrakCount };
}
await connection.ExecuteAsync(query, parameters); await connection.ExecuteAsync(query, parameters, transaction);
}
await transaction.CommitAsync();
} }
catch (Exception e) catch (Exception e)
{ {