Javaで同一IDのフィールドをマージする。

久しぶりにゴリゴリとJavaを書いている。

そんな中、以下の様な要件があってJavaで対応することになった。

今回のお題

①テーブル構造はidとnameのみで両方合わせて一意（PK)である

②同一idのnameをまとめてカンマ区切りで表示したい。

データの例としては以下のイメージ。

id	name
1	ABC
1	DEF
2	ABC
3	ABC
3	DEF
4	ABC
5	ABC
6	ABC
7	ABC
7	DEF
7	GHI
8	ABC
8	ABC
9	ABC
10	ABC
10	DEF

これを以下のようにしたいのだ。

1 : ABC,DEF
2 : ABC
3 : ABC,DEF
4 : ABC
5 : ABC
6 : ABC
7 : ABC,DEF,GHI
8 : ABC,ABC
9 : ABC
10 : ABC,DEF

これのロジックを考えるのにハマったのだ。

考えた結果

できなかったのだけど、１つ１つ考えてみて整理した。

１．初回データは前回のデータとして保持する。

２．前回のデータとIDが同じ場合、nameを結合して追加

３．前のデータとIDが違う場合、前のデータを登録（結果として出力）する

４．最後のデータは登録（結果として出力）する

この考えてソースを書いてみた。

まず、データを保持しているItemBean.java。実質DBから取得したデータだと思ってもらいたい。

1. public class ItemBean {
2.     private int id;
4.     private String name;
6.     // コンストラクタ
7.     public ItemBean(int id, String name){
8.         this.id = id;
9.         this.name = name;
10.     }
12.     public int getId() {
13.         return id;
14.     }
16.     public void setId(int id) {
17.         this.id = id;
18.     }
20.     public String getName() {
21.         return name;
22.     }
24.     public void setName(String name) {
25.         this.name = name;
26.     }
29. }

次に、結果を出力するためのResultBean.java

1. public class ResultBean {
3.     private int id;
5.     private String name;
7.     // コンストラクタ
8.     public ResultBean(){
9.     }
11.     public int getId() {
12.         return id;
13.     }
15.     public void setId(int id) {
16.         this.id = id;
17.     }
19.     public String getName() {
20.         return name;
21.     }
23.     public void setName(String name) {
24.         this.name = name;
25.     }
26. }

最後に上記２つを使ったメインのロジックと出力クラス。

1. import java.util.ArrayList;
2. import java.util.List;
5. public class Test1_1 {
7.     /**
8.      * @param args
9.      */
10.     public static void main(String[] args) {
11.         // 登録されているデータ
12.         List<ItemBean> dataList = new ArrayList<ItemBean>();
13.         dataList.add(new ItemBean(1,"ABC"));
14.         dataList.add(new ItemBean(1,"DEF"));
15.         dataList.add(new ItemBean(2,"ABC"));
16.         dataList.add(new ItemBean(3,"ABC"));
17.         dataList.add(new ItemBean(3,"DEF"));
18.         dataList.add(new ItemBean(4,"ABC"));
19.         dataList.add(new ItemBean(5,"ABC"));
20.         dataList.add(new ItemBean(6,"ABC"));
21.         dataList.add(new ItemBean(7,"ABC"));
22.         dataList.add(new ItemBean(7,"DEF"));
23.         dataList.add(new ItemBean(7,"GHI"));
24.         dataList.add(new ItemBean(8,"ABC"));
25.         dataList.add(new ItemBean(8,"ABC"));
26.         dataList.add(new ItemBean(9,"ABC"));
27.         dataList.add(new ItemBean(10,"ABC"));
28.         dataList.add(new ItemBean(10,"DEF"));
30.         // 結果取得用のリスト
31.         List<ResultBean> resultList = new ArrayList<ResultBean>();
32.         ResultBean result = null;
33.         ResultBean beforeResult = null;
36.         for(ItemBean bean : dataList){
37.             result = new ResultBean();
39.             result.setId(bean.getId());
40.             result.setName(bean.getName());
42.             // 初回
43.             if(beforeResult == null){
44.                 // 前のデータとして保持
45.                 beforeResult = result;
46.             } else if(beforeResult.getId() == result.getId()){//前のIDと同じ時
47.                 beforeResult.setName(beforeResult.getName() + "," + result.getName());
48.             } else if(beforeResult.getId() != result.getId()){//前のIDと違う時
49.                 // 前のデータを登録する
50.                 resultList.add(beforeResult);
51.                 // 前のデータとして保持
52.                 beforeResult = result;
53.             }
54.         }
56.         if(beforeResult != null){
57.             resultList.add(beforeResult);
58.         }
60.         for(ResultBean rslt : resultList){
61.             System.out.println(rslt.getId() + " : " + rslt.getName());
62.         }
65.     }
67. }

この出力結果は想定通りとなっている。

１．初回データは前回のデータとして保持する。⇒41行目のif文

２．前回のデータとIDが同じ場合、nameを結合して追加　⇒46行目のif文

３．前のデータとIDが違う場合、前のデータを登録（結果として出力）する　⇒48行目のif文

４．最後のデータは登録（結果として出力）する　⇒56行目のif文

前のデータの保持は回している中の最後にして、前のIDと同じときは値を設定したらcontinueするのも考えたけど、やりたいこととプログラムを一致させる意味で同じ処理を2回書いている。 

こんなんに時間使っているようではまだまだだぜ。

 こちらからは以上です。

スッキリわかるJava入門 第2版

12時間でわかるJ2EE実践ガイド (J2EE & BEA WebLogic Server)