Java ReadableMetricsRegistry类代码示例

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
@Test
public void testOnMessageUpdatesMetrics() {
  TaskContextImpl mockTaskContext = mock(TaskContextImpl.class);
  ReadableMetricsRegistry mockMetricsRegistry = mock(ReadableMetricsRegistry.class);
  when(mockTaskContext.getMetricsRegistry()).thenReturn(mockMetricsRegistry);
  Counter mockCounter = mock(Counter.class);
  Timer mockTimer = mock(Timer.class);
  when(mockMetricsRegistry.newCounter(anyString(), anyString())).thenReturn(mockCounter);
  when(mockMetricsRegistry.newTimer(anyString(), anyString())).thenReturn(mockTimer);

  Object mockTestOpImplOutput = mock(Object.class);
  OperatorImpl<Object, Object> opImpl = new TestOpImpl(mockTestOpImplOutput);
  opImpl.init(mock(Config.class), mockTaskContext);

  // send a message to this operator
  MessageCollector mockCollector = mock(MessageCollector.class);
  TaskCoordinator mockCoordinator = mock(TaskCoordinator.class);
  opImpl.onMessage(mock(Object.class), mockCollector, mockCoordinator);

  // verify that it updates message count and timer metrics
  verify(mockCounter, times(1)).inc();
  verify(mockTimer, times(1)).update(anyLong());
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
@Test
public void testOnTimerUpdatesMetrics() {
  TaskContextImpl mockTaskContext = mock(TaskContextImpl.class);
  ReadableMetricsRegistry mockMetricsRegistry = mock(ReadableMetricsRegistry.class);
  when(mockTaskContext.getMetricsRegistry()).thenReturn(mockMetricsRegistry);
  Counter mockMessageCounter = mock(Counter.class);
  Timer mockTimer = mock(Timer.class);
  when(mockMetricsRegistry.newCounter(anyString(), anyString())).thenReturn(mockMessageCounter);
  when(mockMetricsRegistry.newTimer(anyString(), anyString())).thenReturn(mockTimer);

  Object mockTestOpImplOutput = mock(Object.class);
  OperatorImpl<Object, Object> opImpl = new TestOpImpl(mockTestOpImplOutput);
  opImpl.init(mock(Config.class), mockTaskContext);

  // send a message to this operator
  MessageCollector mockCollector = mock(MessageCollector.class);
  TaskCoordinator mockCoordinator = mock(TaskCoordinator.class);
  opImpl.onTimer(mockCollector, mockCoordinator);

  // verify that it updates metrics
  verify(mockMessageCounter, times(0)).inc();
  verify(mockTimer, times(1)).update(anyLong());
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
public SamzaRestService(Server server,
                        ReadableMetricsRegistry metricsRegistry,
                        Map<String, MetricsReporter> metricsReporters,
                        ServletContextHandler context) {
  this.server = server;
  this.metricsRegistry = metricsRegistry;
  this.metricsReporters = metricsReporters;
  this.context = context;
  this.context.setContextPath("/");
  server.setHandler(context);
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
/**
 * Command line interface to run the server.
 *
 * @param args arguments supported by {@link org.apache.samza.util.CommandLine}.
 *             In particular, --config-path and --config-factory are used to read the Samza REST config file.
 * @throws Exception if the server could not be successfully started.
 */
public static void main(String[] args)
    throws Exception {
  ScheduledExecutorSchedulingProvider schedulingProvider = null;
  try {
    SamzaRestConfig config = parseConfig(args);
    ReadableMetricsRegistry metricsRegistry = new MetricsRegistryMap();
    log.info("Creating new SamzaRestService with config: {}", config);
    MetricsConfig metricsConfig = new MetricsConfig(config);
    Map<String, MetricsReporter> metricsReporters = MetricsReporterLoader.getMetricsReporters(metricsConfig, Util.getLocalHost().getHostName());
    SamzaRestService restService = new SamzaRestService(new Server(config.getPort()), metricsRegistry, metricsReporters,
                                                        new ServletContextHandler(ServletContextHandler.SESSIONS));

    // Add applications
    SamzaRestApplication samzaRestApplication = new SamzaRestApplication(config);
    ServletContainer container = new ServletContainer(samzaRestApplication);
    restService.addServlet(container, "/*");

    // Schedule monitors to run
    ThreadFactory threadFactory = new ThreadFactoryBuilder().setDaemon(true)
                                                            .setNameFormat("MonitorThread-%d")
                                                            .build();
    ScheduledExecutorService schedulingService = Executors.newScheduledThreadPool(1, threadFactory);
    schedulingProvider = new ScheduledExecutorSchedulingProvider(schedulingService);
    SamzaMonitorService monitorService = new SamzaMonitorService(config,
                                                                 metricsRegistry,
                                                                 schedulingProvider);
    monitorService.start();

    restService.runBlocking();
    monitorService.stop();
  } catch (Throwable t) {
    log.error("Exception in main.", t);
  } finally {
    if (schedulingProvider != null){
      schedulingProvider.stop();
    }
  }
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
@Before
public void setup()
    throws Exception {
  container = mock(YarnContainer.class);
  ReadableMetricsRegistry registry = new MetricsRegistryMap("test-registry");

  yarnAppState =
      new YarnAppState(-1, ConverterUtils.toContainerId("container_1350670447861_0003_01_000001"), "testHost", 1, 1);
  webApp = new HttpServer("/", 0, "", new ServletHolder(new DefaultServlet()));
  webApp.addServlet("/", new YarnContainerHeartbeatServlet(yarnAppState, registry));
  webApp.start();
  mapper = new ObjectMapper();
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
private void startMetrics() {
  for (MetricsReporter reporter: reporters.values()) {
    reporter.register("job-coordinator-" + processorId, (ReadableMetricsRegistry) metrics.getMetricsRegistry());
    reporter.start();
  }
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
@Override
public ReadableMetricsRegistry getMetricsRegistry() {
  return metrics.registry();
}
 

import org.apache.samza.metrics.ReadableMetricsRegistry; //导入依赖的package包/类
public YarnContainerHeartbeatServlet(YarnAppState yarnAppState, ReadableMetricsRegistry registry) {
  this.yarnAppState = yarnAppState;
  this.mapper = new ObjectMapper();
  this.heartbeatsExpiredCount = registry.newCounter(GROUP, "heartbeats-expired");
}