1 files changed, 34 insertions, 2 deletions

diff --git a/juick-server/src/main/java/com/juick/server/xmpp/router/XMPPRouter.java b/juick-server/src/main/java/com/juick/server/xmpp/router/XMPPRouter.java
index 6edecf05..5a6ee055 100644
--- a/juick-server/src/main/java/com/juick/server/xmpp/router/XMPPRouter.java
+++ b/juick-server/src/main/java/com/juick/server/xmpp/router/XMPPRouter.java
@@ -2,6 +2,7 @@ package com.juick.server.xmpp.router;
 
 import com.juick.server.XMPPServer;
 import com.juick.server.xmpp.s2s.BasicXmppSession;
+import com.juick.server.xmpp.s2s.CacheEntry;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
@@ -30,10 +31,12 @@ import java.io.StringWriter;
 import java.net.ServerSocket;
 import java.net.Socket;
 import java.net.SocketException;
+import java.time.Instant;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.Iterator;
 import java.util.List;
+import java.util.concurrent.CopyOnWriteArrayList;
 import java.util.concurrent.ExecutorService;
 
 @Component
@@ -44,6 +47,7 @@ public class XMPPRouter implements StreamHandler {
     private ExecutorService service;
 
     private final List<StreamComponentServer> connections = Collections.synchronizedList(new ArrayList<>());
+    private final List<CacheEntry> outCache = new CopyOnWriteArrayList<>();
 
     private ServerSocket listener;
 
@@ -135,6 +139,20 @@ public class XMPPRouter implements StreamHandler {
                     if (c.isLoggedIn()) {
                         connOut = c;
                         break;
+                    } else {
+                        logger.info("bouncing stanza to {} component until it will be ready", hostname);
+                        boolean haveCache = false;
+                        for (CacheEntry entry : outCache) {
+                            if (entry.hostname != null && entry.hostname.equals(hostname)) {
+                                entry.xml += xml;
+                                entry.updated = Instant.now();
+                                haveCache = true;
+                                break;
+                            }
+                        }
+                        if (!haveCache) {
+                            outCache.add(new CacheEntry(Jid.of(hostname), xml));
+                        }
                     }
                 }
             }
@@ -172,9 +190,23 @@ public class XMPPRouter implements StreamHandler {
         }
     }
 
-    @Override
-    public void ready() {
+    public String getFromCache(Jid to) {
+        final String[] cache = new String[1];
+        outCache.stream().filter(c -> c.hostname != null && c.hostname.equals(to)).findFirst().ifPresent(c -> {
+            cache[0] = c.xml;
+            outCache.remove(c);
+        });
+        return cache[0];
+    }
 
+    @Override
+    public void ready(StreamComponentServer componentServer) {
+        logger.info("component {} ready", componentServer.to);
+        String cache = getFromCache(componentServer.to);
+        if (cache != null) {
+            logger.debug("sending cache to {}", componentServer.to);
+            componentServer.send(cache);
+        }
     }
 
     @Override