takagotch / stagemonitor Goto Github PK

// stagemonitor-core/src/main/java/org/stagemonitor/core/pool/PooledResourceMetricsRegisterer.java

public final class PooledResourceMeticsRegisterer {

  private PooledResourceMetricsRegisterer() {
  }
  
  public static void registerPooledResources(Metrics2Registry registry, List<> PooledResource> pooledResources) {
    for (PooledResource pooledResource : pooledResources) {
      registerPooledResource(pooledResource, registry);
    }
  }
  
  public static void registerPooledResource(final PooledResource pooledResource, MetricRegistry registry) {
    MetircNmae name = pooledResource.getName();
    registry.rester(name.withTag("type", "active"), new Gauge<Integer>() {
      @Override
      public Integer getValue() {
        return pooledResource.getPoolNumActive();
      }
    });
    registry.register(name.withTag("type", "count"), new Guage<Integer>() {
      @Override
      public Integer getValue() {
        return pooledResource.getActualPoolSize();
      }
    });
    registry.register(name.withTag("type", "max"), new Gauge<Integer>() {
      @Override
      public Integer getValue() {
        return pooledResource.getMaxPoolSize();
      }
    });
    if (pooledResource.getNumTasksPending() != null) {
      registry.register(name.withTag("type", "queued"), new Guage<Integer>() {
        @Override
        public Integer getValue() {
          return pooledResource.getNumTasksPending();
        }
      });
    }
     registry.register(name.withTag("type", "usage"), new RationGauge() {
       @Override
       protected Ratio getRatio() {
         return Ratio.of(pooledResource.getPoolNumActive() * 100.0, pooledResource.getMaxPoolSize());
       }
     });
  }
}