Skip to content

Commit 2b1ed9b

Browse files
committed
Changes requeted from review
1 parent 658380e commit 2b1ed9b

3 files changed

Lines changed: 14 additions & 8 deletions

File tree

dd-java-agent/instrumentation/jms/src/main/java/datadog/trace/instrumentation/jms/JMSDecorator.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,10 @@ public static void logJMSException(JMSException ex) {
115115
}
116116

117117
public static String messageTechnology(Message m) {
118+
if (null == m) {
119+
return "null";
120+
}
121+
118122
String messageClass = m.getClass().getName();
119123

120124
if (messageClass.startsWith("com.amazon.sqs")) {

dd-java-agent/instrumentation/jms/src/main/java/datadog/trace/instrumentation/jms/JMSMessageConsumerInstrumentation.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ public static void afterReceive(
153153

154154
if (Config.get().isDataStreamsEnabled()) {
155155
final String tech = messageTechnology(message);
156-
if (tech == "ibmmq") { // Initial release only supports DSM in JMS for IBM MQ
156+
if ("ibmmq".equals(tech)) { // Initial release only supports DSM in JMS for IBM MQ
157157
DataStreamsTags tags =
158158
create(tech, INBOUND, consumerState.getConsumerBaseResourceName().toString());
159159
DataStreamsContext dsmContext = DataStreamsContext.fromTags(tags);

dd-java-agent/instrumentation/jms/src/main/java/datadog/trace/instrumentation/jms/JMSMessageProducerInstrumentation.java

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -82,12 +82,14 @@ public static AgentScope beforeSend(
8282
String destinationName;
8383
try {
8484
// fall-back when producer wasn't created via standard Session.createProducer API
85-
Destination destination = producer.getDestination();
86-
boolean isQueue = PRODUCER_DECORATE.isQueue(destination);
87-
destinationName = PRODUCER_DECORATE.getDestinationName(destination);
8885
if (null != producerState) {
8986
resourceName = producerState.getResourceName();
87+
Destination destination = producer.getDestination();
88+
destinationName = PRODUCER_DECORATE.getDestinationName(destination);
9089
} else {
90+
Destination destination = producer.getDestination();
91+
destinationName = PRODUCER_DECORATE.getDestinationName(destination);
92+
boolean isQueue = PRODUCER_DECORATE.isQueue(destination);
9193
resourceName = PRODUCER_DECORATE.toResourceName(destinationName, isQueue);
9294
}
9395
} catch (Exception ignored) {
@@ -99,11 +101,11 @@ public static AgentScope beforeSend(
99101
PRODUCER_DECORATE.afterStart(span);
100102
PRODUCER_DECORATE.onProduce(span, resourceName);
101103

102-
if (destinationName != null
104+
if (null != destinationName
103105
&& !destinationName.isEmpty()
104106
&& Config.get().isDataStreamsEnabled()) {
105107
final String tech = messageTechnology(message);
106-
if (tech == "ibmmq") { // Initial release only supports DSM in JMS for IBM MQ
108+
if ("ibmmq".equals(tech)) { // Initial release only supports DSM in JMS for IBM MQ
107109
DataStreamsTags tags = create(tech, OUTBOUND, destinationName);
108110
DataStreamsContext dsmContext = DataStreamsContext.fromTags(tags);
109111
AgentTracer.get().getDataStreamsMonitoring().setCheckpoint(span, dsmContext);
@@ -158,11 +160,11 @@ public static AgentScope beforeSend(
158160
PRODUCER_DECORATE.afterStart(span);
159161
PRODUCER_DECORATE.onProduce(span, resourceName);
160162

161-
if (destinationName != null
163+
if (null != destinationName
162164
&& !destinationName.isEmpty()
163165
&& Config.get().isDataStreamsEnabled()) {
164166
final String tech = messageTechnology(message);
165-
if (tech == "ibmmq") { // Initial release only supports DSM in JMS for IBM MQ
167+
if ("ibmmq".equals(tech)) { // Initial release only supports DSM in JMS for IBM MQ
166168
DataStreamsTags tags = create(tech, OUTBOUND, destinationName);
167169
DataStreamsContext dsmContext = DataStreamsContext.fromTags(tags);
168170
AgentTracer.get().getDataStreamsMonitoring().setCheckpoint(span, dsmContext);

0 commit comments

Comments
 (0)