Java中stream處理中map與flatMap的比較和使用案例

Posted on by

前言

使用Java8的新特性Stream流式處理，可以提高對於集合的一些操作效率，再配合lambda表達式，可以極致的簡化代碼，尤其還有並行流這個東東，可以去 瞭解一下，在一些場合還是可以提高效率的，而且編碼起來也不費事。

並且流式處理的核心就是一個淺拷貝和引用管道，其內部實現瞭一個引用管道ReferencePipeline， 他把需要處理的數據的引用拷貝瞭一份，然後處理數據，最後收集結果也是將這些引用放到瞭另一個集合中。

今天要講的就是stream處理中的map和flatMap這倆個的比較和使用

共同點和區別

首先看一下源代碼

  <R> Stream<R> map(Function<? super T, ? extends R> mapper);
   <R> Stream<R> flatMap(Function<? super T, ? extends Stream<? extends R>> mapper);

共同點

  • 都是依賴stream進行轉換，結合lambda的入參和返回值，將一個類型依據程序邏輯轉換成另一種類型。
  • 最後結果返回都還是一個流，還可以對其進行繼續的流式處理或者進行收集。

區別

  • 方法參數不同，第一個參數是一樣的，但是第二個不一樣，對於map第二個參數沒啥要求，但是使用flatMap時，第二個參數還得用流來接收。
  • 所以flatMap多用於多對多，一對多，也就是map是將一個數據流中的一個數據節點，映射成另外一個數據節點，而flatMap是將一個數據流中的一個數據節點映射成另外一個數據流，這個另外的數據流可以是一個數據節點也可以是多個數據節點。
  • flatmap既可以單一轉換也可以一對多/多對多轉換，flatmap要求返回Observable，因此可以再內部進行from/just的再次事件分發，一一取出單一對象（轉換對象的能力不同）

使用案例

比如我們使用map將一個per對象映射成一個字符串對象

  static class Per {
    public String name;
    public int age;

    public Per(String name, int age) {
      this.name = name;
      this.age = age;
    }

    public String getName() {
      return name;
    }

    public void setName(String name) {
      this.name = name;
    }

    public int getAge() {
      return age;
    }

    public void setAge(int age) {
      this.age = age;
    }

    @Override
    public String toString() {
      return "Per{" +
          "name='" + name + '\'' +
          ", age=" + age +
          '}';
    }
  }

  public static void main(String[] args) {
    List<Per> list = new ArrayList<>();
    list.add(new Per("listen", 22));
    list.add(new Per("bike", 24));
    list.add(new Per("milk", 27));

    List<String> collect = list.stream().map(Per::toString).collect(Collectors.toList());

    System.out.println(list);
    System.out.println("---");
    System.out.println(collect);

  }

一對一可以實現，那我們該需求，假設一個per人有多個孩子，我們想獲取這多個孩子，使用map可以實現嗎？比如下面代碼。

  static class Child {
    public String name;
    public int age;

    public Child(String name, int age) {
      this.name = name;
      this.age = age;
    }
        @Override
    public String toString() {
      return "Child{" +
          "name='" + name + '\'' +
          ", age=" + age +
          '}';
    }
  }

  static class Per {
    public String name;
    public int age;
    public Child[] Children;

    public Per(String name, int age) {
      this.name = name;
      this.age = age;
    }

    public Child[] getChildren() {
      return Children;
    }

    public void setChildren(Child[] children) {
      Children = children;
    }

    public String getName() {
      return name;
    }

    public void setName(String name) {
      this.name = name;
    }

    public int getAge() {
      return age;
    }

    public void setAge(int age) {
      this.age = age;
    }

    @Override
    public String toString() {
      return "Per{" +
          "name='" + name + '\'' +
          ", age=" + age +
          '}';
    }
  }

這個時候註意這個孩子是多個，我們用的是數組存儲。如果此時我們還用map，那麼隻能接受每個per的孩子數組，不能和我們所願接受每個孩子。

  public static void main(String[] args) {
    List<Per> list = new ArrayList<>();
    Per per1 = new Per("Listen", 22);
    per1.setChildren(new Child[] {new Child("a", 1), new Child("b", 2)});
    Per per2 = new Per("Milk", 26);
    per2.setChildren(new Child[] {new Child("c", 1), new Child("d", 2)});
    list.add(per1);
    list.add(per2);

    List<Child[]> collect = list.stream().map(Per::getChildren).collect(Collectors.toList());
    collect.forEach(item -> System.out.println(Arrays.toString(item)));
  }

而使用flatMap就可以實現。

  public static void main(String[] args) {
    List<Per> list = new ArrayList<>();
    Per per1 = new Per("Listen", 22);
    per1.setChildren(new Child[] {new Child("a", 1), new Child("b", 2)});
    Per per2 = new Per("Milk", 26);
    per2.setChildren(new Child[] {new Child("c", 1), new Child("d", 2)});
    list.add(per1);
    list.add(per2);

    List<Child> collect = list.stream().flatMap(item -> Arrays.stream(item.getChildren())).collect(Collectors.toList());
    System.out.println(collect);

  }

到此這篇關於Java中stream處理中map與flatMap的比較和使用案例的文章就介紹到這瞭,更多相關Java map與flatMap內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：