From 78a985147ed5cd33e77f0333e1013177d071345d Mon Sep 17 00:00:00 2001
From: Alex <alex@roese.dev>
Date: Sun, 27 Feb 2022 14:44:25 +0100
Subject: [PATCH] added timer task for ack message handling

---
 src/main/java/earth/krakatao/KraSocketClient.java   |  2 ++
 .../krakatao/KraSocketClientMessageTimerTask.java   | 13 +++++++++++++
 src/main/java/earth/krakatao/SocketClient.java      |  4 +++-
 3 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/main/java/earth/krakatao/KraSocketClient.java b/src/main/java/earth/krakatao/KraSocketClient.java
index 834a91f..2cc20f3 100644
--- a/src/main/java/earth/krakatao/KraSocketClient.java
+++ b/src/main/java/earth/krakatao/KraSocketClient.java
@@ -3,8 +3,10 @@ package earth.krakatao;
 import earth.krakatao.events.KraSocketClientEventInitiater;
 import earth.krakatao.events.KraSocketClientEventInterface;
 import earth.krakatao.protocol.KraSocketClientProtocol;
+import earth.krakatao.protocol.KraSocketClientProtocolMessage;
 import java.net.URI;
 import java.net.URISyntaxException;
+import java.util.function.Consumer;
 import java.util.logging.Logger;
 import lombok.Getter;
 
diff --git a/src/main/java/earth/krakatao/KraSocketClientMessageTimerTask.java b/src/main/java/earth/krakatao/KraSocketClientMessageTimerTask.java
index 066bc7c..90f8a50 100644
--- a/src/main/java/earth/krakatao/KraSocketClientMessageTimerTask.java
+++ b/src/main/java/earth/krakatao/KraSocketClientMessageTimerTask.java
@@ -1,12 +1,25 @@
 package earth.krakatao;
 
+import earth.krakatao.protocol.KraSocketClientProtocolMessage;
 import java.util.TimerTask;
+import java.util.function.Consumer;
 
 public class KraSocketClientMessageTimerTask extends TimerTask {
 
+  private final SocketClient socketClient;
+
+  public KraSocketClientMessageTimerTask(SocketClient socketClient) {
+    this.socketClient = socketClient;
+  }
 
   @Override
   public void run() {
     KraSocketClient.getLogger().info("MessageTimerTask is running");
+
+    for (Consumer<KraSocketClientProtocolMessage> consumer : this.socketClient.getKraProtocol()
+        .getConsumerHashMap().values()) {
+
+      KraSocketClient.getLogger().info("consumer: " + consumer);
+    }
   }
 }
diff --git a/src/main/java/earth/krakatao/SocketClient.java b/src/main/java/earth/krakatao/SocketClient.java
index 14b79ae..38f2893 100644
--- a/src/main/java/earth/krakatao/SocketClient.java
+++ b/src/main/java/earth/krakatao/SocketClient.java
@@ -8,11 +8,13 @@ import java.net.URI;
 import java.nio.ByteBuffer;
 import java.util.Timer;
 import java.util.TimerTask;
+import lombok.Getter;
 import org.java_websocket.client.WebSocketClient;
 import org.java_websocket.handshake.ServerHandshake;
 
 public class SocketClient extends WebSocketClient {
 
+  @Getter
   private final KraSocketClientProtocol kraProtocol;
   private final KraSocketClientEventInitiater kraSocketClientEventInitiater;
   private TimerTask timerTask;
@@ -42,7 +44,7 @@ public class SocketClient extends WebSocketClient {
     this.kraSocketClientEventInitiater.callOnOpen(serverHandshake);
     KraSocketClient.getLogger().info("connection opened");
 
-    this.timerTask = new KraSocketClientMessageTimerTask();
+    this.timerTask = new KraSocketClientMessageTimerTask(this);
     this.timer = new Timer(true);
     this.timer.scheduleAtFixedRate(timerTask, 0, 3 * 1000);
   }