Examples with TransactionSegment com.newrelic.agent.trace.TransactionSegment used on opensource projects

Example 1 with TransactionSegment

use of com.newrelic.agent.trace.TransactionSegment in project newrelic-java-agent by newrelic.

the class DefaultSqlTracerTest method testInsertValuesTruncates.

@Test
public void testInsertValuesTruncates() throws Exception {
    String sql = " Insert  Into  test       \t\t   VALUES ";
    for (int i = 0; i < 100; i++) {
        sql += "(333,444,555,666,777,888,999,111,222),";
    }
    DefaultSqlTracer tracer = newTracer(sql);
    tracer.finish(Opcodes.RETURN, null);
    assertEquals(getScopedMetric("test", "insert"), tracer.getMetricName());
    SqlObfuscator sqlObfuscator = ServiceFactory.getDatabaseService().getDefaultSqlObfuscator();
    TransactionTracerConfig ttConfig = ServiceFactory.getConfigService().getDefaultAgentConfig().getTransactionTracerConfig();
    TransactionSegment segment = new TransactionSegment(ttConfig, sqlObfuscator, 0, tracer);
    JSONArray json = (JSONArray) AgentHelper.serializeJSON(segment);
    JSONObject params = (JSONObject) json.get(3);
    String truncatedSql = (String) params.get(SqlTracer.SQL_OBFUSCATED_PARAMETER_NAME);
    assertNotNull(truncatedSql);
    assertTrue(truncatedSql.length() < sql.length());
    assertTrue(truncatedSql.endsWith("more chars)"));
}

Example 2 with TransactionSegment

use of com.newrelic.agent.trace.TransactionSegment in project newrelic-java-agent by newrelic.

the class DefaultSqlTracerTest method stackTrace.

@SuppressWarnings("unchecked")
@Test
public void stackTrace() throws Exception {
    DefaultSqlTracer tracer = newTracer("select * from dude");
    tracer.storeStackTrace();
    SqlObfuscator sqlObfuscator = ServiceFactory.getDatabaseService().getDefaultSqlObfuscator();
    TransactionTracerConfig ttConfig = ServiceFactory.getConfigService().getDefaultAgentConfig().getTransactionTracerConfig();
    TransactionSegment segment = new TransactionSegment(ttConfig, sqlObfuscator, 0, tracer);
    JSONArray json = (JSONArray) AgentHelper.serializeJSON(segment);
    Map params = (Map) json.get(3);
    List backtrace = (List) params.get(DefaultSqlTracer.BACKTRACE_PARAMETER_NAME);
    assertTrue(backtrace.size() > 2);
}

Example 3 with TransactionSegment

use of com.newrelic.agent.trace.TransactionSegment in project newrelic-java-agent by newrelic.

the class DefaultSqlTracerTest method testHighSecurityNormalFinish.

@SuppressWarnings("unchecked")
@Test
public void testHighSecurityNormalFinish() throws Exception {
    TransactionTracerConfig ttConfig = mock(TransactionTracerConfig.class);
    // These are set to -1 to allow this tracer to always go through slow sql logic. Otherwise if the tracer
    // executes in less than 1 millisecond (e.g. 0 ms rounded) it won't be greater than the threshold (0) so
    // we won't run this through the slow sql code.
    when(ttConfig.getExplainThresholdInMillis()).thenReturn(-1d);
    when(ttConfig.getExplainThresholdInNanos()).thenReturn(-1d);
    when(ttConfig.getRecordSql()).thenReturn(RecordSql.obfuscated.name());
    when(ttConfig.isExplainEnabled()).thenReturn(true);
    when(ttConfig.isEnabled()).thenReturn(true);
    AgentConfig agentConfig = AgentHelper.mockAgentConfig(ttConfig);
    when(agentConfig.isHighSecurity()).thenReturn(true);
    when(agentConfig.getTransactionTracerConfig()).thenReturn(ttConfig);
    String inputSql = "select * from dudette where ssn = 123456789";
    String obfuscatedInputSql = "select * from dudette where ssn = ?";
    DefaultSqlTracer tracer = newTracer(inputSql);
    assertTrue(tracer.getAgentAttributes().isEmpty());
    tracer.finish(Opcodes.RETURN, null);
    SqlObfuscator sqlObfuscator = ServiceFactory.getDatabaseService().getDefaultSqlObfuscator();
    TransactionSegment segment = new TransactionSegment(ttConfig, sqlObfuscator, 0, tracer);
    // exclusive_duration_millis, sql, sql_obfuscated, host, port_path_or_id
    assertEquals(5, tracer.getAgentAttributes().size());
    // shouldn't be obfuscated yet
    assertEquals(inputSql, (String) tracer.getAgentAttributes().get("sql"));
    JSONArray json = (JSONArray) AgentHelper.serializeJSON(segment);
    Map params = (Map) json.get(3);
    // the name changes when it is obfuscated
    String sqlStatement = (String) params.get("sql_obfuscated");
    assertEquals(obfuscatedInputSql, sqlStatement);
    String rawSql = (String) params.get("sql");
    // raw sql should be null since high security is on
    assertNull(rawSql);
    // Ensure that we do not send up unobfuscated sql in the intrinsics
    Transaction transaction = Transaction.getTransaction(false);
    assertNotNull(transaction);
    Map<String, Object> intrinsics = transaction.getIntrinsicAttributes();
    assertNotNull(intrinsics);
    assertNull(intrinsics.get(SqlTracer.SQL_PARAMETER_NAME));
}

Example 4 with TransactionSegment

use of com.newrelic.agent.trace.TransactionSegment in project newrelic-java-agent by newrelic.

the class ExternalTracerTest method testExternalTracerExcludeUri.

@Test
public void testExternalTracerExcludeUri() throws Exception {
    before("/com/newrelic/agent/config/exclude_request_uri.yml");
    DefaultTracer tracer = prepareTracer();
    TransactionTracerConfig ttConfig = ServiceFactory.getConfigService().getDefaultAgentConfig().getTransactionTracerConfig();
    SqlObfuscator sqlObfuscator = ServiceFactory.getDatabaseService().getDefaultSqlObfuscator();
    TransactionSegment segment = tracer.getTransactionSegment(ttConfig, sqlObfuscator, 0, null);
    TransactionStats stats = tracer.getTransactionActivity().getTransactionStats();
    String uri = "http://myhost:1234/Parameters";
    String queryParams = "?data=confidential";
    tracer.reportAsExternal(GenericParameters.library("MyLibrary").uri(URI.create(uri + queryParams)).procedure("other").build());
    tracer.recordMetrics(stats);
    String segmentUri = segment.getUri();
    tracer.finish(Opcodes.RETURN, null);
    Assert.assertEquals(null, segmentUri);
}

Example 5 with TransactionSegment

use of com.newrelic.agent.trace.TransactionSegment in project newrelic-java-agent by newrelic.

the class ExternalTracerTest method testExternalTracerIncludeUri.

@Test
public void testExternalTracerIncludeUri() throws Exception {
    before("/com/newrelic/agent/config/span_events.yml");
    DefaultTracer tracer = prepareTracer();
    TransactionTracerConfig ttConfig = ServiceFactory.getConfigService().getDefaultAgentConfig().getTransactionTracerConfig();
    SqlObfuscator sqlObfuscator = ServiceFactory.getDatabaseService().getDefaultSqlObfuscator();
    TransactionSegment segment = tracer.getTransactionSegment(ttConfig, sqlObfuscator, 0, null);
    TransactionStats stats = tracer.getTransactionActivity().getTransactionStats();
    String uri = "http://myhost:1234/Parameters";
    String queryParams = "?data=confidential";
    tracer.reportAsExternal(GenericParameters.library("MyLibrary").uri(URI.create(uri + queryParams)).procedure("other").build());
    tracer.recordMetrics(stats);
    String segmentUri = segment.getUri();
    tracer.finish(Opcodes.RETURN, null);
    Assert.assertNotNull(segmentUri);
}