A conversão de uma lista de valores de objectos a grupo

StonecoldIM:

Eu tenho o seguinte pedaço de código

OrderCriteria o1 = new OrderCriteria(1, 1, 101, 201);
OrderCriteria o2 = new OrderCriteria(1, 1, 102, 202);
OrderCriteria o4 = new OrderCriteria(1, 1, 102, 201);
OrderCriteria o5 = new OrderCriteria(2, 2, 501, 601);
OrderCriteria o6 = new OrderCriteria(2, 2, 501, 602);
OrderCriteria o7 = new OrderCriteria(2, 2, 502, 601);
OrderCriteria o8 = new OrderCriteria(2, 2, 502, 602);
OrderCriteria o9 = new OrderCriteria(2, 2, 503, 603);

Onde OrderCriteriaparece com abaixo:

public class OrderCriteria {
    private final long orderId;
    private final long orderCatalogId;
    private final long procedureId;
    private final long diagnosisId;

    public OrderCriteria(long orderId, long orderCatalogId, long procedureId, long diagnosisId) {
        this.orderId = orderId;
        this.orderCatalogId = orderCatalogId;
        this.procedureId = procedureId;
        this.diagnosisId = diagnosisId;
    }

    // Getters
}

O que eu quero é para obter uma lista de procedimentos e lista de diagnósticos agrupados por ID de ordem. Portanto, ele deve retornar:

{1, {101, 102}, {201, 202}}
{2, {501, 502, 503}, {601, 602, 603}}

o que significa Ordem com id 1 está tendo procedimento ids 101, 102 e ids de diagnóstico 201, 202 etc. Eu tentei usando a tabela google goiaba, mas não conseguiu chegar a qualquer solução válida.

shmosel:

Primeiro você vai precisar de uma nova estrutura para armazenar os dados agrupados:

class OrderCriteriaGroup {
    final Set<Long> procedures = new HashSet<>();
    final Set<Long> diagnoses = new HashSet<>();

    void add(OrderCriteria o) {
        procedures.add(o.getProcedureId());
        diagnoses.add(o.getDiagnosisId());
    }

    OrderCriteriaGroup merge(OrderCriteriaGroup g) {
        procedures.addAll(g.procedures);
        diagnoses.addAll(g.diagnoses);
        return this;
    }
}

add()e merge()são métodos de conveniência que nos ajudarão a transmitir e recolher os dados, assim:

Map<Long, OrderCriteriaGroup> grouped = criteriaList.stream()
        .collect(Collectors.groupingBy(OrderCriteria::getOrderId,
                Collector.of(
                        OrderCriteriaGroup::new,
                        OrderCriteriaGroup::add,
                        OrderCriteriaGroup::merge)));