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

package org.mercury_im.messenger.xmpp_core;

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

public class ConnectionCenter {

    private static ConnectionCenter INSTANCE;

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

    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);
    }
}