Javaのコードの標準化がうまくできない

Question

前提・実現したいこと

Springboot + ThymeleafでWebページを作成したいです。
Thymeleafで表示したい項目にJavaのTreeNodeを使用したものがあるのですが
その標準化がうまくできません。

試したこと

日付を親ノード、ファイル名を子ノードとするTreeNodeを作成したいです。
試してみて固定値を適当に入れた場合下記のような記述でTreeNode[] nodeが作成されることが分かりました。

TreeNode[] node = {
		new TreeNode("20200101", new TreeNode[] {
                new TreeNode("aaa.txt"),
                new TreeNode("bbb.txt")
        }),
        new TreeNode("20200301", new TreeNode[] {
                new TreeNode("ccc.txt")
        })
};

上記のサンプルコードだと固定の値しか入れられないので親の日付をkeyにしてファイル名のリストをValueとしてMapで値を持つようにしてみたのですが

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

List<String> fileList1 = new ArrayList<String>();
fileList1.add("aaa.txt");
fileList1.add("bbb.txt");
map.put("20200101", fileList1);

List<String> fileList2 = new ArrayList<String>();
fileList2.add("ccc.txt");
map.put("20200301", fileList2);

mapにどんな値が入っていてもTreeNode[] nodeが作成できるような標準化をするにはどうすればよいかがわかりませんでした。
どなたかお力を貸していただきたいです。
よろしくお願いいたします。

Comments

[m.ts10806]

> Springboot + ThymeleafでWebページを作成したいです。

とのことですがタグ「Java」だけで問題ないでしょうか。

[moshi]

＞m.ts10806さん
ご指摘ありがとうございます。
タグを追加させていただきました。

Answer 1

とりあえず一般的によく使われるMapへのマージ処理とTreeNodeへの変換を
Stream APIを使用、未使用の2パターン書いてみましたけど、
こういう物が欲しかったのでしょうか。

public class Main {

    public static void main(String[] args) {
        // 入力引数
        String[][] nodeStr = {
                {"20200101", "aaa.txt"},
                {"20200101", "bbb.txt"},
                {"20200301", "ccc.txt"},
                };

        // Streamを使った方法
        // Mapのマージ処理
        Map<String, List<String>> map = new TreeMap<String, List<String>>();
        for (String[] lNode : nodeStr) {
            // mergeメソッドでキーによる追加、変更を処理する
            map.merge(lNode[0], new ArrayList<String>(Arrays.asList(lNode[1])), (v1, v2) -> {
                v1.add(lNode[1]);
                return v1;
            });
        }

        // MapをTreeNode[]に書き出し
        TreeNode[] node = map.entrySet().stream()
                .map(e -> new TreeNode(e.getKey(),
                        e.getValue().stream()
                        .map(v -> new TreeNode(v))
                        .toArray(TreeNode[]::new)
                )).toArray(TreeNode[]::new);

        // 出力
        Arrays.stream(node)
            .flatMap(n -> Arrays.stream(n.getChildNode())
                    .map(cn -> n.getNode() +":"+ cn.getNode()))
            .forEach(System.out::println);



        // Streamを使わない方法
        // Mapのマージ処理
        map = new TreeMap<String, List<String>>();
        for (String[] lNode : nodeStr) {
            // 日次がMapに存在しない場合
            if (!map.containsKey(lNode[0])) {
                map.put(lNode[0], new ArrayList<String>(Arrays.asList(lNode[1])));
            } else {
                List<String> list = map.get(lNode[0]);
                list.add(lNode[1]);
            }
        }

        // MapをTreeNode[]に書き出し
        node = new TreeNode[map.size()];
        int i = 0;
        for (Entry<String, List<String>> entry : map.entrySet()) {
            List<String> nList = entry.getValue();
            TreeNode[] cNode = new TreeNode[nList.size()];
            for (int j = 0; j < nList.size(); j++) {
                cNode[j] = new TreeNode(nList.get(j));
            }
            node[i] = new TreeNode(entry.getKey(), cNode);
            i++;
        }

        // 出力
        for (TreeNode pNode : node) {
            for (TreeNode cNode : pNode.getChildNode()) {
                System.out.println(pNode.getNode() +":"+ cNode.getNode());
            }
        }
    }

}

class TreeNode {
    public TreeNode(String node) {
        this.node = node;
    }

    public TreeNode(String node, TreeNode[] childNode) {
        this.node = node;
        this.childNode = childNode;
    }

    private String node;
    private TreeNode[] childNode;

    public String getNode() {
        return node;
    }

    public void setNode(String node) {
        this.node = node;
    }

    public TreeNode[] getChildNode() {
        return childNode;
    }

    public void setChildNode(TreeNode[] childNode) {
        this.childNode = childNode;
    }
}

Comments

[moshi]

＞yureighostさん
ご回答ありがとうございます。

まさにこう言った処理がしたかったです。
こうして書いていただけるとそれを追って理解ができたのですが自分では思いつきませんでした。
大変助かりました。
ありがとうございます。

Answer 2

それぞれの固定値のところに、文字列が入った変数を当てはめればどうなるんでしょうか

Comments

[moshi]

＞y_waiwaiさん
ご回答ありがとうございます。

固定値のところに変数を当てはめれば同じことができるのですが、
例えば一つ目の「20200101」のマップのValueのリストの要素にxxx.txtというものが加えられた場合、new TreeNode("xxx.txt")というのを追加しなければいけなくなります。
そのあたりをFor文などでうまく作りたいのですがやり方が思い浮かばなく質問させていただきました…

Answer 3

「それぞれ引数で受け取ってそういうList/Mapを作る関数」を作れば良いのでは？

※動作未確認

Map<String, List<String>> createNodeList(String key ,String[] filenamelist ){
    Map<String, List<String>> nodelist = new Map<String, List<String>>();

    List<String> fileList = new ArrayList<String>();
    for(String filename : filenamelist ){
        fileList.add(filename);
    }
    nodelist.put(key ,fileList);
    return nodelist;
}

あとは、そのファイル名一覧とキーと日付がどのようにリクエストとしてくるか次第ですね。

Comments

[moshi]

>m.ts10806さん
ご回答ありがとうございます。

質問の仕方が悪く申し訳ありません。
「それぞれ引数で受け取ってそういうList/Mapを作る関数」は作れていたのですがそれをTreeNodeに変換する部分がわかりませんでした。