java stream : get sum of field in nested list of objects

Question

From a List containing a List , how do i get the sum of a specific field in Tax class after filtering List based on a field in Item class?

This is what I have done so far -

public class Item {
    private long itemClass;
    private List<Tax> taxList;

    public long getItemClass() {
        return itemClass;
    }
    public void setItemClass(long itemClass) {
        this.itemClass = itemClass;
    }
    public List<Tax> getTaxes() {
        return taxList;
    }
    public void setTaxList(List<Tax> taxList) {
        this.taxList = taxList;
    }

}


public class Tax {
    private double taxRate;

    public double getTaxRate() {
        return taxRate;
    }
    public void setTaxRate(double taxRate) {
        this.taxRate = taxRate;
    }
}


public class App {
    public static void main(String[] args) {
        Item item1 = new Item();
        item1.setItemClass(100);

        Item item2 = new Item();
        item1.setItemClass(200);

        Item item3 = new Item();
        item1.setItemClass(300);

        Tax tax1 = new Tax();
        tax1.setTaxRate(0.01);

        Tax tax2 = new Tax();
        tax1.setTaxRate(0.02);

        Tax tax3 = new Tax();
        tax1.setTaxRate(0.03);

        item1.setTaxList(Arrays.asList(tax1, tax2));
        item2.setTaxList(Arrays.asList(tax1, tax3));
        item3.setTaxList(Arrays.asList(tax1, tax2, tax3));

        List<Item> itemList = Arrays.asList(item1, item2, item3);

        List<Item> filteredItemList = itemList.stream().filter(i -> i.getItemClass() != 200).collect(Collectors.toList());
        List<Tax> filteredTaxList = filteredItemList.stream().map(i -> i.getTaxes()).flatMap(t -> t.stream()).collect(Collectors.toList());
        Double totalTaxRate = filteredTaxList.stream().mapToDouble(t -> t.getTaxRate()).sum();

        System.out.println(totalTaxRate); //expected result 0.09
    }
}

I am able to get the desired result. But 3 streams seems to be overkill. How do I make this more efficient?

Answer 1

The equivalent one-liner :

Double totalTaxRate = itemList.stream()
                .filter(i -> i.getItemClass() != 200)
                .flatMap(i -> i.getTaxes().stream())
                .mapToDouble(t -> t.getTaxRate()).sum();