io.vertx.ext.metrics.impl.HttpMetricsImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertx-metrics Show documentation
The newest version!
/*
 * Copyright 2014 Red Hat, Inc.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */

package io.vertx.ext.metrics.impl;

import com.codahale.metrics.Timer;

import java.util.concurrent.TimeUnit;

/**
 * @author Nick Scavelli
 */
abstract class HttpMetricsImpl extends NetMetricsImpl {

  private Timer requests;

  public HttpMetricsImpl(AbstractMetrics metrics, String baseName, boolean client) {
    super(metrics, baseName, client);
  }

  @Override
  protected void initialize() {
    super.initialize();
    requests = timer("requests");
  }

  /**
   * Provides a timed context to measure http request latency.
   *
   * @param method the http method or null if it's not to be recorded
   * @param uri the request uri or path of the request, or null if it's not to be recorded
   * @return the TimedContext to be measured
   */
  protected TimedContext time(String method, String uri) {
    return new TimedContext(method, uri);
  }

  protected class TimedContext {
    private String method;
    private String uri;
    private long start;

    private TimedContext(String method, String uri) {
      this.method = (method == null) ? null : method.toLowerCase();
      this.uri = uri;
      start = System.nanoTime();
    }

    protected void stop() {
      if (closed) return;

      long duration = System.nanoTime() - start;

      // Update generic requests metric
      requests.update(duration, TimeUnit.NANOSECONDS);

      // Update specific method / uri request metrics
      if (method != null) {
        timer(method + "-requests").update(duration, TimeUnit.NANOSECONDS);
        if (uri != null) {
          timer(method + "-requests", uri).update(duration, TimeUnit.NANOSECONDS);
        }
      } else if (uri != null) {
        timer("requests", uri).update(duration, TimeUnit.NANOSECONDS);
      }

      if (closed) {
        removeAll();
      }
    }
  }
}