io.syndesis.integration.component.proxy.Processors Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of integration-component-proxy Show documentation
The newest version!
/*
 * Copyright (C) 2016 Red Hat, 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.
 */
package io.syndesis.integration.component.proxy;

import java.util.Collection;

import org.apache.camel.CamelContext;
import org.apache.camel.Exchange;
import org.apache.camel.Message;
import org.apache.camel.Processor;
import org.apache.camel.model.language.SimpleExpression;
import org.apache.camel.processor.Pipeline;
import org.apache.camel.processor.PollEnricher;
import org.apache.camel.processor.aggregate.AggregationStrategy;

public final class Processors {

    private static final class EnrichmentAggregator implements AggregationStrategy {
        public static final AggregationStrategy INSTANCE = new EnrichmentAggregator();

        @Override
        public Exchange aggregate(final Exchange previousExchange, final Exchange currentExchange) {
            final Message previousMessage = previousExchange.getIn();
            if (currentExchange==null) {
                previousExchange.getIn().setBody(null);
                previousExchange.setOut(null);
            } else {
                final Message currentMessage = currentExchange.getIn();
                previousMessage.setBody(currentMessage.getBody());
                if (currentMessage.hasAttachments()) {
                    previousMessage.getAttachmentObjects().putAll(currentMessage.getAttachmentObjects());
                }
                previousMessage.getHeaders().putAll(currentMessage.getHeaders());
                previousExchange.getProperties().putAll(currentExchange.getProperties());
            }
            return previousExchange;
        }
    }

    private Processors() {
        // utility class
    }

    public static void addAfterConsumer(final ComponentProxyComponent component, final Processor processor) {
        final Processor existing = component.getAfterConsumer();

        if (existing == null) {
            component.setAfterConsumer(processor);
        } else if (existing instanceof Pipeline) {
            final Pipeline pipeline = (Pipeline) existing;
            final Collection processors = pipeline.getProcessors();
            processors.add(processor);
        } else {
            final Processor pipeline = Pipeline.newInstance(component.getCamelContext(), existing, processor);
            component.setAfterConsumer(pipeline);
        }
    }

    public static void addAfterProducer(final ComponentProxyComponent component, final Processor processor) {
        final Processor existing = component.getAfterProducer();

        if (existing == null) {
            component.setAfterProducer(processor);
        } else if (existing instanceof Pipeline) {
            final Pipeline pipeline = (Pipeline) existing;
            final Collection processors = pipeline.getProcessors();
            processors.add(processor);
        } else {
            final Processor pipeline = Pipeline.newInstance(component.getCamelContext(), existing, processor);
            component.setAfterProducer(pipeline);
        }
    }

    public static void addBeforeConsumer(final ComponentProxyComponent component, final Processor processor) {
        final Processor existing = component.getBeforeConsumer();

        if (existing == null) {
            component.setBeforeConsumer(processor);
        } else if (existing instanceof Pipeline) {
            final Pipeline pipeline = (Pipeline) existing;
            final Collection processors = pipeline.getProcessors();
            processors.add(processor);
        } else {
            final Processor pipeline = Pipeline.newInstance(component.getCamelContext(), existing, processor);
            component.setBeforeConsumer(pipeline);
        }
    }

    public static void addBeforeProducer(final ComponentProxyComponent component, final Processor processor) {
        final Processor existing = component.getBeforeProducer();

        if (existing == null) {
            component.setBeforeProducer(processor);
        } else if (existing instanceof Pipeline) {
            final Pipeline pipeline = (Pipeline) existing;
            final Collection processors = pipeline.getProcessors();
            processors.add(processor);
        } else {
            final Processor pipeline = Pipeline.newInstance(component.getCamelContext(), existing, processor);
            component.setBeforeProducer(pipeline);
        }
    }

    public static Processor pollEnricher(final String endpointUri, final ComponentProxyComponent proxyComponent) {
        final PollEnricher pollEnricher = new PollEnricher(new SimpleExpression(endpointUri), 0);
        final CamelContext camelContext = proxyComponent.getCamelContext();
        pollEnricher.setCamelContext(camelContext);
        pollEnricher.setAggregationStrategy(EnrichmentAggregator.INSTANCE);

        return Pipeline.newInstance(camelContext, proxyComponent.getBeforeConsumer(), pollEnricher, proxyComponent.getAfterConsumer());
    }
}