package chapter1;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

public class stream {
  //比如說 你需要在一個(gè)列表中 篩選出所以重量大于150的蘋果，然后按照顏色分組
  //按常規(guī)的篩選寫法 就是在循環(huán)里面 迭代篩選

  public static void main(String[] args) {
    List<Apple> appleList = new ArrayList<>();

    //常規(guī)寫法
    Map<String, List<Apple>> AppMap = new HashMap<>();
    for (Apple apple : appleList) {
      if (apple.getWeight() > 150) { //如果重量大于150
        if (AppMap.get(apple.getColor()) == null) { //該顏色還沒分類
          List<Apple> list = new ArrayList<>(); //新建該顏色的列表
          list.add(apple);//將蘋果放進(jìn)去列表
          AppMap.put(apple.getColor(),list);//將列表放到map中
        }else { //該顏色分類已存在
          AppMap.get(apple.getColor()).add(apple);//該顏色分類已存在，則直接放進(jìn)去即可
        }
      }
    }

    //如上方式 就可以篩選出來所有的150克大小以上的蘋果，并按顏色分類



    //方式二 使用java8提供的流api實(shí)現(xiàn) 這種叫內(nèi)部迭代
    Map<String, List<Apple>> AppMap2=appleList.stream().filter((Apple a)->a.getWeight()>150) //篩選出大于150的
        .collect(groupingBy(Apple::getColor)); //按顏色分組 最后得到map
    

  }


  class Apple {

    private String color;//顏色
    private Integer weight; //重量

    public String getColor() {
      return color;
    }

    public void setColor(String color) {
      this.color = color;
    }

    public Integer getWeight() {
      return weight;
    }

    public void setWeight(Integer weight) {
      this.weight = weight;
    }
  }
}