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

package org.mercury_im.messenger;

import org.mercury_im.messenger.data.repository.AccountRepository;
import org.mercury_im.messenger.data.repository.PeerRepository;
import org.mercury_im.messenger.entity.contact.Peer;
import org.mercury_im.messenger.transport.connection.ConnectionMethod;
import org.mercury_im.messenger.entity.Account;

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

import javax.inject.Inject;

public class Messenger {

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

    @Inject
    AccountRepository accountRepository;

    @Inject
    PeerRepository peerRepository;

    @Inject
    public Messenger() {

    }

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

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