ru.yandex.qatools.camelot.common.ClayProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camelot-commons Show documentation
There is a newer version: 2.5.4
package ru.yandex.qatools.camelot.common;

import org.apache.camel.CamelContext;
import org.apache.camel.CamelContextAware;
import org.apache.camel.Exchange;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import ru.yandex.qatools.camelot.api.annotations.InjectHeader;
import ru.yandex.qatools.camelot.api.annotations.InjectHeaders;

import java.lang.reflect.Field;

import static ru.yandex.qatools.camelot.util.ContextUtils.autowireFields;
import static ru.yandex.qatools.camelot.util.ReflectUtil.*;

/**
 * @author Ilya Sadykov (mailto: [email protected])
 */
public abstract class ClayProcessor implements CamelContextAware, ApplicationContextAware {
    protected final Logger LOGGER = LoggerFactory.getLogger(getClass());
    protected final MessagesSerializer serializer;
    protected ClassLoader classLoader;
    protected ApplicationContext applicationContext;
    protected CamelContext camelContext;

    protected ClayProcessor(ClassLoader classLoader, MessagesSerializer serializer) {
        this.classLoader = classLoader;
        this.serializer = serializer;
    }

    /**
     * Returns the currently set classloader
     */
    public ClassLoader getClassLoader() {
        return classLoader;
    }

    /**
     * Set the classloader which is used to serialize/fromBytes & load objects from input exchange
     */
    public void setClassLoader(ClassLoader classLoader) {
        this.classLoader = classLoader;
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

    @Override
    public CamelContext getCamelContext() {
        return camelContext;
    }

    @Override
    public void setCamelContext(CamelContext camelContext) {
        this.camelContext = camelContext;
    }

    protected Exchange processAfterIn(Exchange exchange) {
        return (serializer != null) ? serializer.preProcess(exchange, classLoader) : exchange;
    }

    protected Exchange processBeforeOut(Exchange exchange) {
        return (serializer != null) ? serializer.postProcess(exchange, classLoader) : exchange;
    }

    protected void injectFields(Object procInstance, Exchange exchange) {
        try {
            autowireFields(procInstance, applicationContext, camelContext);
        } catch (Exception e) {
            LOGGER.error("Could not autowire the Spring or Camel context fields: ", e);
        }
        for (Field field : getFieldsInClassHierarchy(procInstance.getClass())) {
            try {
                boolean oldAccessible = field.isAccessible();
                if (!field.isAccessible()) {
                    field.setAccessible(true);
                }
                if (getAnnotation(field, InjectHeader.class) != null) {
                    String headerName = (String) getAnnotationValue(field, InjectHeader.class, "value");
                    field.set(procInstance, exchange.getIn().getHeader(headerName));
                }
                if (getAnnotation(field, InjectHeaders.class) != null) {
                    field.set(procInstance, exchange.getIn().getHeaders());
                }
                field.setAccessible(oldAccessible);
            } catch (Exception e) {
                LOGGER.error("Inject field " + field.getName() + " of FSM " + procInstance + " error: ", e);
            }
        }
    }

}