Examples with DatabaseService com.newrelic.agent.database.DatabaseService used on opensource projects

Example 6 with DatabaseService

use of com.newrelic.agent.database.DatabaseService in project newrelic-java-agent by newrelic.

the class TransactionTraceBucketTest method createServiceManager.

private void createServiceManager(Map<String, Object> configMap) throws Exception {
    MockServiceManager serviceManager = new MockServiceManager();
    ServiceFactory.setServiceManager(serviceManager);
    AgentConfig agentConfig = createConfig(configMap);
    ConfigService configService = ConfigServiceFactory.createConfigService(agentConfig, configMap);
    serviceManager.setConfigService(configService);
    DatabaseService dbService = new DatabaseService();
    serviceManager.setDatabaseService(dbService);
    TransactionTraceService ttService = new TransactionTraceService();
    serviceManager.setTransactionTraceService(ttService);
    AttributesService attService = new AttributesService();
    serviceManager.setAttributesService(attService);
}

Example 7 with DatabaseService

use of com.newrelic.agent.database.DatabaseService in project newrelic-java-agent by newrelic.

the class IntrospectorServiceManager method setup.

private void setup(Map<String, Object> config) {
    configService = ConfigServiceFactory.createConfigService(AgentConfigImpl.createAgentConfig(config), Collections.<String, Object>emptyMap());
    ServiceFactory.setServiceManager(this);
    coreService = new IntrospectorCoreService();
    threadService = new ThreadService();
    environmentService = new EnvironmentServiceImpl();
    transactionService = new TransactionService();
    rpmConnectionService = new IntrospectorRPMConnectService();
    rpmServiceManager = new IntrospectorRPMServiceManager();
    transactionTraceService = new IntrospectorTransactionTraceService();
    asyncTxService = new AsyncTransactionService();
    profilerService = new ProfilerService();
    statsService = new IntrospectorStatsService();
    harvestService = new IntrospectorHarvestService();
    sqlTraceService = new SqlTraceServiceImpl();
    insightsService = new IntrospectorInsightsService();
    logSenderService = new IntrospectorLogSenderService();
    expirationService = new ExpirationService();
    dbService = new DatabaseService();
    jarCollectorService = new IgnoringJarCollectorService();
    distributedTraceService = new DistributedTraceServiceImpl();
    TransactionDataToDistributedTraceIntrinsics transactionDataToDistributedTraceIntrinsics = new TransactionDataToDistributedTraceIntrinsics(distributedTraceService);
    transactionEventsService = new TransactionEventsService(transactionDataToDistributedTraceIntrinsics);
    normalizationService = new NormalizationServiceImpl();
    extensionService = new ExtensionService(configService, ExtensionsLoadedListener.NOOP);
    tracerService = new TracerService();
    commandParser = new CommandParser();
    remoteInstrumentationService = new RemoteInstrumentationServiceImpl();
    sourceLanguageService = new SourceLanguageService();
    classTransformerService = new NoOpClassTransformerService();
    jmxService = new JmxService(configService.getDefaultAgentConfig().getJmxConfig());
    attributesService = new AttributesService();
    circuitBreakerService = new CircuitBreakerService();
    AgentConfig agentConfig = createAgentConfig(config, (Map) config.get("distributed_tracing"));
    distributedTraceService.connected(null, agentConfig);
    ReservoirManager<SpanEvent> reservoirManager = new CollectorSpanEventReservoirManager(configService);
    ReservoirManager.EventSender<SpanEvent> collectorSender = new CollectorSpanEventSender(rpmServiceManager);
    Consumer<SpanEvent> infiniteTracing = new Consumer<SpanEvent>() {

        @Override
        public void accept(SpanEvent spanEvent) {
        }
    };
    SpanEventCreationDecider spanEventCreationDecider = new SpanEventCreationDecider(configService);
    spanEventsService = new IntrospectorSpanEventService(agentConfig, reservoirManager, collectorSender, infiniteTracing, spanEventCreationDecider, environmentService, transactionDataToDistributedTraceIntrinsics);
    configService.addIAgentConfigListener((IntrospectorSpanEventService) spanEventsService);
    transactionService.addTransactionListener((IntrospectorSpanEventService) spanEventsService);
    try {
        transactionTraceService.start();
        transactionEventsService.start();
        transactionService.start();
    } catch (Exception e) {
    // fall through
    }
}

Example 8 with DatabaseService

use of com.newrelic.agent.database.DatabaseService 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 9 with DatabaseService

use of com.newrelic.agent.database.DatabaseService in project newrelic-java-agent by newrelic.

the class AgentTestUtils method createServiceManager.

public static MockServiceManager createServiceManager(Map<String, Object> configMap) throws Exception {
    AgentConfig config = AgentConfigImpl.createAgentConfig(configMap);
    MockServiceManager serviceManager = new MockServiceManager();
    ServiceFactory.setServiceManager(serviceManager);
    ConfigService configService = ConfigServiceFactory.createConfigService(config, configMap);
    serviceManager.setConfigService(configService);
    ThreadService threadService = new ThreadService();
    serviceManager.setThreadService(threadService);
    HarvestService harvestService = new MockHarvestService();
    serviceManager.setHarvestService(harvestService);
    TransactionService transactionService = new TransactionService();
    serviceManager.setTransactionService(transactionService);
    StatsService statsService = new StatsServiceImpl();
    serviceManager.setStatsService(statsService);
    DatabaseService dbService = new DatabaseService();
    serviceManager.setDatabaseService(dbService);
    SqlTraceService sqlTraceService = new SqlTraceServiceImpl();
    serviceManager.setSqlTraceService(sqlTraceService);
    MockCoreService agent = new MockCoreService();
    serviceManager.setCoreService(agent);
    TransactionTraceService transactionTraceService = new TransactionTraceService();
    serviceManager.setTransactionTraceService(transactionTraceService);
    MockRPMServiceManager rpmServiceManager = new MockRPMServiceManager();
    serviceManager.setRPMServiceManager(rpmServiceManager);
    MockRPMService rpmService = new MockRPMService();
    rpmService.setApplicationName("name");
    rpmService.setEverConnected(true);
    rpmService.setErrorService(new ErrorServiceImpl("name"));
    rpmServiceManager.setRPMService(rpmService);
    configService.start();
    serviceManager.start();
    sqlTraceService.start();
    return serviceManager;
}

Example 10 with DatabaseService

use of com.newrelic.agent.database.DatabaseService in project newrelic-java-agent by newrelic.

the class SlowQueryInfo method asSqlTrace.

public SqlTrace asSqlTrace() {
    Tracer tracer = getTracer();
    if (tracer instanceof SqlTracer) {
        DatabaseService dbService = ServiceFactory.getDatabaseService();
        dbService.runExplainPlan((SqlTracer) tracer);
    }
    return new SqlTraceImpl(this);
}