POJOをキーとして2つのオブジェクトをマージ

Question

お世話になっております。

以下のようにするとリストから任意のキーでマップに変換できると思います。

Map<String, List<ChildA>>map=list.stream().collect(Collectors.groupby(ChildA::getId));

このマップのキーをstringからParentにし、
かつ変換後mapに対してParentをキーにChildBをマージしたいのですが、レガシーなfor文を使う以外にステップ数を抑えて可読性が上がる方法はないでしょうか？

// 変換前のイメージ
class BeforeParent {
  private Parent parent;
  private ChildA childA; // 中身はBと同様
}
class childB {
  private Integer pk1; // parentのpk
  private String pk2; // parentのpk
  private String pk3;
  private String data1;
  private String data2;
}

// 変換後のイメージ
class After {
  private Parent parent;
  private List<ChildA> childA;
  private List<ChildB> childB;
}

上記のとおり、java8が使用可能です。

やりたいのは、
M:Nのデータを結合して一括抽出した後で、M:Oとなるデータを同様に抽出し、
Mごとにループ処理を行うということです。

追記
抽出したデータを編集することはありません(→イミュータブル？)。
equalsはオーバーライドせずに、parentが持つ複合キー(全て基本型)それぞれに対してforループ内でequalsチェックする想定でした。

→ふと、思いましたが、equalsをオンバーライドすれば上記のstreamで行けそう？(^^;

Comments

[xebme]

ParentはイミュータブルかつhashCode(),equals()が適切にオーバーライドされていないと、マージする際にふたつのParentの一致判定ができないと思われますが、それはできているという前提ですね。一方にキーエントリが存在して、他方に存在しない場合は想定しますか。できれば、追記をお願いします。

[lupus_dingo]

回答ありがとうございます。追記しました。

[xebme]

Parentがキーなので、<M:N>.forEach( .... <M:O>.merge() ) というような処理を行うなら、Parentの equals()/hashCode()がオーバーライドされていないと、ふたつのMapのキー（違うインスタンスの同じParentのキーどおし）を突合できないと思っているだけですが、Stringをキーにしておくことも含めて回答者が考えることだと理解しました。

[lupus_dingo]

そうですね、equalsをオーバーライドするだけの話な気がしてきました…

Answer 1

参考にした情報 Merging Two Maps with Java このなかの、Stream を結合する方法を参考にしました。入力は２つの Map 、出力は List です。

public static List<After> merge(Map<Parent,List<ChildA>> mapA, Map<Parent,List<ChildB>> mapB) {
  return Stream.concat(
      mapA.entrySet().stream().map(e -> new After(e.getKey(),e.getValue(),null)),
      mapB.entrySet().stream().map(e -> new After(e.getKey(),null,e.getValue()))
    ).collect(Collectors.groupingBy(e -> e.getParent()))
     .values().stream()
     .map(l -> l.stream().reduce(new After(),(a,b)-> new After(a,b)))
     .collect(Collectors.toList());
}

Afterにコンストラクタが必要なので追加しました。Parentの取得メソッドも追加。(フィールドが private なので)

class After {
  private Parent parent;
  private List<ChildA> childA;
  private List<ChildB> childB;

  public After(Parent parent, List<ChildA> childA, List<ChildB> childB) {
    this.parent = parent;
    this.childA = (childA == null) ? new ArrayList<>() : childA;
    this.childB = (childB == null) ? new ArrayList<>() : childB;
  }
  public After(After a, After b) { // reduce マージ用コンストラクタ
    this((a.parent!=null)?a.parent:b.parent,a.childA,a.childB);
    this.childA.addAll(b.childA);
    this.childB.addAll(b.childB);
  }
  public After() { this(null,null,null); } // reduce 初期値

  public Parent getParent() { return this.parent; } // Parent の取得

}

POJO をキーにするということなので、Parentの equals()/hahsCode() をオーバーライドしました。自作オブジェクトを HashMap に格納するには必要です。equals()/hahsCode() に必要なフィールドは、final 変更できません。変更すると HashMap から取り出せなくなるから。意味がわからなければ、ご自分で調べてください。

class Parent {
  private final Integer pk1; // parentのpk
  private final String pk2; // parentのpk
	
  public Parent(Integer pk1, String pk2) {
    this.pk1 = pk1;
    this.pk2 = pk2;
  }

  @Override
  public boolean equals(Object obj) {
    if (obj == this) {
      return true;
    }
    if (!(obj instanceof Parent)) {
      return false;
    }
    Parent right = (Parent)obj;
    if (!this.pk1.equals(right.pk1)) {
      return false;
    }
    if (!this.pk2.equals(right.pk2)) {
      eturn false;
    }
    return true;
  }
	
  @Override
  public int hashCode() {
    return pk1.hashCode() + pk2.hashCode();
  }

}

After のマージコンストラクター (2019-05-08)

Map.merge()を使うと map に格納する型が合わず型変換が必要になる。After のコンストラクターにマージ機能を持たせるほうが凝集度が高くなるし、畳み込みにより２件以上のマージができると判断。

キーオブジェクトの equals()/hashCode()、immutable (2019-05-08)

参考書は『Effective Java』 第１版〜第３版 のどれでもよい、読んでみてください。

Mapのキーについての議論
HashMap/Setにまつわる2つの事
Immutable objects and hashmap keys

Java の基礎として知ておくべき。調べたうえで疑問が残るなら、別の質問をポストしてください。

hashCode() を Effective Java にあわせる (2019-05-09)

J. Broch さんはこれがベストではないと断っていますが、Effective Java にあわせます。

  @Override
  public int hashCode() {
    int result = 17;
    result = result * 31 + pk1.hashCode();
    result = result * 31 + pk2.hashCode();
    return result;
  }

pk1、pk2 はプライマリキーなので null になりえないことが前提です。また final な変数ゆえに１度だけ計算してキャッシュすることもできます。

Comments

[lupus_dingo]

回答ありがとうございます。
非常に参考になりました。
effective javaは本棚の肥やしにしているので引っ張りだして読んでみます。