org.apache.activemq.artemis.jms.example.OpenTelemetryPlugin Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.activemq.artemis.jms.example;
import java.io.InputStream;
import java.util.Properties;
import io.opentelemetry.api.GlobalOpenTelemetry;
import io.opentelemetry.api.trace.Span;
import io.opentelemetry.api.trace.SpanBuilder;
import io.opentelemetry.api.trace.SpanKind;
import io.opentelemetry.api.trace.StatusCode;
import io.opentelemetry.api.trace.Tracer;
import io.opentelemetry.sdk.OpenTelemetrySdk;
import io.opentelemetry.sdk.autoconfigure.AutoConfiguredOpenTelemetrySdk;
import org.apache.activemq.artemis.api.core.ActiveMQException;
import org.apache.activemq.artemis.api.core.Message;
import org.apache.activemq.artemis.core.postoffice.RoutingStatus;
import org.apache.activemq.artemis.core.server.MessageReference;
import org.apache.activemq.artemis.core.server.ServerConsumer;
import org.apache.activemq.artemis.core.server.ServerSession;
import org.apache.activemq.artemis.core.server.plugin.ActiveMQServerPlugin;
import org.apache.activemq.artemis.core.transaction.Transaction;
public class OpenTelemetryPlugin implements ActiveMQServerPlugin {
private static final String OPERATION_NAME = "ArtemisMessageDelivery";
private static OpenTelemetrySdk sdk = initopentelemetry();
private static Tracer tracer = GlobalOpenTelemetry.getTracer(OpenTelemetryPlugin.class.getName());
public static OpenTelemetrySdk initopentelemetry() {
try {
InputStream input = OpenTelemetryPlugin.class.getClassLoader().getResourceAsStream("tracing.properties");
if (input == null) {
throw new NullPointerException("Unable to find tracing.properties file");
}
Properties prop = new Properties(System.getProperties());
prop.load(input);
System.setProperties(prop);
sdk = AutoConfiguredOpenTelemetrySdk.initialize().getOpenTelemetrySdk();
} catch (Throwable t) {
t.printStackTrace();
}
return sdk;
}
@Override
public void beforeSend(ServerSession session,
Transaction tx,
Message message,
boolean direct,
boolean noAutoCreateQueue) throws ActiveMQException {
// TODO: find a way to inject a context based in https://github.com/kittylyst/OTel/blob/8faea2aab7b19680f78804ddff3d59b7b1135aab/src/main/java/io/opentelemetry/examples/utils/OpenTelemetryConfig.java#L96-L100
// if a client has the metadata, we should get the parent context here
SpanBuilder spanBuilder = getTracer().spanBuilder(OPERATION_NAME).setAttribute("message", message.toString()).setSpanKind(SpanKind.SERVER);
Span span = spanBuilder.startSpan();
message.setUserContext(Span.class, span);
}
@Override
public void afterSend(Transaction tx,
Message message,
boolean direct,
boolean noAutoCreateQueue,
RoutingStatus result) throws ActiveMQException {
Span span = getSpan(message);
span.addEvent("send " + result.name());
}
@Override
public void afterDeliver(ServerConsumer consumer, MessageReference reference) throws ActiveMQException {
Span span = (Span) reference.getMessage().getUserContext(Span.class);
span.addEvent("deliver " + consumer.getSessionName());
span.end();
}
@Override
public void onSendException(ServerSession session,
Transaction tx,
Message message,
boolean direct,
boolean noAutoCreateQueue,
Exception e) throws ActiveMQException {
getSpan(message).setStatus(StatusCode.ERROR).recordException(e);
}
public Tracer getTracer() {
return tracer;
}
public void setTracer(Tracer myTracer) {
tracer = myTracer;
}
private Span getSpan(Message message) {
Span span = (Span) message.getUserContext(Span.class);
return span;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy