From 8ec2d34a6096cd38c9b25c2706b3237be0b007d6 Mon Sep 17 00:00:00 2001
From: LOOHP <jamesloohp@gmail.com>
Date: Thu, 18 Dec 2025 16:59:52 +0000
Subject: [PATCH] Refactor

---
 .../com/loohp/limbo/network/ClientConnection.java    | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/loohp/limbo/network/ClientConnection.java b/src/main/java/com/loohp/limbo/network/ClientConnection.java
index dab48de..1d0b10e 100644
--- a/src/main/java/com/loohp/limbo/network/ClientConnection.java
+++ b/src/main/java/com/loohp/limbo/network/ClientConnection.java
@@ -155,7 +155,6 @@ public class ClientConnection extends Thread {
 
     private static final Key DEFAULT_HANDLER_NAMESPACE = Key.key("default");
     private static final String BRAND_ANNOUNCE_CHANNEL = Key.key("brand").toString();
-    private final AtomicLong lastKeepAliveResponse = new AtomicLong(System.currentTimeMillis());
 
     private final Random random = new Random();
     private final Socket clientSocket;
@@ -163,10 +162,12 @@ public class ClientConnection extends Thread {
     private boolean running;
     private volatile ClientState state;
 
+    private final AtomicLong lastPacketTimestamp;
+    private final AtomicLong lastKeepAlivePayLoad;
+    private final AtomicLong lastKeepAliveResponse;
+
     private Player player;
     private TimerTask keepAliveTask;
-    private AtomicLong lastPacketTimestamp;
-    private AtomicLong lastKeepAlivePayLoad;
     private InetAddress inetAddress;
     private boolean ready;
 
@@ -175,6 +176,7 @@ public class ClientConnection extends Thread {
         this.inetAddress = clientSocket.getInetAddress();
         this.lastPacketTimestamp = new AtomicLong(-1);
         this.lastKeepAlivePayLoad = new AtomicLong(-1);
+        this.lastKeepAliveResponse = new AtomicLong(-1);
         this.channel = null;
         this.running = false;
         this.ready = false;
@@ -192,6 +194,10 @@ public class ClientConnection extends Thread {
         this.lastKeepAlivePayLoad.set(payLoad);
     }
 
+    public long getLastKeepAliveResponse() {
+        return lastKeepAliveResponse.get();
+    }
+
     public long getLastPacketTimestamp() {
         return lastPacketTimestamp.get();
     }