CODEWITHSUNDEEP
javacollectionsjava-8java-streamcollectors
karthik selvaraj
karthik selvaraj

Reputation: 499

Map with LIst using Collectors.toMap in Java 8

I like convert below code to java stream,

HashMap<String, List<Data>> heMap = new HashMap<String, List<Data>>();
for (Data heData : obj) {
    String id = heData.getData().getId() + heData.getPlanData().getCode()
            + heData.getPlanData().getId();
    if (!heMap.containsKey(id)) {
        CitizenHElist = new ArrayList<Data>();

        CitizenHElist.add(heData);
        heMap.put(id, CitizenHElist);

    } else {
        heMap.get(id).add(heData);
    }
}

I tried the below code using stream, but i am not succeed on this.

heMap=obj.stream().collect(Collectors.toMap(t->getKey(t), obj.stream().collect(Collectors.toList())));

private String getKey(Data heData){
    String id = heData.getData().getId() + heData.getPlanData().getCode()
                    + heData.getPlanData().getId();
    return id;
}

Upvotes: 5

Views: 15267

Answers (3)

Zigri2612
Zigri2612

Reputation: 2310

Grouping on the bases of a field - 

import java.util.*;
import java.util.stream.*;
public class Main
{
  public static void main (String[]args)
  {
    System.out.println ("Hello World");

    List < Data > dataList = getDataList();

    System.out.println (dataList);

    Map < String, List < Data >> dataMap =
      dataList.stream ().collect (Collectors.groupingBy (d->d.code));

    System.out.println (dataMap);
  }

 static List < Data > getDataList(){
    List < Data > dataList = new ArrayList <> ();
    dataList.add (new Data (1, "Prince", "102"));
    dataList.add (new Data (2, "Rahul", "102"));
    dataList.add (new Data (3, "Sunny", "103"));
    dataList.add (new Data (4, "Mitul", "104"));
    dataList.add (new Data (5, "Amit", "105"));
    dataList.add (new Data (6, "Ashish", "105"));
    return dataList;
  }
}

class Data
{
   int id;
   String name;
   String code;
  public Data (int id, String name, String code)
  {
    this.id = id;
    this.name = name;
    this.code = code;
  }
  public String toString ()
  {
    return String.format ("id:%s,name:%s,code:%s", id, name, code);
  }
}

Upvotes: 1

Chirag
Chirag

Reputation: 1508

not sure your data structure but you want to do something like below, which is working.

import java.util.*;
import java.util.stream.Collectors;
class Data {

    String stud_id;
    String stud_name;
    String stud_location;
    public Data(String string, String string2, String string3) {
    this.stud_id=string;
    this.stud_location=string2;
    this.stud_name=string3;
    }
    public Object getData() {
        return this.stud_id;
    }


}

class Temp3
{
    public static void main(String args[])
    {

        Map<String, List<Data>> heMap=new HashMap<String, List<Data>>();
        Data data1=new Data("1","11","111");
        Data data2=new Data("2","22","222");
        List<Data> obj=new ArrayList<Data>();
        obj.add(data1);
        obj.add(data2);
        for (Data heData : obj) 
        {
                    String id = "2";
                    if (!heMap.containsKey(id)) 
                    {
                        ArrayList<Data> CitizenHElist = new ArrayList<Data>();
                        CitizenHElist.add(heData);
                        heMap.put(id, CitizenHElist);
                    } 
                    else 
                    {
                        heMap.get(id).add(heData);

                    }

        }
        heMap=obj.stream().collect(Collectors.groupingBy(w -> w.stud_location));
        System.out.println(heMap);

    }
}

Upvotes: 0

Misha
Misha

Reputation: 28133

This is the job for groupingBy collector:

import static java.util.stream.Collectors.groupingBy;

Map<String, List<Data>> heMap = obj.stream().collect(groupingBy(d -> getKey(d)));

Note that this will use some unspecified implementations of Map and List. Currently, it happens to be HashMap and ArrayList, but that might change in the future.

Upvotes: 19

Related Questions