juick-xmpp/src/main/java/com/juick/components/s2s/CleaningUp.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

package com.juick.components.s2s;

import com.juick.components.XMPPServer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.Scheduled;

import javax.inject.Inject;
import java.util.Iterator;

/**
 *
 * @author ugnich
 */
public class CleaningUp {

    private static final Logger logger = LoggerFactory.getLogger(CleaningUp.class);

    @Inject
    XMPPServer xmpp;

    @Scheduled(fixedDelay = 10000)
    public void cleanUp() {
        long now = System.currentTimeMillis();

        synchronized (xmpp.getOutConnections()) {
            for (Iterator<ConnectionOut> i = xmpp.getOutConnections().iterator(); i.hasNext(); ) {
                ConnectionOut c = i.next();
                int inactive = (int) ((double) (now - c.tsLocalData) / 1000.0);
                if (inactive > 900) {
                    logger.info("closing idle outgoing connection to {}", c.to);
                    c.closeConnection();
                    i.remove();
                }
            }
        }

        synchronized (xmpp.getInConnections()) {
            for (Iterator<ConnectionIn> i = xmpp.getInConnections().iterator(); i.hasNext(); ) {
                ConnectionIn c = i.next();
                int inactive = (int) ((double) (now - c.tsRemoteData) / 1000.0);
                if (inactive > 900) {
                    logger.info("closing idle incoming connection from {}", c.from);
                    c.closeConnection();
                    i.remove();
                }
            }
        }
    }
}