org.talend.camel.TalendProducer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camel-talendjob Show documentation
There is a newer version: 8.0.1.R2024-05-RT
/*
 * #%L
 * Talend ESB :: Camel Talend Job Component
 * %%
 * Copyright (C) 2011 - 2014 Talend Inc.
 * %%
 * 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.
 * #L%
 */

package org.talend.camel;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Map;

import org.apache.camel.Exchange;
import org.apache.camel.RuntimeCamelException;
import org.apache.camel.impl.DefaultProducer;
import org.apache.camel.util.ObjectHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import routines.system.api.TalendJob;

/**
 * 
 * The Talend producer.
 * 
 */
public class TalendProducer extends DefaultProducer {

    private static final transient Logger LOG = LoggerFactory.getLogger(TalendProducer.class);

    private Thread workingThread;

    public TalendProducer(TalendEndpoint endpoint) {
        super(endpoint);
    }

    public void process(Exchange exchange) throws Exception {
        final TalendEndpoint talendEndpoint = (TalendEndpoint) getEndpoint();
        final String context = talendEndpoint.getContext();
        final Collection args = new ArrayList();
        if (context != null) {
            args.add("--context=" + context);
        }
        if (talendEndpoint.isPropagateHeader()) {
            getParamsFromHeaders(exchange, args);
        }
        getParamsFromProperties(getEndpoint().getCamelContext().getProperties(), args);
        getParamsFromProperties(talendEndpoint.getEndpointProperties(), args);
        invokeTalendJob(talendEndpoint.getJobInstance(), args.toArray(new String[args.size()]), exchange);
    }

    private static void getParamsFromProperties(Map propertiesMap, Collection args) {
        if (propertiesMap != null) {
            for (Map.Entry entry : propertiesMap.entrySet()) {
                args.add("--context_param " + entry.getKey() + '=' + entry.getValue());
            }
        }
    }

    private static void getParamsFromHeaders(
            Exchange exchange, Collection args) {
        Map headers = exchange.getIn().getHeaders();
        for (Map.Entry header : headers.entrySet()) {
            Object headerValue = header.getValue();
            if (headerValue != null) {
                String headerStringValue = exchange.getContext().getTypeConverter()
                        .convertTo(String.class, exchange, headerValue);
                args.add("--context_param " + header.getKey() + '=' + headerStringValue);
            }
        }
    }

    private void invokeTalendJob(final TalendJob jobInstance, String[] args, Exchange exchange) {
        try {
            final Method setExchangeMethod =
                    jobInstance.getClass().getMethod("setExchange", new Class[]{Exchange.class});
            LOG.debug("Pass the exchange from route to Job");
            ObjectHelper.invokeMethod(setExchangeMethod, jobInstance, exchange);
        } catch (NoSuchMethodException e) {
            LOG.debug("No setExchange(exchange) method found in Job, the message data will be ignored");
        }
        if (LOG.isDebugEnabled()) {
            LOG.debug("Invoking Talend job '" + jobInstance.getClass().getCanonicalName() 
                    + ".runJob(String[] args)' with args: " + Arrays.toString(args));
        }

        // use local variable due to single component instance during parallel processing
        final Thread thread = Thread.currentThread();
        workingThread = thread;
        final ClassLoader oldContextCL = thread.getContextClassLoader();
        try {
            thread.setContextClassLoader(jobInstance.getClass().getClassLoader());
            int result = jobInstance.runJobInTOS(args);
            if (result != 0) {
                throw new RuntimeCamelException("Execution of Talend job '" 
                        + jobInstance.getClass().getCanonicalName() + "' with args: "
                        + Arrays.toString(args) + "' failed, see stderr for details");
                // Talend logs errors using System.err.println
            }
        } finally {
            thread.setContextClassLoader(oldContextCL);
            workingThread = null;
        }
    }

    @Override
    protected void doStop() throws Exception {
        super.doStop();
        if (null != workingThread) {
            LOG.info("Force terminate Talend job");
            workingThread.interrupt();
        }
    }
}