使用 Java8 的 stream()、collect() 和 groupingBy() 等根据两个相同的 beans 属性值添加 integer 值

Question

我有一个豆子：

class Foo { 
  private String category;
  private int qty;
  private int price;
}

现在我有一个清单：

  List foos = new ArrayList();
  foos.add(new Foo("A", 1, 10));
  foos.add(new Foo("A", 2, 10));
  foos.add(new Foo("A", 2, 20));
  foos.add(new Foo("B", 1, 10));
  foos.add(new Foo("C", 1, 30));
  foos.add(new Foo("C", 5, 10));

我想添加类别和数量相同的价格，例如两个相同的A类别有两个相同的数量(Foo("A", 2, 10), new Foo("A", 2, 20))，然后是另一个或相同的列表应为“A 1 10”和“A 2 30”

Answer 1

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;

public class Main {

    private static <T> Predicate<T> distinctByKeys(Function<? super T, ?>... keyExtractors) {
        final Map<List<?>, Boolean> seen = new ConcurrentHashMap<>();

        return t -> {
            final List<?> keys = Arrays.stream(keyExtractors).map(ke -> ke.apply(t)).collect(Collectors.toList());

            return seen.putIfAbsent(keys, Boolean.TRUE) == null;
        };
    }

    public static void main(String[] args) {
        List<Foo> foos = new ArrayList<>();
        List<Foo> resultfoos = new ArrayList<>();
        foos.add(new Foo("A", 1, 10));
        foos.add(new Foo("A", 2, 10));
        foos.add(new Foo("A", 2, 20));
        foos.add(new Foo("B", 1, 10));
        foos.add(new Foo("C", 1, 30));
        foos.add(new Foo("C", 5, 10));


        foos.stream().filter(distinctByKeys(Foo::getCategory, Foo::getQty)).forEach(o -> {

            o.setPrice(foos.stream().distinct()
                    .filter(x -> x.getCategory().equalsIgnoreCase(o.getCategory()) && x.getQty() == o.getQty())
                    .mapToInt(t -> t.getPrice()).sum());

            System.out.println(o.toString());

            resultfoos.add(o), o.getPrice()));
        });

    }

}

class Foo {
    private String category;
    private int qty;
    private int price;

    Foo(String category, int qty, int price) {
        this.category = category;
        this.qty = qty;
        this.price = price;

    }

    public String getCategory() {
        return category;
    }

    public void setCategory(String category) {
        this.category = category;
    }

    public int getQty() {
        return qty;
    }

    public void setQty(int qty) {
        this.qty = qty;
    }

    public int getPrice() {
        return price;
    }

    public void setPrice(int price) {
        this.price = price;
    }

    @Override
    public String toString() {
        return "Foo [category=" + category + ", qty=" + qty + ", price=" + price + "]";
    }

}