在Vaadin Flow / 14 TreeGrid组件中使用节点图标

// 导入所需的依赖
import com.vaadin.flow.component.icon.Icon;
import com.vaadin.flow.component.treegrid.TreeGrid;
import com.vaadin.flow.data.provider.hierarchy.TreeDataProvider;
import com.vaadin.flow.router.Route;

@Route("treegrid")
public class TreeGridExample extends Div {

    public TreeGridExample() {
        // 创建TreeGrid实例
        TreeGrid<String> treeGrid = new TreeGrid<>();

        // 设置列
        treeGrid.addHierarchyColumn(String::toString).setHeader("Name");

        // 设置数据源
        TreeDataProvider<String> dataProvider = new TreeDataProvider<>(getData());
        treeGrid.setDataProvider(dataProvider);

        // 定义节点图标
        Icon folderIcon = new Icon(VaadinIcon.FOLDER);
        Icon fileIcon = new Icon(VaadinIcon.FILE);

        // 设置节点图标
        treeGrid.setItemIconGenerator(item -> {
            if (item.endsWith(".txt")) {
                return fileIcon;
            } else {
                return folderIcon;
            }
        });

        // 注册节点图标生成器
        treeGrid.setItemIconGenerator(treeGrid.getItemIconGenerator());

        // 将TreeGrid添加到UI中
        add(treeGrid);
    }

    private HierarchicalDataProvider<String, Void> getData() {
        // 返回层次结构数据源
        // 可以根据实际需求自定义数据源
    }
}