com.nitorcreations.willow.metrics.AccessLogMetric Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of willow-servers Show documentation
Willow operational servlets and servers
The newest version!
package com.nitorcreations.willow.metrics;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import javax.inject.Named;

import com.nitorcreations.willow.messages.AccessLogEntry;
import com.nitorcreations.willow.messages.metrics.MetricConfig;

@Named("/access")
public class AccessLogMetric extends FullMessageMultiseriesMetric {
  long[] limitValues = new long[] { 10L, 100L, 1000L, 10000L };

  private interface ValueGetter {
    long getValue(AccessLogEntry next);
  }
  private static class DurationValueGetter implements ValueGetter {
    @Override
    public long getValue(AccessLogEntry next) {
      return next.getDuration();
    }
  }
  private static class ReturnCodeValueGetter implements ValueGetter {
    @Override
    public long getValue(AccessLogEntry next) {
      return next.getStatus();
    }
  }
  private static final DurationValueGetter DURATION = new DurationValueGetter();
  private static final ReturnCodeValueGetter RETURNCODE = new ReturnCodeValueGetter();
  private ValueGetter getter = DURATION;

  @Override
  public Collection> calculateMetric(MetricConfig conf) {
    if (conf.getLimits().length > 0) {
      limitValues = new long[conf.getLimits().length];
      for (int i = 0; i < conf.getLimits().length; i++) {
        limitValues[i] = Long.parseLong(conf.getLimits()[i]);
      }
    } else if (conf.hasType("statuses")) {
      getter = RETURNCODE;
      if (conf.getLimits().length == 0) {
        limitValues = new long[] { 200L, 300L, 400L, 500L, 600L };
      }
    }
    return super.calculateMetric(conf);
  }

  @Override
  protected void addValue(Map> values, List preceeding, long stepTime, int stepLen, MetricConfig conf) {
    long[] buckets = new long[limitValues.length + 1];
    for (AccessLogEntry next : preceeding) {
      boolean inLimits = false;
      for (int i = 0; i < limitValues.length; i++) {
        if (getter.getValue(next) < limitValues[i]) {
          buckets[i]++;
          inLimits = true;
          break;
        }
      }
      if (!inLimits) {
        buckets[limitValues.length]++;
      }
    }
    String lower = "";
    String upper = "" + limitValues[0];
    for (int i = 0; i < limitValues.length; i++) {
      addBucket(values, lower + "-" + upper, stepTime, buckets[i]);
      lower = Long.toString(limitValues[i]);
      if (i < limitValues.length - 1) {
        upper = "" + limitValues[i+1];
      } else {
        upper = "";
      }
    }
    addBucket(values, lower + "-", stepTime, buckets[buckets.length - 1]);
  }

  private void addBucket(Map> values, String label, long time, long value) {
    SeriesData bucket = values.get(label);
    if (bucket == null) {
      bucket = new SeriesData<>();
      bucket.key = label;
      values.put(label, bucket);
    }
    Point p = new Point<>();
    p.x = time;
    p.y = value;
    bucket.values.add(p);
  }
}