com.yugabyte.sample.common.metrics.Metric Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yb-sample-apps Show documentation

Sample applications for benchmarking YugaByte DB functionality on various workload types.

The newest version!

// Copyright (c) YugaByte, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
// in compliance with the License.  You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied.  See the License for the specific language governing permissions and limitations
// under the License.
//

package com.yugabyte.sample.common.metrics;

import org.apache.log4j.Logger;

public class Metric {
  private static final Logger LOG = Logger.getLogger(Metric.class);
  String name;
  private final Object lock = new Object();
  private long curOpCount = 0;
  private long curOpLatencyNanos = 0;
  private long totalOpCount = 0;
  private long lastSnapshotNanos;

  public Metric(String name) {
    this.name = name;
    lastSnapshotNanos = System.nanoTime();
  }

  /**
   * Accumulate metrics with operations processed as one batch.
   * @param numOps number of ops processed as one batch
   * @param batchLatencyNanos whole batch latency
   */
  public void accumulate(long numOps, long batchLatencyNanos) {
    synchronized(lock) {
      curOpCount += numOps;
      curOpLatencyNanos += batchLatencyNanos * numOps;
      totalOpCount += numOps;
    }
  }

  public String getMetricsAndReset() {
    String msg = null;
    synchronized(lock) {
      long currNanos = System.nanoTime();
      long elapsedNanos = currNanos - lastSnapshotNanos;
      LOG.debug("currentOpLatency: " + curOpLatencyNanos + ", currentOpCount: " + curOpCount);
      double ops_per_sec =
          (elapsedNanos == 0) ? 0 : (curOpCount * 1000000000 * 1.0 / elapsedNanos);
      double latency = (curOpCount == 0) ? 0 : (curOpLatencyNanos / 1000000 * 1.0 / curOpCount);
      msg = String.format("%s: %.2f ops/sec (%.2f ms/op), %d total ops",
                          name, ops_per_sec, latency, totalOpCount);
      curOpCount = 0;
      curOpLatencyNanos = 0;
      lastSnapshotNanos = currNanos;
    }
    return msg;
  }
}