Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add marker attribute for Logback #6652

Merged
merged 5 commits into from
Sep 22, 2022
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -56,4 +56,5 @@ tasks.withType<Test>().configureEach {
jvmArgs("-Dotel.instrumentation.logback-appender.experimental.capture-mdc-attributes=*")
jvmArgs("-Dotel.instrumentation.logback-appender.experimental-log-attributes=true")
jvmArgs("-Dotel.instrumentation.logback-appender.experimental.capture-code-attributes=true")
jvmArgs("-Dotel.instrumentation.logback-appender.experimental.capture-marker-attributes=true")
}
Original file line number Diff line number Diff line change
Expand Up @@ -24,14 +24,20 @@ public final class LogbackSingletons {
boolean captureCodeAttributes =
config.getBoolean(
"otel.instrumentation.logback-appender.experimental.capture-code-attributes", false);
boolean captureMarkerAttributes =
config.getBoolean(
"otel.instrumentation.logback-appender.experimental.capture-marker-attributes", false);
trask marked this conversation as resolved.
Show resolved Hide resolved
List<String> captureMdcAttributes =
config.getList(
"otel.instrumentation.logback-appender.experimental.capture-mdc-attributes",
emptyList());

mapper =
new LoggingEventMapper(
captureExperimentalAttributes, captureMdcAttributes, captureCodeAttributes);
captureExperimentalAttributes,
captureMdcAttributes,
captureCodeAttributes,
captureMarkerAttributes);
}

public static LoggingEventMapper mapper() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,8 @@
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;
import org.slf4j.Marker;
import org.slf4j.MarkerFactory;

class LogbackTest extends AgentInstrumentationSpecification {

Expand All @@ -34,6 +36,7 @@ class LogbackTest extends AgentInstrumentationSpecification {

private static final Logger abcLogger = LoggerFactory.getLogger("abc");
private static final Logger defLogger = LoggerFactory.getLogger("def");
private static final Logger ghiLogger = LoggerFactory.getLogger("ghi");

private static Stream<Arguments> provideParameters() {
return Stream.of(
Expand Down Expand Up @@ -219,6 +222,25 @@ void testMdc() {
.containsEntry(SemanticAttributes.CODE_FILEPATH, "LogbackTest.java"));
}

@Test
public void testMarker() {

String markerName = "aMarker";
Marker marker = MarkerFactory.getMarker(markerName);

ghiLogger.info(marker, "Message");
trask marked this conversation as resolved.
Show resolved Hide resolved

await().untilAsserted(() -> assertThat(testing.logs().size()).isEqualTo(1));

LogData log = getLogs().get(0);

assertThat(log)
.hasAttributesSatisfying(
attributes ->
assertThat(attributes)
.containsEntry(AttributeKey.stringKey("log.marker"), markerName));
}

private static void performLogging(
Logger logger,
OneArgLoggerMethod oneArgLoggerMethod,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@

<logger name="abc" level="INFO"/>
<logger name="def" level="WARN"/>
<logger name="ghi" level="INFO"/>

<root level="INFO">
<appender-ref ref="console"/>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ public class OpenTelemetryAppender extends UnsynchronizedAppenderBase<ILoggingEv

private volatile boolean captureExperimentalAttributes = false;
private volatile boolean captureCodeAttributes = false;
private volatile boolean captureMarkerAttributes = false;
trask marked this conversation as resolved.
Show resolved Hide resolved
private volatile List<String> captureMdcAttributes = emptyList();

private volatile LoggingEventMapper mapper;
Expand All @@ -36,7 +37,10 @@ public OpenTelemetryAppender() {}
public void start() {
mapper =
new LoggingEventMapper(
captureExperimentalAttributes, captureMdcAttributes, captureCodeAttributes);
captureExperimentalAttributes,
captureMdcAttributes,
captureCodeAttributes,
captureMarkerAttributes);
super.start();
}

Expand Down Expand Up @@ -77,6 +81,16 @@ public void setCaptureCodeAttributes(boolean captureCodeAttributes) {
this.captureCodeAttributes = captureCodeAttributes;
}

/**
* Sets whether the marker attributes (file name, class name, method name and line number) should
* be set to logs.
*
* @param captureMarkerAttributes To enable or disable the marker attributes
*/
public void setCaptureMarkerAttributes(boolean captureMarkerAttributes) {
this.captureMarkerAttributes = captureMarkerAttributes;
}

/** Configures the {@link MDC} attributes that will be copied to logs. */
public void setCaptureMdcAttributes(String attributes) {
if (attributes != null) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import org.slf4j.Marker;

/**
* This class is internal and is hence not for public use. Its APIs are unstable and can change at
Expand All @@ -31,18 +32,23 @@ public final class LoggingEventMapper {

private static final Cache<String, AttributeKey<String>> mdcAttributeKeys = Cache.bounded(100);

private static final AttributeKey<String> LOG_MARKER = AttributeKey.stringKey("log.marker");
trask marked this conversation as resolved.
Show resolved Hide resolved

private final boolean captureExperimentalAttributes;
private final List<String> captureMdcAttributes;
private final boolean captureAllMdcAttributes;
private final boolean captureCodeAttributes;
private final boolean captureMarkerAttributes;

public LoggingEventMapper(
boolean captureExperimentalAttributes,
List<String> captureMdcAttributes,
boolean captureCodeAttributes) {
boolean captureCodeAttributes,
boolean captureMarkerAttributes) {
this.captureExperimentalAttributes = captureExperimentalAttributes;
this.captureCodeAttributes = captureCodeAttributes;
this.captureMdcAttributes = captureMdcAttributes;
this.captureMarkerAttributes = captureMarkerAttributes;
this.captureAllMdcAttributes =
captureMdcAttributes.size() == 1 && captureMdcAttributes.get(0).equals("*");
}
Expand Down Expand Up @@ -125,6 +131,14 @@ private void mapLoggingEvent(LogRecordBuilder builder, ILoggingEvent loggingEven
}
}

if (captureMarkerAttributes) {
Marker marker = loggingEvent.getMarker();
if (marker != null) {
String markerName = marker.getName();
attributes.put(LOG_MARKER, markerName);
}
}

builder.setAllAttributes(attributes.build());

// span context
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,8 @@
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;
import org.slf4j.Marker;
import org.slf4j.MarkerFactory;

class OpenTelemetryAppenderConfigTest {

Expand Down Expand Up @@ -101,7 +103,9 @@ private static Span runWithSpan(String spanName, Runnable runnable) {
@Test
void logWithExtras() {
Instant start = Instant.now();
logger.info("log message 1", new IllegalStateException("Error!"));
String markerName = "aMarker";
Marker marker = MarkerFactory.getMarker(markerName);
logger.info(marker, "log message 1", new IllegalStateException("Error!"));

List<LogData> logDataList = logExporter.getFinishedLogItems();
assertThat(logDataList).hasSize(1);
Expand All @@ -114,7 +118,8 @@ void logWithExtras() {
.isLessThan(TimeUnit.MILLISECONDS.toNanos(Instant.now().toEpochMilli()));
assertThat(logData.getSeverity()).isEqualTo(Severity.INFO);
assertThat(logData.getSeverityText()).isEqualTo("INFO");
assertThat(logData.getAttributes().size()).isEqualTo(3 + 4); // 4 code attributes
assertThat(logData.getAttributes().size())
.isEqualTo(3 + 4 + 1); // 3 exception attributes, 4 code attributes, 1 marker attribute
assertThat(logData.getAttributes().get(SemanticAttributes.EXCEPTION_TYPE))
.isEqualTo(IllegalStateException.class.getName());
assertThat(logData.getAttributes().get(SemanticAttributes.EXCEPTION_MESSAGE))
Expand All @@ -135,6 +140,9 @@ void logWithExtras() {

Long lineNumber = logData.getAttributes().get(SemanticAttributes.CODE_LINENO);
assertThat(lineNumber).isGreaterThan(1);

String logMarker = logData.getAttributes().get(AttributeKey.stringKey("log.marker"));
assertThat(logMarker).isEqualTo(markerName);
}

@Test
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ class LoggingEventMapperTest {
@Test
void testDefault() {
// given
LoggingEventMapper mapper = new LoggingEventMapper(false, emptyList(), false);
LoggingEventMapper mapper = new LoggingEventMapper(false, emptyList(), false, false);
Map<String, String> contextData = new HashMap<>();
contextData.put("key1", "value1");
contextData.put("key2", "value2");
Expand All @@ -38,7 +38,7 @@ void testDefault() {
@Test
void testSome() {
// given
LoggingEventMapper mapper = new LoggingEventMapper(false, singletonList("key2"), false);
LoggingEventMapper mapper = new LoggingEventMapper(false, singletonList("key2"), false, false);
Map<String, String> contextData = new HashMap<>();
contextData.put("key1", "value1");
contextData.put("key2", "value2");
Expand All @@ -55,7 +55,7 @@ void testSome() {
@Test
void testAll() {
// given
LoggingEventMapper mapper = new LoggingEventMapper(false, singletonList("*"), false);
LoggingEventMapper mapper = new LoggingEventMapper(false, singletonList("*"), false, false);
Map<String, String> contextData = new HashMap<>();
contextData.put("key1", "value1");
contextData.put("key2", "value2");
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
class="io.opentelemetry.instrumentation.logback.appender.v1_0.OpenTelemetryAppender">
<captureExperimentalAttributes>false</captureExperimentalAttributes>
<captureCodeAttributes>true</captureCodeAttributes>
<captureMarkerAttributes>true</captureMarkerAttributes>
<captureMdcAttributes>*</captureMdcAttributes>
</appender>

Expand Down