Java 8 Stream API toMap converts to TreeMap

public class Message {
    private int id;
    private User sender;
    private User receiver;
    private String text;   
    private Date senddate;
..
}

I have

List<Message> list= new ArrayList<>();

I need to convert them to

TreeMap<User,List<Message>> map

I know how to make a conversion to HashMap using

list.stream().collect(Collectors.groupingBy(Message::getSender));

But I need TreeMap with: Key is the first user with the first senddate message. Value - A list sorted by the first first address of senddate

Custom Class Part

    public class User{
    ...
    private List<Message> sendMessages;
    ...

   public List<Message> getSendMessages() {
        return sendMessages;
    }

}

Custom Comparator:

     public class Usercomparator implements Comparator<User> {
        @Override
        public int compare(User o1, User o2) {
            return o2.getSendMessages().stream()
.map(message -> message.getSenddate())
.max(Date::compareTo).get()
           .compareTo(o1.getSendMessages().stream()
.map(message1 -> message1.getSenddate())
.max(Date::compareTo).get());
        }
    }
+4
source share
1 answer

You can use the overloaded method groupingByand pass TreeMapas Supplier:

TreeMap<User, List<Message>> map = list
            .stream()
            .collect(Collectors.groupingBy(Message::getSender,
                    () -> new TreeMap<>(new Usercomparator()), toList()));
+13
source

Source: https://habr.com/ru/post/1626760/


All Articles