根据下划线生成tree

/**
     * 获取指标表
     *
     * @return
     */
    @GetMapping("/metricsTablesTree")
    public R getMetricsTables() {
        List<MetricsTablesVo> vos = metricsTablesService.queryList(new MetricsTablesBo());
        ArrayList<MetricsTablesTree> metricsTablesTrees = new ArrayList<>();
        vos.forEach(vo -> {
            getAllMetrics(vos, metricsTablesTrees, vo.getName(), vo.getTableKey(), vo.getFrequency());
        });
        TreeNodeConfig config = new TreeNodeConfig();
        config.setIdKey("label");//默认为id可以不设置
        config.setParentIdKey("parentLabel");//默认为parentId可以不设置
        // 3.转树,Tree<>里面泛型为id的类型
        List<Tree<String>> trees = TreeUtil.build(metricsTablesTrees, "root", config, (object, tree) -> {
            // 也可以使用 tree.setId(object.getId());等一些默认值
            tree.putExtra("id", object.getId());
            tree.putExtra("label", object.getLabel());
            tree.putExtra("frequency", object.getFrequency());
            tree.putExtra("parentLabel", object.getParentLabel());
        });

        return R.ok(trees);
    }

    private void getAllMetrics(List<MetricsTablesVo> vos, List<MetricsTablesTree> metricsTablesTrees, String name, String tableKey, String frequency) {
        // 查找最后一个下划线的位置
        int lastUnderscoreIndex = name.lastIndexOf('_');
        // 如果找到了下划线,并且它不是字符串的第一个字符
        if (lastUnderscoreIndex > 0) {
            // 分割字符串为两部分
            String firstPart = name.substring(0, lastUnderscoreIndex);
            String secondPart = name.substring(lastUnderscoreIndex + 1);
            MetricsTablesTree obj = new MetricsTablesTree(name, firstPart, tableKey, frequency);
            metricsTablesTrees.add(obj);
            if (!exist(vos, firstPart)) {
                getAllMetrics(vos, metricsTablesTrees, firstPart, "root-" + UUID.randomUUID(), frequency);
            } else {
                getAllMetrics(vos, metricsTablesTrees, firstPart, tableKey, frequency);
            }
        } else {
            // 如果没有找到下划线或者下划线位于开头或结尾,则处理这种情况
            MetricsTablesTree obj = new MetricsTablesTree(name, "root", "root-" + UUID.randomUUID(), frequency);
            metricsTablesTrees.add(obj);
        }
    }

    private boolean exist(List<MetricsTablesVo> vos, String name) {
        for (MetricsTablesVo vo : vos) {
            if (vo.getName().equals(name)) {
                return true;
            }
        }
        return false;
    }
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容