Mercury-IM/domain/src/main/java/org/mercury_im/messenger/Messenger.java

package org.mercury_im.messenger;

import org.mercury_im.messenger.transport.connection.ConnectionMethod;
import org.mercury_im.messenger.entity.Account;

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

public class Messenger {

    private final Map<Long, ConnectionMethod> connections = new HashMap<>();

    public void addConnection(ConnectionMethod connection) {
        connections.put(connection.getAccount().getId(), connection);
    }

    public ConnectionMethod getConnection(Account account) {
        return connections.get(account.getId());
    }
}