JavaFX按子目录和文件对TreeView进行排序

Question

public void createTreeView(String dbpath) {

    TreeItem<Object> ROOTTREE = new TreeItem<>(dbpath.substring(dbpath.lastIndexOf("\\")));

    try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(dbpath))) {
        for(Path path: directoryStream) {
            if (Files.isDirectory(path)) {
                TreeItem<Object> leafs = new TreeItem<>(strPath.substring(1 + strPath.lastIndexOf("\\")));
                parent.getChildren().add(leafs);
                return leafs;
            } else {
                getLeafs(path, tree);
            }

        }
    } catch (IOException e) {
        e.printStackTrace();
    }
    tree.setExpanded(ROOTTREE);
    TreeView_DBDirectory.setRoot(ROOTTREE);
    TreeView_DBDirectory.setShowRoot(true);
}

也许有更合适的解决方案？：/

Answer 1

将目录存储在列表中。将文件存储在另一个列表中。首先将目录列表中的所有项添加到树子项中。然后将文件列表的所有项添加到树子项中。

TreeItem<Object> tree = new TreeItem<>(dbPath.substring(dbPath.lastIndexOf(File.separator)));

List<TreeItem<Object>> dirs = new ArrayList<>();
List<TreeItem<Object>> files = new ArrayList<>();

try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(dbPath))) {
    for(Path path: directoryStream) {
        if (Files.isDirectory(path)) {
            TreeItem<Object> subDirectory = new TreeItem<>(path);
            getSubLeafs(path, subDirectory);
            dirs.add(subDirectory);
        } else {
            files.add(getLeafs(path));
        }
    }

    tree.getChildren().addAll(dirs);
    tree.getChildren().addAll(files);
} catch (IOException e) {
    e.printStackTrace();
}

在以下目录结构上运行的示例：

$ tree /tmp/test
/tmp/test
├── a1.txt
├── d1
│   ├── b1.txt
│   └── b2.txt
├── d2
│   └── c1.txt
└── f1.txt

我单击树节点来展开它们，因此您可以看到它们的值（默认情况下树节点不展开）。

import javafx.application.Application;
import javafx.scene.Scene;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeView;
import javafx.stage.Stage;

import java.io.File;
import java.io.IOException;
import java.nio.file.*;
import java.util.ArrayList;
import java.util.List;

public class MenuClickSample extends Application {
    TreeView<Object> treeView = new TreeView<>();

    @Override public void start(Stage stage) throws IOException {
        createTreeView("/tmp/test");
        Scene scene = new Scene(treeView);
        stage.setScene(scene);
        stage.show();
    }

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

    private void createTreeView(String dbPath) {
        TreeItem<Object> tree = new TreeItem<>(dbPath.substring(dbPath.lastIndexOf(File.separator)));

        List<TreeItem<Object>> dirs = new ArrayList<>();
        List<TreeItem<Object>> files = new ArrayList<>();

        try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(dbPath))) {
            for(Path path: directoryStream) {
                if (Files.isDirectory(path)) {
                    TreeItem<Object> subDirectory = new TreeItem<>(path);
                    getSubLeafs(path, subDirectory);
                    dirs.add(subDirectory);
                } else {
                    files.add(getLeafs(path));
                }
            }

            tree.getChildren().addAll(dirs);
            tree.getChildren().addAll(files);
        } catch (IOException e) {
            e.printStackTrace();
        }

        tree.setExpanded(true);
        treeView.setRoot(tree);
        treeView.setShowRoot(true);
    }

    private TreeItem<Object> getLeafs(Path subPath) {
        String strPath = subPath.toString();
        TreeItem<Object> leafs = new TreeItem<>(strPath.substring(1 + strPath.lastIndexOf(File.separator)));
        return leafs;
    }

    private void getSubLeafs(Path subPath, TreeItem<Object> parent) {
        try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(subPath.toString()))) {
            for(Path subDir: directoryStream) {
                // explicit search for files because we dont want to get sub-sub-directories
                if (!Files.isDirectory(subDir)) {
                    String subTree = subDir.toString();
                    TreeItem<Object> subLeafs = new TreeItem<>(subTree.substring(1 + subTree.lastIndexOf(File.separator)));
                    parent.getChildren().add(subLeafs);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}