com.wavefront.agent.config.ReportableConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of proxy Show documentation
Service for batching and relaying metric traffic to Wavefront
There is a newer version: 13.4
package com.wavefront.agent.config;

import com.wavefront.common.TaggedMetricName;
import com.yammer.metrics.Metrics;
import com.yammer.metrics.core.Gauge;
import com.yammer.metrics.core.MetricName;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Properties;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.annotation.Nullable;

/**
 * Wrapper class to simplify access to .properties file + track values as metrics as they are
 * retrieved
 *
 * @author [email protected]
 */
public class ReportableConfig {
  private static final Logger logger = Logger.getLogger(ReportableConfig.class.getCanonicalName());

  private final Properties prop = new Properties();

  public ReportableConfig(String fileName) throws IOException {
    prop.load(new FileInputStream(fileName));
  }

  public ReportableConfig() {}

  /** Returns string value for the property without tracking it as a metric */
  public String getRawProperty(String key, String defaultValue) {
    return prop.getProperty(key, defaultValue);
  }

  public int getInteger(String key, Number defaultValue) {
    return getNumber(key, defaultValue).intValue();
  }

  public long getLong(String key, Number defaultValue) {
    return getNumber(key, defaultValue).longValue();
  }

  public double getDouble(String key, Number defaultValue) {
    return getNumber(key, defaultValue).doubleValue();
  }

  public Number getNumber(String key, Number defaultValue) {
    return getNumber(key, defaultValue, null, null);
  }

  public Number getNumber(
      String key,
      @Nullable Number defaultValue,
      @Nullable Number clampMinValue,
      @Nullable Number clampMaxValue) {
    String property = prop.getProperty(key);
    if (property == null && defaultValue == null) return null;
    double d;
    try {
      d = property == null ? defaultValue.doubleValue() : Double.parseDouble(property.trim());
    } catch (NumberFormatException e) {
      throw new NumberFormatException(
          "Config setting \"" + key + "\": invalid number format \"" + property + "\"");
    }
    if (clampMinValue != null && d < clampMinValue.longValue()) {
      logger.log(
          Level.WARNING,
          key
              + " ("
              + d
              + ") is less than "
              + clampMinValue
              + ", will default to "
              + clampMinValue);
      reportGauge(clampMinValue, new MetricName("config", "", key));
      return clampMinValue;
    }
    if (clampMaxValue != null && d > clampMaxValue.longValue()) {
      logger.log(
          Level.WARNING,
          key
              + " ("
              + d
              + ") is greater than "
              + clampMaxValue
              + ", will default to "
              + clampMaxValue);
      reportGauge(clampMaxValue, new MetricName("config", "", key));
      return clampMaxValue;
    }
    reportGauge(d, new MetricName("config", "", key));
    return d;
  }

  public String getString(String key, String defaultValue) {
    return getString(key, defaultValue, null);
  }

  public String getString(
      String key, String defaultValue, @Nullable Function converter) {
    String s = prop.getProperty(key, defaultValue);
    if (s == null || s.trim().isEmpty()) {
      reportGauge(0, new MetricName("config", "", key));
    } else {
      reportGauge(
          1,
          new TaggedMetricName("config", key, "value", converter == null ? s : converter.apply(s)));
    }
    return s;
  }

  public Boolean getBoolean(String key, Boolean defaultValue) {
    Boolean b = Boolean.parseBoolean(prop.getProperty(key, String.valueOf(defaultValue)).trim());
    reportGauge(b ? 1 : 0, new MetricName("config", "", key));
    return b;
  }

  public Boolean isDefined(String key) {
    return prop.getProperty(key) != null;
  }

  public static void reportSettingAsGauge(Supplier numberSupplier, String key) {
    reportGauge(numberSupplier, new MetricName("config", "", key));
  }

  public static void reportGauge(Supplier numberSupplier, MetricName metricName) {
    Metrics.newGauge(
        metricName,
        new Gauge() {
          @Override
          public Double value() {
            return numberSupplier.get().doubleValue();
          }
        });
  }

  public static void reportGauge(Number number, MetricName metricName) {
    Metrics.newGauge(
        metricName,
        new Gauge() {
          @Override
          public Double value() {
            return number.doubleValue();
          }
        });
  }
}