nl.vpro.camel.newrelic.TraceProducer Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2014 All rights reserved
 * VPRO The Netherlands
 */
package nl.vpro.camel.newrelic;

import com.newrelic.api.agent.NewRelic;
import org.apache.camel.Endpoint;
import org.apache.camel.Exchange;
import org.apache.camel.support.DefaultConsumer;
import org.apache.camel.support.DefaultProducer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Roelof Jan Koekoek
 * @since 1.1
 */
public class TraceProducer extends DefaultProducer {
    private static final Logger LOG = LoggerFactory.getLogger(TraceProducer.class);

    private TraceEndpoint endpoint;

    public TraceProducer(Endpoint endpoint) {
        super(endpoint);
        this.endpoint = (TraceEndpoint)endpoint;
    }

    //    @Trace(dispatcher = true)
    @Override
    public void process(Exchange exchange) throws Exception {
        if(endpoint.getConsumers().isEmpty()) {
            LOG.warn("No consumers available on endpoint: {} to process: {}", endpoint, exchange);
        } else {
            for(DefaultConsumer consumer : endpoint.getConsumers()) {
                String traceId = endpoint.getEndpointKey();
                NewRelic.setTransactionName(null, traceId);

                long start = System.nanoTime();
                consumer.getProcessor().process(exchange);
                long elapsedTime = System.nanoTime() - start;
                NewRelic.recordMetric(traceId, elapsedTime / 1000f);

                Exception error = exchange.getException();
                if(error == null) {
                    // handled exception
                    error = exchange.getProperty(Exchange.EXCEPTION_CAUGHT, Exception.class);
                }
                if(error != null) {
                    NewRelic.noticeError(error);
                }

                NewRelic.incrementCounter(traceId);
            }
        }
    }
}