How to easily implement conditional count in groupby using Java Stream

Question

I have the following data(A class structure):

[
  {
    "name": "a",
    "enable": true
  },
  {
    "name": "a",
    "enable": false
  },
  {
    "name": "b",
    "enable": false
  },
  {
    "name": "b",
    "enable": false
  },
  {
    "name": "c",
    "enable": false
  }
]

How can I replace the complex code below with a simple steam:

List<A> list = xxxxx;
Map<String, Integer> aCountMap = new HashMap<>();
list.forEach(item -> {
    Integer count = aCountMap.get(item.getName());
    if (Objects.isNull(count)) {
        aCountMap.put(item.getName(), 0);
    }
    if (item.isEnable()) {
        aCountMap.put(item.getName(), ++count);
    }
});

The correct result is as follows:

{"a":1}
{"b":0}
{"c":0}

Naman · Accepted Answer · 2020-03-20 04:17:32Z

3

You might simply be counting the objects filtered by a condition:

Map<String, Long> aCountMap = list.stream()
        .filter(A::isEnable)
        .collect(Collectors.groupingBy(A::getName, Collectors.counting()));

But since you are looking for the 0 count as well, Collectors.filtering with Java-9+ provides you the implementation such as :

Map<String, Long> aCountMap = List.of().stream()
        .collect(Collectors.groupingBy(A::getName,
                Collectors.filtering(A::isEnable,
                        Collectors.counting())));

edited Mar 20, 2020 at 4:17

answered Mar 20, 2020 at 4:11

Naman

32.7k32 gold badges240 silver badges385 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

赵彬杰 Over a year ago

Is there a similar approach for Java 8？

Naman Over a year ago

@赵彬杰 This answer shares an approach of how you can create a static method in Java-8 as well.

Collectives™ on Stack Overflow

How to easily implement conditional count in groupby using Java Stream

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related