Java Stream filter by two values using collectors

Question

I have the below working code using one value to compare but want to compare with two values. I need to add one more filter as "getAddressID" along with getAction. Could some help me to modify the below code?

List<Data> finalData = dataList.stream().collect(
    Collectors.collectingAndThen(
        Collectors.toCollection(
            () -> new TreeSet<>(Comparator.comparing(Data::getAction))),
            ArrayList::new))

public class TestCollection {

/**
 * @param args
 */
public static void main(String[] args) {
    
    List<Data> dataList = new ArrayList<Data>();
    
    Data data1 = new Data("DELIVERED", "1112");
    Data data2 = new Data("DELIVERED", "1113");
    Data data3 = new Data("PICKEUP", "1112");
    Data data4 = new Data("PICKEUP", "1112");
    
    dataList.add(data1);
    dataList.add(data2);
    dataList.add(data3);
    dataList.add(data4);
    
    //Filer by Action
    List<Data> finalData = dataList.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<> (Comparator.comparing(Data::getAction))), ArrayList::new));
    
    for(Data data : finalData) {
        System.out.println(" Action " + data.getAction() + " Address ID " + data.getAddressID());
    }
    
    

}

}

The result is : "Action DELIVERED Address ID 1112" "Action PICKEUP Address ID 1112"

But I want result as: "DELIVERED Address ID 1112" "DELIVERED Address ID 1113" "PICKEUP Address ID 1112"

Can you explain a bit more in detail, what your code should do actually? Maybe also add an example of input and desired output. — cyberbrain
– cyberbrain, Commented Mar 16, 2022 at 17:11
Input is a list of Data objects. Data object has attributes inside of it. Above code is able to filetr by "Action" attrubute and returns a list. I just need to add one more filer to the code. I am going to write a smaple code and post in few minutes. — user3067524
– user3067524, Commented Mar 16, 2022 at 17:15
Does your Data class only have the two fields addressID & action or are there more? Are both fields strings like shown in your code or was is just an example? — Eritrean
– Eritrean, Commented Mar 16, 2022 at 17:59
Have you tried adding a second comparator, ie: Comparator.comparing(Data::getAction).thenComparing(Comparator.comparing(Data::otherMethod) — Zeke Rogers
– Zeke Rogers, Commented Mar 16, 2022 at 18:15
@Zeke Rogers. That's it. It worked. Thank you so much.Pleas put that as a solution. I will accept it. — user3067524
– user3067524, Commented Mar 16, 2022 at 18:21

Zeke Rogers · Accepted Answer · 2022-03-16 18:29:34Z

2

You can chain your comparators using thenComparing, ie

new TreeSet<>(Comparator.comparing(Data::getAction).thenComparing(Comparator.comparing(Data::otherMethod))

answered Mar 16, 2022 at 18:29

Zeke Rogers

1466 bronze badges

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

Comments

Eritrean · Accepted Answer · 2022-03-16 18:30:47Z

0

If I get it right, you are trying to get distinct items from your list by the combination of the two properties addressID & action. Here is a an accepted answer, thanks to @Stuart Marks, which works to get distinct elements by one property. That could be further modified to solve your use case like:

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;


....

List<Data> dataList = // your data objects

List<Data> distinctData = dataList.stream()
            .filter(distinctByKeys(Data::getAction, Data::getAddressID))
            .collect(Collectors.toList());

where the method distinctByKeys could look like

private static <T> Predicate<T> distinctByKeys(final 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;
    };
}

answered Mar 16, 2022 at 18:30

Eritrean

16.6k3 gold badges25 silver badges28 bronze badges

1 Comment

user3067524 Over a year ago

Your answer also look good to me. However, I just want to make a change to the exiting code. Thank you!

Collectives™ on Stack Overflow

Java Stream filter by two values using collectors

2 Answers 2

Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related