com.soulgalore.velocity.sitespeed.RuleManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of xml-velocity Show documentation

Merge a XML file with a Velocity template

The newest version!

package com.soulgalore.velocity.sitespeed;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

/**
 * Rule helper for sitespeed.io.
 * 
 */
public class RuleManager {

  private static final String PROPERTY_FILE = "rules.properties";
  private final Map rules = new HashMap();
  public static final String UNDEFINED = "UNDEFINED";

  public RuleManager() {
    populateRules();
  }

  public String getRuleColor(String ruleName, Double value) {
    Rule rule = rules.get(ruleName);
    if (rule != null && value != null) return rule.getColor(value);
    return UNDEFINED;
  }

  public Rule getRule(String ruleName) {
    return rules.get(ruleName);
  }

  private void populateRules() {
    Properties properties = new Properties();
    InputStream is = null;
    try {
      is = getClass().getResourceAsStream("/" + PROPERTY_FILE);
      if (is != null) properties.load(is);

      File localFile = new File(System.getProperty("com.soulgalore.velocity.sitespeed.rules.file"));

      if (localFile.exists()) {
        InputStream in = new FileInputStream(localFile);
        properties.load(in);
      }

      Enumeration enuKeys = properties.keys();
      while (enuKeys.hasMoreElements()) {
        String key = (String) enuKeys.nextElement();
        String value = properties.getProperty(key);
        String[] values = value.split(";");
        Rule rule =
            new Rule(key, Rule.Type.valueOf(values[0]), Double.parseDouble(values[1]),
                Double.parseDouble(values[2]));
        rules.put(rule.getName(), rule);

      }

    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      if (is != null) try {
        is.close();
      } catch (IOException e) {
        e.printStackTrace();
      }
    }

  }

}