io.ebeaninternal.server.profile.DQueryPlanMetric Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ebean-core Show documentation
ebean core module
There is a newer version: 15.8.0
package io.ebeaninternal.server.profile;

import io.ebean.meta.MetaQueryMetric;
import io.ebean.meta.MetricVisitor;
import io.ebean.metric.QueryPlanMetric;
import io.ebean.metric.TimedMetric;
import io.ebean.metric.TimedMetricStats;

final class DQueryPlanMetric implements QueryPlanMetric {

  private final DQueryPlanMeta meta;
  private final DTimedMetric metric;
  private boolean collected;
  private String reportName;

  DQueryPlanMetric(DQueryPlanMeta meta, DTimedMetric metric) {
    this.meta = meta;
    this.metric = metric;
  }

  @Override
  public void visit(MetricVisitor visitor) {
    TimedMetricStats stats = metric.collect(visitor.reset());
    if (stats != null) {
      String name = reportName != null ? reportName : reportName(visitor);
      visitor.visitQuery(new Stats(name, meta, stats, collected));
      collected = true;
    }
  }

  String reportName(MetricVisitor visitor) {
    final String tmp = visitor.namingConvention().apply(meta.getName());
    this.reportName = tmp;
    return tmp;
  }

  @Override
  public TimedMetric metric() {
    return metric;
  }

  private static class Stats implements MetaQueryMetric {

    private final String name;
    private final DQueryPlanMeta meta;
    private final TimedMetricStats stats;
    private final boolean collected;

    private Stats(String name, DQueryPlanMeta meta, TimedMetricStats stats, boolean collected) {
      this.name = name;
      this.meta = meta;
      this.stats = stats;
      this.collected = collected;
    }

    @Override
    public String toString() {
      return meta + " " + stats + " sql:" + sql();
    }

    @Override
    public Class type() {
      return meta.getType();
    }

    @Override
    public boolean initialCollection() {
      return !collected;
    }

    @Override
    public String hash() {
      return meta.getHash();
    }

    @Override
    public String label() {
      return meta.getLabel();
    }

    @Override
    public String sql() {
      return meta.getSql();
    }

    @Override
    public String name() {
      return name;
    }

    @Override
    public String location() {
      return meta.getLocation();
    }

    @Override
    public long count() {
      return stats.count();
    }

    @Override
    public long total() {
      return stats.total();
    }

    @Override
    public long max() {
      return stats.max();
    }

    @Override
    public long mean() {
      return stats.mean();
    }
  }
}