MercadoEficiente.java

package br.com.atentatecnologia.mercado;

import java.util.*;
import java.util.stream.Collectors;

public class MercadoEficiente implements Mercado {

	private Map<String, PosicaoVendedor> mapaFonte = new HashMap<>();
	private Set<PosicaoVendedor> melhoresTaxas = new TreeSet<>((o1, o2) -> Double.compare(o2.getTaxa(), o1.getTaxa()));

	@Override
	public void atualizar(PosicaoVendedor pos) {
		PosicaoVendedor posAnterior = mapaFonte.put(pos.getFonte(), pos);
		if (posAnterior != null) {
			melhoresTaxas.remove(posAnterior);
		}
		melhoresTaxas.add(pos);
	}

	@Override
	public List<PosicaoVendedor> melhoresPosicoes(int quantidadePosicoesEsperada) {
		return melhoresTaxas.stream().limit(quantidadePosicoesEsperada).collect(Collectors.toList());
	}

}