Add LOG4J2 domain to the Log4j 2.x agent

[lttng-ust.git] / src / lib / lttng-ust-java-agent / java / lttng-ust-agent-log4j2 / org / lttng / ust / agent / log4j2 / LttngLogAppender.java

diff --git a/src/lib/lttng-ust-java-agent/java/lttng-ust-agent-log4j2/org/lttng/ust/agent/log4j2/LttngLogAppender.java b/src/lib/lttng-ust-java-agent/java/lttng-ust-agent-log4j2/org/lttng/ust/agent/log4j2/LttngLogAppender.java
index 3b7afa5b72dadb4bd6510b42c94e423d402773d9..cebd18945e152cdb81ccc8461aa59e5a3b90eac3 100644 (file)
--- a/src/lib/lttng-ust-java-agent/java/lttng-ust-agent-log4j2/org/lttng/ust/agent/log4j2/LttngLogAppender.java
+++ b/src/lib/lttng-ust-java-agent/java/lttng-ust-agent-log4j2/org/lttng/ust/agent/log4j2/LttngLogAppender.java
@@ -92,7 +92,9 @@ public final class LttngLogAppender extends AbstractAppender implements ILttngHa
 
                /* Register to the relevant agent. */
                if (domain == LttngLog4j2Agent.Domain.LOG4J) {
-                       agent = LttngLog4j2Agent.getInstance();
+                       agent = LttngLog4j2Agent.getLog4j1Instance();
+               } else if (domain == LttngLog4j2Agent.Domain.LOG4J2) {
+                       agent = LttngLog4j2Agent.getLog4j2Instance();
                } else {
                        throw new IllegalArgumentException("Unsupported domain '" + domain + "'");
                }
@@ -239,6 +241,6 @@ public final class LttngLogAppender extends AbstractAppender implements ILttngHa
 
                LttngLog4j2Api.tracepointWithContext(message, loggername, classname, methodname, filename, line,
                                event.getTimeMillis(), event.getLevel().intLevel(), event.getThreadName(),
-                               contextInfo.getEntriesArray(), contextInfo.getStringsArray());
+                               contextInfo.getEntriesArray(), contextInfo.getStringsArray(), agent.getDomain() == Domain.LOG4J);
        }
 }