Mercury-IM/core/src/main/java/org/mercury_im/messenger/core/ConnectionCenter.java

package org.mercury_im.messenger.core;

import org.jivesoftware.smack.AbstractXMPPConnection;
import org.jivesoftware.smack.tcp.XMPPTCPConnection;
import org.jivesoftware.smack.tcp.XMPPTCPConnectionConfiguration;
import org.mercury_im.messenger.persistence.model.AccountModel;

import java.util.HashMap;
import java.util.Map;

import javax.inject.Inject;
import javax.inject.Singleton;

@Singleton
public class ConnectionCenter {

    private static ConnectionCenter INSTANCE;

    private final Map<Long, MercuryConnection> connectionMap = new HashMap<>();

    @Inject
    EntityCapsStore capsStore;

    private ConnectionCenter() {

    }

    public static ConnectionCenter get() {
        if (INSTANCE == null) {
            INSTANCE = new ConnectionCenter();
        }
        return INSTANCE;
    }

    public MercuryConnection getConnection(long accountId) {
        return connectionMap.get(accountId);
    }

    public void putConnection(MercuryConnection connection) {
        connectionMap.put(connection.getAccountId(), connection);
    }

    public MercuryConnection createConnection(AccountModel accountModel) {
        XMPPTCPConnectionConfiguration configuration = XMPPTCPConnectionConfiguration.builder()
                .setHost(accountModel.getJid().getDomain().toString())
                .setXmppAddressAndPassword(accountModel.getJid(), accountModel.getPassword())
                .setConnectTimeout(2 * 60 * 1000)
                .build();

        AbstractXMPPConnection tcpConnection = new XMPPTCPConnection(configuration);
        return new MercuryConnection(tcpConnection, accountModel.getId());
    }
}