JAVA Stream APIを使用してDB参照データをリクエストIDの順番で出力したいです

Question

前提

JAVAで以下のような実装をしているのですが、こちらをfor文を使用せずstream一本で
おそらくflatMap等、使用して実装できるかと思うのですが、具体的にどのように実装したら良いのでしょうか？

for (ID id : idLists) {
  returnList.addAll(
      idDatas.stream()
          .filter(data -> data.getId().equals(id))
          .collect(Collectors.toList()));
}

idListはリクエストされたList<ID>のリストで
idDatasはidListを使用しDBから取得したList<dataEntity>となっております。
※dataEntityはidとnameだけのクラスです。

最終的に、idDatasを出力するのですが、idListを使用しDBから取得するとリクエストされた順番とは異なってしまいます。
（↑SQL実行でリクエストパラメータ順にorder by を指定する方法は無しで考えたいです。）
ですので出力前に上記のような処理をかませ並び替えております。

実現したいこと

• stream API一本で記載したい
• リクエストされたidListの順でDB参照値のidDatasが出力されること

試したこと

以下のような記載で実現できるかと考えているのですが、具体的にどのように記載すればよいのか分からずエラーとなっています

returnList.addAll(
    idDatas.stream()
        .flatMap(x -> idLists.stream().filter(id -> x.getId().equals(id))
        .collect(Collectors.toList()));

Comments

[jimbe]

>idListはリクエストされたList<ID>のリストで ～

データベースは直接は関係無さそうですし、構造を言葉で表すよりはコードで示されたほうが確実です。
必要なクラスと必要なリストに必要な(ダミー)データを入れ、ストリームのコードを実行した結果どのようになる(表示処理を入れれば実行したらこう表示される)のか、入力・処理・出力のうち入力と出力がハッキリすればコードの成否もハッキリします。
(stream である必要も無いかもしれませんし。)

[ksppy3500]

ありがとうございます。
確かに言葉で説明するだけでは分かりづらく
入出力の例があるだけで、だいぶイメージしやすくなりますね！

Answer 1

とりあえず並び替えなら (for はともかく stream でも無く) List.sort で十分なのではということで。
(どうしても stream なら sorted 。)

import java.util.*;

public class Main {
  public static void main(String[] args) {
    List<Data> idDatas = new ArrayList<>();
    idDatas.add(new Data(1, "AAA"));
    idDatas.add(new Data(2, "BBB"));
    idDatas.add(new Data(3, "CCC"));
    idDatas.add(new Data(4, "DDD"));
    idDatas.add(new Data(5, "EEE"));
    idDatas.add(new Data(6, "FFF"));
    System.out.println("-- idDatas ------");
    idDatas.stream().forEach(System.out::println);

    List<Long> idLists = new ArrayList<>();
    for(Data d : idDatas) idLists.add(d.id);
    Collections.shuffle(idLists);
    System.out.println("-- idLists ------");
    idLists.stream().forEach(System.out::println);

    List<Data> returnList = new ArrayList<>(idDatas);
    returnList.sort((a,b)->idLists.indexOf(a.id)-idLists.indexOf(b.id));
    /*
    List<Data> returnList = idDatas.stream()
        .sorted((a,b)->idLists.indexOf(a.id)-idLists.indexOf(b.id))
        .collect(Collectors.toList());
    */
    System.out.println("-- returnList ------");
    returnList.stream().forEach(System.out::println);
  }

  private static class Data {
    final long id;
    final String name;
    Data(long id, String name) {
      this.id = id;
      this.name = name;
    }
    @Override
    public String toString() {
      return getClass().getSimpleName()+"[id="+id+",name="+name+"]";
    }
  }
}

実行結果

-- idDatas ------
Data[id=1,name=AAA]
Data[id=2,name=BBB]
Data[id=3,name=CCC]
Data[id=4,name=DDD]
Data[id=5,name=EEE]
Data[id=6,name=FFF]
-- idLists ------
3
1
6
2
5
4
-- returnList ------
Data[id=3,name=CCC]
Data[id=1,name=AAA]
Data[id=6,name=FFF]
Data[id=2,name=BBB]
Data[id=5,name=EEE]
Data[id=4,name=DDD]

Comments

[ksppy3500]

ご回答ありがとうございます！
このような簡単な方法が、あったのですね！
streamを使用する場合の実装例までありがとうございます。
とても助かりました🙇‍♂️