io.opentracing.contrib.jms2.TracingJMSProducer Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017-2018 The OpenTracing Authors
 *
 * Licensed 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 io.opentracing.contrib.jms2;

import io.opentracing.Span;
import io.opentracing.Tracer;
import io.opentracing.contrib.jms.common.SpanJmsDecorator;
import io.opentracing.contrib.jms.common.TracingMessageUtils;
import io.opentracing.util.GlobalTracer;
import java.io.Serializable;
import java.util.Map;
import java.util.Set;
import javax.jms.BytesMessage;
import javax.jms.CompletionListener;
import javax.jms.Destination;
import javax.jms.JMSContext;
import javax.jms.JMSException;
import javax.jms.JMSProducer;
import javax.jms.MapMessage;
import javax.jms.Message;
import javax.jms.ObjectMessage;
import javax.jms.Session;
import javax.jms.TextMessage;


/**
 * Tracing decorator for JMS JMSProducer
 */
public class TracingJMSProducer implements JMSProducer {

  private final JMSProducer jmsProducer;
  private JMSContext jmsContext = null;
  private Session jmsSession = null;
  private final Tracer tracer;

  /**
   * GlobalTracer is used to get tracer
   */
  public TracingJMSProducer(JMSProducer jmsProducer, JMSContext jmsContext) {
    this(jmsProducer, jmsContext, GlobalTracer.get());
  }

  public TracingJMSProducer(JMSProducer jmsProducer, JMSContext jmsContext, Tracer tracer) {
    this.jmsProducer = jmsProducer;
    this.jmsContext = jmsContext;
    this.tracer = tracer;
  }

  /**
   * GlobalTracer is used to get tracer
   */
  public TracingJMSProducer(JMSProducer jmsProducer, Session jmsSession) {
    this(jmsProducer, jmsSession, GlobalTracer.get());
  }

  public TracingJMSProducer(JMSProducer jmsProducer, Session jmsSession, Tracer tracer) {
    this.jmsProducer = jmsProducer;
    this.jmsSession = jmsSession;
    this.tracer = tracer;
  }

  @Override
  public JMSProducer clearProperties() {
    jmsProducer.clearProperties();
    return this;
  }

  @Override
  public CompletionListener getAsync() {
    return jmsProducer.getAsync();
  }

  @Override
  public boolean getBooleanProperty(String arg0) {
    return jmsProducer.getBooleanProperty(arg0);
  }

  @Override
  public byte getByteProperty(String arg0) {
    return jmsProducer.getByteProperty(arg0);
  }

  @Override
  public long getDeliveryDelay() {
    return jmsProducer.getDeliveryDelay();
  }

  @Override
  public int getDeliveryMode() {
    return jmsProducer.getDeliveryMode();
  }

  @Override
  public boolean getDisableMessageID() {
    return jmsProducer.getDisableMessageID();
  }

  @Override
  public boolean getDisableMessageTimestamp() {
    return jmsProducer.getDisableMessageTimestamp();
  }

  @Override
  public double getDoubleProperty(String arg0) {
    return jmsProducer.getDoubleProperty(arg0);
  }

  @Override
  public float getFloatProperty(String arg0) {
    return jmsProducer.getFloatProperty(arg0);
  }

  @Override
  public int getIntProperty(String arg0) {
    return jmsProducer.getIntProperty(arg0);
  }

  @Override
  public String getJMSCorrelationID() {
    return jmsProducer.getJMSCorrelationID();
  }

  @Override
  public byte[] getJMSCorrelationIDAsBytes() {
    return jmsProducer.getJMSCorrelationIDAsBytes();
  }

  @Override
  public Destination getJMSReplyTo() {
    return jmsProducer.getJMSReplyTo();
  }

  @Override
  public String getJMSType() {
    return jmsProducer.getJMSType();
  }

  @Override
  public long getLongProperty(String arg0) {
    return jmsProducer.getLongProperty(arg0);
  }

  @Override
  public Object getObjectProperty(String arg0) {
    return jmsProducer.getObjectProperty(arg0);
  }

  @Override
  public int getPriority() {
    return jmsProducer.getPriority();
  }

  @Override
  public Set getPropertyNames() {
    return jmsProducer.getPropertyNames();
  }

  @Override
  public short getShortProperty(String arg0) {
    return jmsProducer.getShortProperty(arg0);
  }

  @Override
  public String getStringProperty(String arg0) {
    return jmsProducer.getStringProperty(arg0);
  }

  @Override
  public long getTimeToLive() {
    return jmsProducer.getTimeToLive();
  }

  @Override
  public boolean propertyExists(String arg0) {
    return jmsProducer.propertyExists(arg0);
  }

  @Override
  public JMSProducer send(Destination destination, Message message) {
    Span span = TracingMessageUtils.buildAndInjectSpan(destination, message, tracer);
    try {
      jmsProducer.send(destination, message);
    } catch (Throwable e) {
      SpanJmsDecorator.onError(e, span);
      throw e;
    } finally {
      span.finish();
    }
    return this;
  }

  @Override
  public JMSProducer send(Destination destination, String message) {
    TextMessage textMsg;
    try {
      textMsg = getTextMessage();
    } catch (JMSException e) {
      e.printStackTrace();
      jmsProducer.send(destination, message);
      return this;
    }
    if (textMsg == null) {
      // if textMsg is null, conversion failed
      jmsProducer.send(destination, message);
      return this;
    }

    try {
      textMsg.setText(message);
    } catch (JMSException e) {
      e.printStackTrace();
      jmsProducer.send(destination, message);
      return this;
    }
    return send(destination, textMsg);
  }

  private TextMessage getTextMessage() throws JMSException {
    TextMessage textMsg = null;
    if (jmsContext != null) {
      textMsg = jmsContext.createTextMessage();
    } else if (jmsSession != null) {
      textMsg = jmsSession.createTextMessage();
    }
    return textMsg;
  }

  @Override
  public JMSProducer send(Destination destination, Map arg1) {
    MapMessage mapMsg;
    try {
      mapMsg = getMapMessage();
    } catch (JMSException e) {
      e.printStackTrace();
      jmsProducer.send(destination, arg1);
      return this;
    }
    // if mapMsg is null, conversion failed
    if (mapMsg == null) {
      jmsProducer.send(destination, arg1);
      return this;
    }

    for (Map.Entry entry : arg1.entrySet()) {
      try {
        mapMsg.setObject(entry.getKey(), entry.getValue());
      } catch (JMSException e) {
        e.printStackTrace();
        jmsProducer.send(destination, arg1);
        return this;
      }
    }
    return send(destination, mapMsg);
  }

  private MapMessage getMapMessage() throws JMSException {
    MapMessage mapMsg = null;
    if (jmsContext != null) {
      mapMsg = jmsContext.createMapMessage();
    } else if (jmsSession != null) {
      mapMsg = jmsSession.createMapMessage();
    }
    return mapMsg;
  }

  @Override
  public JMSProducer send(Destination destination, byte[] arg1) {
    BytesMessage bytesMsg;
    try {
      bytesMsg = getBytesMessage();

    } catch (JMSException e) {
      e.printStackTrace();
      jmsProducer.send(destination, arg1);
      return this;
    }

    if (bytesMsg == null) {
      // if bytesMsg is null, conversion failed
      jmsProducer.send(destination, arg1);
      return this;
    }

    try {
      bytesMsg.writeBytes(arg1);
    } catch (JMSException e) {
      e.printStackTrace();
      jmsProducer.send(destination, arg1);
      return this;
    }

    return send(destination, bytesMsg);
  }

  private BytesMessage getBytesMessage() throws JMSException {
    BytesMessage bytesMsg = null;
    if (jmsContext != null) {
      bytesMsg = jmsContext.createBytesMessage();
    } else if (jmsSession != null) {
      bytesMsg = jmsSession.createBytesMessage();
    }
    return bytesMsg;
  }

  @Override
  public JMSProducer send(Destination destination, Serializable obj) {
    Message message;
    try {
      message = createJMSMessage(obj);
    } catch (JMSException e) {
      e.printStackTrace();
      jmsProducer.send(destination, obj);
      return this;
    }

    if (message == null) {
      // if message is null, conversion failed
      jmsProducer.send(destination, obj);
      return this;
    }

    return send(destination, message);
  }

  private Message createJMSMessage(Serializable obj) throws JMSException {
    if (obj instanceof String) {
      TextMessage textMsg = getTextMessage();
      // if textMsg is null, conversion failed
      if (textMsg == null) {
        return null;
      }
      textMsg.setText((String) obj);
      return textMsg;
    } else {
      ObjectMessage objMsg = getObjectMessage();
      if (objMsg == null) {
        return null;
      }
      objMsg.setObject(obj);
      return objMsg;
    }
  }

  private ObjectMessage getObjectMessage() throws JMSException {
    ObjectMessage objectMsg = null;
    if (jmsContext != null) {
      objectMsg = jmsContext.createObjectMessage();
    } else if (jmsSession != null) {
      objectMsg = jmsSession.createObjectMessage();
    }
    return objectMsg;
  }

  @Override
  public JMSProducer setAsync(CompletionListener arg0) {
    jmsProducer.setAsync(arg0);
    return this;
  }

  @Override
  public JMSProducer setDeliveryDelay(long arg0) {
    jmsProducer.setDeliveryDelay(arg0);
    return this;
  }

  @Override
  public JMSProducer setDeliveryMode(int arg0) {
    jmsProducer.setDeliveryMode(arg0);
    return this;
  }

  @Override
  public JMSProducer setDisableMessageID(boolean arg0) {
    jmsProducer.setDisableMessageID(arg0);
    return this;
  }

  @Override
  public JMSProducer setDisableMessageTimestamp(boolean arg0) {
    jmsProducer.setDisableMessageTimestamp(arg0);
    return this;
  }

  @Override
  public JMSProducer setJMSCorrelationID(String arg0) {
    jmsProducer.setJMSCorrelationID(arg0);
    return this;
  }

  @Override
  public JMSProducer setJMSCorrelationIDAsBytes(byte[] arg0) {
    jmsProducer.setJMSCorrelationIDAsBytes(arg0);
    return this;
  }

  @Override
  public JMSProducer setJMSReplyTo(Destination destination) {
    jmsProducer.setJMSReplyTo(destination);
    return this;
  }

  @Override
  public JMSProducer setJMSType(String arg0) {
    jmsProducer.setJMSType(arg0);
    return this;
  }

  @Override
  public JMSProducer setPriority(int arg0) {
    jmsProducer.setPriority(arg0);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, boolean arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, byte arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, short arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, int arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, long arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, float arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, double arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, String arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setProperty(String arg0, Object arg1) {
    jmsProducer.setProperty(arg0, arg1);
    return this;
  }

  @Override
  public JMSProducer setTimeToLive(long arg0) {
    jmsProducer.setTimeToLive(arg0);
    return this;
  }

}