Listに重複する名前があったら値を変えたい

Question

前提・実現したいこと

あるフォルダにあるファイル名を元にリストを作成したいです。
ファイル名は

20200227_0001_A.text

のようになっていて、(日付)(ID)(型).txtのように規則的につけられています。

フォルダに
・20200225_0001_A.txt
・20200226_0001_B.txt
・20200227_0001_A.txt
・20200227_0001_B.txt
・20200227_0002_A.txt
・20200227_0002_C.txt
とファイルがあった時に関する質問です。

該当のソースコード

public String ok() {
	
	List<TestInfoBean> list = new ArrayList<TestInfoBean>();

	// ファイル置き場
	String path = "C:\tmp\test\_test";
	File dir = new File(path);                                              
	File files[] = dir.listFiles();
	// ファイル数分繰り返し
	for(int i=0; i<files.length; i++){                                      
		// ファイル名取得
		String fileName = files[i].getName();
		// ファイル名分割
		String[] param = fileName.split("_");
		
		// TestInfoBeanにセット
		TestInfoBean info = new TestInfoBean();
		info.setHiduke(param[0]);
		info.setId(param[1]);
		
		String type = param[2].substring(0, 1);
		switch(type) {
		case "A":
			info.setTypeA(true);
			break;
		case "B":
			info.setTypeB(true);
			break;
		case "C":
			info.setTypeC(true);
			break;
		}
		
		list.add(info);
	}                                                

	return "ok";
}

@Data
public class TestInfoBean {
	private String hiduke;
	private String id;
	private boolean typeA;
	private boolean typeB;
	private boolean typeC;
}

発生している問題・エラーメッセージ

今のままだとlistには6つの要素があるのですが
本当は日付とIDが同じだった場合のinfoを一つにまとめてlistに追加したいです。
例）
・20200227_0001_A.txt
・20200227_0001_B.txt
は日付とIDが同じのため

info.hidukeが20200227、
info.idが0001、
info.typeAがtrue、
info.typeBがtrue、
info.typeCがfalse

のように一つのinfoとして扱いたいです。(今回のファイルの場合、合計のlistの要素を4つにしたいです)

しばらく考えてはみたのですがどうやって比較すれば良いかがわからず…
わかる方がいたら教えてください。

Comments

[jimbe]

ok メソッドの中で list は貯めるだけで使用されていませんが, List である必要はあるのでしょうか.

Answer 1

簡単(?)には Map を使います.
一応最後に List に入れ直していますが, 必要無ければ map.values() を使えばよろしいかと思います.

import java.io.File;
import java.util.*;

public class Q243809 {

  public static void main(String[] args) {
    ok();
  }

  static class TestInfoBean {
    private String hiduke;
    private String id;
    private boolean typeA;
    private boolean typeB;
    private boolean typeC;
    void setHiduke(String s) { hiduke=s; }
    void setId(String s) { id=s; }
    void setTypeA(boolean f) { typeA=f; }
    void setTypeB(boolean f) { typeB=f; }
    void setTypeC(boolean f) { typeC=f; }
    public String toString() {
      return "[hiduke="+hiduke+",id="+id+
        ",typeA="+typeA+
        ",typeB="+typeB+
        ",typeC="+typeC+"]";
    }
  }

  public static String ok() {

    Map<String,TestInfoBean> map = new HashMap<>();

    // ファイル置き場
    String path = "C:\tmp\test\_test";
    //String path = "tmp";
    File dir = new File(path);
    File files[] = dir.listFiles();
    // ファイル数分繰り返し
    for(int i=0; i<files.length; i++){
      // ファイル名取得
      String fileName = files[i].getName();
      // ファイル名分割
      String[] param = fileName.split("_");

      TestInfoBean info = map.get(param[0]+"_"+param[1]);
      if(info == null) {
        info = new TestInfoBean();
        info.setHiduke(param[0]);
        info.setId(param[1]);
        map.put(param[0]+"_"+param[1], info);
      }

      String type = param[2].substring(0, 1);
      switch(type) {
      case "A":
        info.setTypeA(true);
        break;
      case "B":
        info.setTypeB(true);
        break;
      case "C":
        info.setTypeC(true);
        break;
      }
    }

    List<TestInfoBean> list = new ArrayList<>(map.values());

    for(TestInfoBean info : list) System.out.println(info);

    return "ok";
  }
}

[hiduke=20200225,id=0001,typeA=true,typeB=false,typeC=false]
[hiduke=20200227,id=0001,typeA=true,typeB=true,typeC=false]
[hiduke=20200227,id=0002,typeA=true,typeB=false,typeC=true]
[hiduke=20200226,id=0001,typeA=false,typeB=true,typeC=false]

Answer 2

特に指定がなかったですがjava8対応の書き方です。
continueラベルとか使ってて美しくはありませんが・・・。

List<TestInfoBean> list = new ArrayList<TestInfoBean>();

// ファイル置き場
String path = "C:\tmp\test\_test";
File dir = new File(path);
File files[] = dir.listFiles();
// ファイル数分繰り返し
file_loop: for(int i=0; i<files.length; i++){
    // ファイル名取得
    String fileName = files[i].getName();
    // ファイル名分割
    String[] param = fileName.split("_");

    Consumer<TestInfoBean> setType = info -> {
        String type = param[2].substring(0, 1);
        switch(type) {
        case "A":
            info.setTypeA(true);
            break;
        case "B":
            info.setTypeB(true);
            break;
        case "C":
            info.setTypeC(true);
            break;
        }
    };

    // リストをループする
    for (TestInfoBean info : list) {
        // 日付とID両方が一致する場合
        if (info.getHiduke().equals(param[0])
                && info.getId().equals(param[1])) {
            // リスト内のTestInfoBeanにTypeを設定
            setType.accept(info);
            // ファイルループに戻って処理を繰り返す
            continue file_loop;
        }
    }

    // TestInfoBeanにセット
    TestInfoBean info = new TestInfoBean();
    info.setHiduke(param[0]);
    info.setId(param[1]);
    setType.accept(info);

    list.add(info);
}

// リスト内の確認(デバッグコード)
list.forEach(l ->
    System.out.println(
        "hiduke=" + l.getHiduke() +
        "、id=" + l.getId() +
        "、typeA=" + l.isTypeA() +
        "、typeB=" + l.isTypeB() +
        "、typeC=" + l.isTypeC()));

Comments

[moshi]

回答ありがとうございます。
上記でやりたいことはできました。助かります。
もし可能であれば少し教えていただきたいのですが
Consumer<TestInfoBean> setType = info -> {
と
setType.accept(info);
のところは何をしているのでしょうか・・・調べてみたけどよくわからなく・・・

continue file_loop;はfile_loopと名前を付けたループ処理にcontinueするというのはわかるのですが
お時間ありましたら回答お願いします

[yureighost]

その辺りのコードはJava8で追加されたメソッドを変数として定義できる機能です。
Consumerは引数を1つ取りつつ返り値は返さない。
メソッドで言うとvoid型の物と同等になります。
acceptはその定義した変数の処理を実行するメソッドです。
「java8 関数型インターフェース」などで検索すると解説してるサイトがありますので調べてみてください。