LegendTreeViewGroupRuleProvider.java

package com.griddynamics.jagger.dbapi.model.rules;

import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.regex.Pattern;

/**
 * Provides TreeGroupRule with predefined groups.
 * Requires format for group.
 */
@Component
public class LegendTreeViewGroupRuleProvider {

    public TreeViewGroupRule provide(String rootId,
                                     Collection<String> legendGroups, String idFormatRegex) {

        List<TreeViewGroupRule> firstLevelFilters = new ArrayList<>();

        for (String legendGroup: legendGroups) {

            String regex = String.format(idFormatRegex, Pattern.quote(legendGroup));
            firstLevelFilters.add(new TreeViewGroupRule(Rule.By.ID, legendGroup, legendGroup, regex));
        }

        // Root filter - will match all metrics
        return new TreeViewGroupRule(Rule.By.ID, rootId, rootId, ".*", firstLevelFilters);
    }
}