修复: Servertools 添加数据库错误

UnrealMultiple · May 4, 2024 · 3402eac · 3402eac
1 parent daf3ea7
commit 3402eac
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 9 deletions.
diff --git a/ServerTools/DB/PlayerDeath.cs b/ServerTools/DB/PlayerDeath.cs
@@ -7,7 +7,7 @@ namespace ServerTools.DB;
 
 public class PlayerDeath : Dictionary<string, int>
 {
-    public new int this[string key]
+    private new int this[string key]
     {
         get
         { 
@@ -47,9 +47,15 @@ private void ReadAll()
 
     public void Add(string name)
     {
-        this[name] += 1;
-        if (database.Query("UPDATE Death SET Count = @0 WHERE Name = @1", this[name], name) != 1)
-            database.Query("INSERT INTO `Death` (`Name`, `Count`) VALUES (@0, @1)", this[name], 1);
-
+        if (ContainsKey(name))
+        {
+            this[name] += 1;
+            database.Query("UPDATE Death SET Count = @0 WHERE Name = @1", this[name], name);
+        }
+        else
+        {
+            this[name] = 1;
+            database.Query("INSERT INTO `Death` (`Name`, `Count`) VALUES (@0, @1)", name, 1);
+        }    
     }
 }
diff --git a/ServerTools/DB/PlayerOnline.cs b/ServerTools/DB/PlayerOnline.cs
@@ -7,6 +7,7 @@ namespace ServerTools.DB;
 
 public class PlayerOnline : Dictionary<string, int>
 {
+    private HashSet<string> _players = new();
     public new int this[string key]
     { 
         get 
@@ -42,6 +43,7 @@ public void ReadAll()
             string username = reader.Get<string>("username");
             int duration = reader.Get<int>("duration");
             this[username] = duration;
+            _players.Add(username);
         }
     }
 
@@ -67,6 +69,7 @@ public bool Update(string Name, int duration)
     }
     public bool Insert(string Name, int duration)
     {
+        _players.Add(Name);
         return 1 == database.Query("INSERT INTO `OnlineDuration` (`username`, `duration`) VALUES (@0, @1)", Name, duration);
     }
 
@@ -75,7 +78,9 @@ public bool Insert(string Name, int duration)
 
     public void AddOrUpdate(string name, int duration)
     {
-        if (!Update(name, duration))
+        if (_players.Contains(name))
+            Update(name, duration);
+        else
             Insert(name, duration);
     }
 

diff --git a/ServerTools/Plugin.cs b/ServerTools/Plugin.cs
@@ -126,13 +126,17 @@ private void OnUpdate(EventArgs e)
             if (TimerCount % 60 == 0)
             {
                 Timer?.Invoke(e);
-                foreach (var ply in Deads)
+                if (Config.DeadTimer)
                 {
-                    if (ply != null && ply.Active && ply.Dead)
+                    foreach (var ply in Deads)
                     {
-                        ply.SendInfoMessage(Config.DeadFormat, ply.RespawnTimer);
+                        if (ply != null && ply.Active && ply.Dead)
+                        {
+                            ply.SendInfoMessage(Config.DeadFormat, ply.RespawnTimer);
+                        }
                     }
                 }
+
             }  
         }