Examples with HarvestServiceImpl com.newrelic.agent.HarvestServiceImpl used on opensource projects

Example 1 with HarvestServiceImpl

use of com.newrelic.agent.HarvestServiceImpl in project newrelic-java-agent by newrelic.

the class CommandParserTest method createServiceManager.

private MockServiceManager createServiceManager(Map<String, Object> config) throws Exception {
    MockServiceManager serviceManager = new MockServiceManager();
    ServiceFactory.setServiceManager(serviceManager);
    ThreadService threadService = new ThreadService();
    serviceManager.setThreadService(threadService);
    ConfigService configService = ConfigServiceFactory.createConfigService(AgentConfigImpl.createAgentConfig(config), config);
    serviceManager.setConfigService(configService);
    TransactionService transactionService = new TransactionService();
    serviceManager.setTransactionService(transactionService);
    TransactionTraceService transactionTraceService = new TransactionTraceService();
    serviceManager.setTransactionTraceService(transactionTraceService);
    RPMConnectionService rpmConnectionService = new RPMConnectionServiceImpl();
    serviceManager.setRPMConnectionService(rpmConnectionService);
    ProfilerService profilerService = new ProfilerService();
    serviceManager.setProfilerService(profilerService);
    EnvironmentService envService = new EnvironmentServiceImpl();
    serviceManager.setEnvironmentService(envService);
    HarvestService harvestService = new HarvestServiceImpl();
    serviceManager.setHarvestService(harvestService);
    return serviceManager;
}

Example 2 with HarvestServiceImpl

use of com.newrelic.agent.HarvestServiceImpl in project newrelic-java-agent by newrelic.

the class CommandParserThreadProfilerTest method createServiceManager.

private MockServiceManager createServiceManager(Map<String, Object> config) throws Exception {
    MockServiceManager serviceManager = new MockServiceManager();
    ServiceFactory.setServiceManager(serviceManager);
    ThreadService threadService = new ThreadService();
    serviceManager.setThreadService(threadService);
    ConfigService configService = ConfigServiceFactory.createConfigService(AgentConfigImpl.createAgentConfig(config), config);
    serviceManager.setConfigService(configService);
    TransactionService transactionService = new TransactionService();
    serviceManager.setTransactionService(transactionService);
    TransactionTraceService transactionTraceService = new TransactionTraceService();
    serviceManager.setTransactionTraceService(transactionTraceService);
    RPMConnectionService rpmConnectionService = new RPMConnectionServiceImpl();
    serviceManager.setRPMConnectionService(rpmConnectionService);
    ProfilerService profilerService = new ProfilerService();
    serviceManager.setProfilerService(profilerService);
    EnvironmentService envService = new EnvironmentServiceImpl();
    serviceManager.setEnvironmentService(envService);
    HarvestService harvestService = new HarvestServiceImpl();
    serviceManager.setHarvestService(harvestService);
    return serviceManager;
}

Example 3 with HarvestServiceImpl

use of com.newrelic.agent.HarvestServiceImpl in project newrelic-java-agent by newrelic.

the class InsightsServiceImplTest method testCustomEventFasterHarvest.

@Test
public void testCustomEventFasterHarvest() throws Exception {
    Map<String, Object> config = new HashMap<>();
    config.put(AgentConfigImpl.APP_NAME, EventTestHelper.APP_NAME);
    EventTestHelper.createServiceManager(config);
    String appName = ServiceFactory.getRPMService().getApplicationName();
    InsightsServiceImpl insightsService = new InsightsServiceImpl();
    ((MockServiceManager) ServiceFactory.getServiceManager()).setInsights(insightsService);
    ServiceManager serviceManager = spy(ServiceFactory.getServiceManager());
    ServiceFactory.setServiceManager(serviceManager);
    HarvestServiceImpl harvestService = spy(new HarvestServiceImpl());
    doReturn(harvestService).when(serviceManager).getHarvestService();
    doReturn(0L).when(harvestService).getInitialDelay();
    insightsService.addHarvestableToService(appName);
    insightsService.configureHarvestables(60, 10);
    assertEquals(10, insightsService.getMaxSamplesStored());
    insightsService.start();
    Map<String, Object> connectionInfo = new HashMap<>();
    Map<String, Object> eventHarvest = new HashMap<>();
    Map<String, Object> harvestLimits = new HashMap<>();
    // 5 is the lowest allowable value
    eventHarvest.put("report_period_ms", 5000L);
    eventHarvest.put("harvest_limits", harvestLimits);
    harvestLimits.put("custom_event_data", 1L);
    connectionInfo.put("event_harvest_config", eventHarvest);
    harvestService.startHarvestables(ServiceFactory.getRPMService(), AgentConfigImpl.createAgentConfig(connectionInfo));
    Thread.sleep(500);
    Map<String, String> m = Collections.singletonMap("key", "value");
    insightsService.recordCustomEvent("CustomEvent", m);
    Thread.sleep(5050);
    checkForEvent();
    ((MockRPMService) ServiceFactory.getRPMService()).clearEvents();
    insightsService.recordCustomEvent("CustomEvent", m);
    Thread.sleep(5050);
    checkForEvent();
}

Example 4 with HarvestServiceImpl

use of com.newrelic.agent.HarvestServiceImpl in project newrelic-java-agent by newrelic.

the class ServiceManagerImpl method doStart.

@Override
protected synchronized void doStart() throws Exception {
    // The ConfigService has been created, but not started. This means it
    // is safe to consult the config, but it will be the local config only.
    coreService.start();
    threadService = new ThreadService();
    circuitBreakerService = new CircuitBreakerService();
    classTransformerService = new ClassTransformerServiceImpl(coreService.getInstrumentation());
    AgentConfig config = configService.getDefaultAgentConfig();
    JmxConfig jmxConfig = config.getJmxConfig();
    jmxService = new JmxService(jmxConfig);
    Logger jarCollectorLogger = Agent.LOG.getChildLogger("com.newrelic.jar_collector");
    boolean jarCollectorEnabled = configService.getDefaultAgentConfig().getJarCollectorConfig().isEnabled();
    AtomicBoolean shouldSendAllJars = new AtomicBoolean(true);
    TrackedAddSet<JarData> analyzedJars = new TrackedAddSet<>();
    Function<URL, JarData> processor = new JarCollectorServiceProcessor(jarCollectorLogger, configService.getDefaultAgentConfig());
    JarAnalystFactory jarAnalystFactory = new JarAnalystFactory(processor, analyzedJars, jarCollectorLogger);
    ExecutorService executorService = Executors.newSingleThreadExecutor(new DefaultThreadFactory("New Relic Jar Analysis Thread", true));
    JarCollectorInputs jarCollectorInputs = JarCollectorInputs.build(jarCollectorEnabled, jarAnalystFactory, executorService, jarCollectorLogger);
    jarCollectorService = new JarCollectorServiceImpl(jarCollectorLogger, jarCollectorEnabled, shouldSendAllJars, analyzedJars, jarCollectorInputs.getClassNoticingFactory());
    extensionService = new ExtensionService(configService, jarCollectorInputs.getExtensionAnalysisProducer());
    String defaultAppName = configService.getDefaultAgentConfig().getApplicationName();
    JarCollectorConnectionListener jarCollectorConnectionListener = new JarCollectorConnectionListener(defaultAppName, shouldSendAllJars);
    JarCollectorHarvestListener jarCollectorHarvestListener = new JarCollectorHarvestListener(defaultAppName, jarCollectorService);
    sourceLanguageService = new SourceLanguageService();
    expirationService = new ExpirationService();
    tracerService = new TracerService();
    // this allows async parts of transaction to be registered
    // it must be created before the first class transformation occurs
    asyncTxService = new AsyncTransactionService();
    // this is called in a transaction finish - it needs to be created before the first class transformation
    environmentService = new EnvironmentServiceImpl();
    /*
         * Before this point the ClassTransformer is not initialized, so be careful not to load classes that should be
         * instrumented.
         */
    cacheService = new CacheService();
    extensionService.start();
    classTransformerService.start();
    boolean realAgent = coreService.getInstrumentation() != null;
    statsService = new StatsServiceImpl();
    replayStartupStatsWork();
    utilizationService = new UtilizationService();
    // Start as early as possible.
    if (realAgent) {
        utilizationService.start();
    }
    rpmConnectionService = new RPMConnectionServiceImpl();
    transactionService = new TransactionService();
    InfiniteTracing infiniteTracing = buildInfiniteTracing(configService);
    InfiniteTracingEnabledCheck infiniteTracingEnabledCheck = new InfiniteTracingEnabledCheck(configService);
    SpanEventCreationDecider spanEventCreationDecider = new SpanEventCreationDecider(configService);
    AgentConnectionEstablishedListener agentConnectionEstablishedListener = new UpdateInfiniteTracingAfterConnect(infiniteTracingEnabledCheck, infiniteTracing);
    JfrConfig jfrConfig = config.getJfrConfig();
    jfrService = new JfrService(jfrConfig, configService.getDefaultAgentConfig());
    AgentConnectionEstablishedListener jfrServiceConnectionListener = new JfrServiceConnectionListener(jfrService);
    distributedTraceService = new DistributedTraceServiceImpl();
    TransactionDataToDistributedTraceIntrinsics transactionDataToDistributedTraceIntrinsics = new TransactionDataToDistributedTraceIntrinsics(distributedTraceService);
    rpmServiceManager = new RPMServiceManagerImpl(agentConnectionEstablishedListener, jarCollectorConnectionListener, jfrServiceConnectionListener);
    normalizationService = new NormalizationServiceImpl();
    harvestService = new HarvestServiceImpl();
    gcService = realAgent ? new GCService() : new NoopService("GC Service");
    transactionTraceService = new TransactionTraceService();
    transactionEventsService = new TransactionEventsService(transactionDataToDistributedTraceIntrinsics);
    profilerService = new ProfilerService();
    commandParser = new CommandParser();
    cpuSamplerService = realAgent ? new CPUSamplerService() : new NoopService("CPU Sampler");
    deadlockDetectorService = new DeadlockDetectorService();
    samplerService = realAgent ? new SamplerServiceImpl() : new NoopSamplerService();
    sqlTraceService = new SqlTraceServiceImpl();
    databaseService = new DatabaseService();
    browserService = new BrowserServiceImpl();
    remoteInstrumentationService = new RemoteInstrumentationServiceImpl();
    attsService = new AttributesService();
    insightsService = new InsightsServiceImpl();
    logSenderService = new LogSenderServiceImpl();
    spanEventsService = SpanEventsServiceFactory.builder().configService(configService).rpmServiceManager(rpmServiceManager).transactionService(transactionService).infiniteTracingConsumer(infiniteTracing).spanEventCreationDecider(spanEventCreationDecider).environmentService(environmentService).transactionDataToDistributedTraceIntrinsics(transactionDataToDistributedTraceIntrinsics).build();
    // Register harvest listeners that started before harvest service was created.
    harvestService.addHarvestListener(extensionService);
    harvestService.addHarvestListener(jarCollectorHarvestListener);
    asyncTxService.start();
    threadService.start();
    statsService.start();
    environmentService.start();
    rpmConnectionService.start();
    tracerService.start();
    jarCollectorService.start();
    sourceLanguageService.start();
    harvestService.start();
    gcService.start();
    transactionService.start();
    transactionTraceService.start();
    transactionEventsService.start();
    profilerService.start();
    commandParser.start();
    jmxService.start();
    cpuSamplerService.start();
    deadlockDetectorService.start();
    samplerService.start();
    sqlTraceService.start();
    browserService.start();
    cacheService.start();
    normalizationService.start();
    databaseService.start();
    configService.start();
    remoteInstrumentationService.start();
    attsService.start();
    insightsService.start();
    logSenderService.start();
    circuitBreakerService.start();
    distributedTraceService.start();
    spanEventsService.start();
    startServices();
    // start last so other services can add connection listeners
    rpmServiceManager.start();
    // used for debugging purposes to quickly determine slow service startups
    ServiceTiming.setEndTime();
    ServiceTiming.logServiceTimings(getLogger());
}

Example 5 with HarvestServiceImpl

use of com.newrelic.agent.HarvestServiceImpl in project newrelic-java-agent by newrelic.

the class ErrorServiceTest method testErrorEventFasterHarvest.

@Test
public void testErrorEventFasterHarvest() throws Exception {
    String appName = ServiceFactory.getRPMService().getApplicationName();
    ServiceManager serviceManager = spy(ServiceFactory.getServiceManager());
    ServiceFactory.setServiceManager(serviceManager);
    HarvestServiceImpl harvestService = spy(new HarvestServiceImpl());
    doReturn(harvestService).when(serviceManager).getHarvestService();
    doReturn(0L).when(harvestService).getInitialDelay();
    ErrorServiceImpl errorService = (ErrorServiceImpl) ServiceFactory.getRPMService().getErrorService();
    errorService.addHarvestableToService();
    errorService.harvestable.configure(60, 10);
    Assert.assertEquals(10, errorService.getMaxSamplesStored());
    Map<String, Object> connectionInfo = new HashMap<>();
    Map<String, Object> eventHarvest = new HashMap<>();
    Map<String, Object> harvestLimits = new HashMap<>();
    // 5 is the lowest allowable value
    eventHarvest.put("report_period_ms", 5000L);
    eventHarvest.put("harvest_limits", harvestLimits);
    harvestLimits.put("error_event_data", 100L);
    connectionInfo.put("event_harvest_config", eventHarvest);
    harvestService.startHarvestables(ServiceFactory.getRPMService(), AgentConfigImpl.createAgentConfig(connectionInfo));
    Thread.sleep(500);
    TracedError error = ThrowableError.builder(ServiceFactory.getConfigService().getErrorCollectorConfig(appName), appName, "metricName", new Throwable(), 100).requestUri("requestUri").errorAttributes(Collections.singletonMap("test_attribute", "value")).build();
    ServiceFactory.getRPMService().getErrorService().reportError(error);
    Thread.sleep(5050);
    checkForEvent();
    ((MockRPMService) ServiceFactory.getRPMService()).clearEvents();
    ServiceFactory.getRPMService().getErrorService().reportError(error);
    Thread.sleep(5050);
    checkForEvent();
}