diff --git a/src/main/java/earth/krakatao/protocol/KraSocketClientProtocol.java b/src/main/java/earth/krakatao/protocol/KraSocketClientProtocol.java index 8006718..d69e405 100644 --- a/src/main/java/earth/krakatao/protocol/KraSocketClientProtocol.java +++ b/src/main/java/earth/krakatao/protocol/KraSocketClientProtocol.java @@ -23,12 +23,15 @@ public class KraSocketClientProtocol { @Getter private final HashMap sendQueueMessages; + @Getter + private final HashMap receivedQueueMessages; public KraSocketClientProtocol() { //this.cmdIDs = new ArrayList<>(); this.currentCmdIDIndex = 9; //this.consumerHashMap = new HashMap<>(); this.sendQueueMessages = new HashMap<>(); + this.receivedQueueMessages = new HashMap<>(); } public int generateCmdID() { @@ -152,4 +155,8 @@ public class KraSocketClientProtocol { this.consumerHashMap.remove(cmdID); */ } + + public void removeMessageFromReceivedQueue(int cmdID) { + this.receivedQueueMessages.remove(cmdID); + } }