io.atlasmap.examples.camel.main.Application Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2017 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.atlasmap.examples.camel.main;

import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;

import org.apache.camel.Exchange;
import org.apache.camel.Message;
import org.apache.camel.Processor;
import org.apache.camel.builder.RouteBuilder;
import org.apache.camel.impl.DefaultMessage;
import org.apache.camel.main.Main;

public class Application extends RouteBuilder {

    private static final String SOURCE_MAP_NAME = "atlasSourceMap";

    @Override
    public void configure() throws Exception {
        from("timer:main?period=5000")
            .to("direct:order-producer")
            .to("direct:contact-producer")
            .to(String.format("atlas:atlasmap-mapping.adm?sourceMapName=%s", SOURCE_MAP_NAME))
            .to("direct:outcome-consumer");

        from("direct:order-producer")
            .process(new MessageCaptureProcessor().setSource("order-schema", "data/order.json"))
            .log("--> Order: [${body}]");
        
        from("direct:contact-producer")
            .process(new MessageCaptureProcessor().setSource("contact-schema", "data/contact.xml"))
            .log("--> Contact: [${body}]");
        
        from("direct:outcome-consumer")
            .log("--< Outcome: body:[")
            .log("${body}")
            .log("], Headers:[")
            .log("Order Document ID:${header.order-id}, Order path:${header.order-path}")
            .log("Contact Document ID:${header.contact-id}, Contact path:${header.contact-path}")
            .log("]");
    }

    public static void main(String args[]) throws Exception {
        Main camelMain = new Main();
        camelMain.addRouteBuilder(new Application());
        camelMain.run(args);
    }

    public class MessageCaptureProcessor implements Processor {
        private String id;
        private String path;

        public MessageCaptureProcessor setSource(String id, String path) {
            this.id = id;
            this.path = path;
            return this;
        }

        public void process(Exchange ex) throws Exception {
            Message msg = new DefaultMessage(ex.getContext());
            msg.setHeader("id", id);
            msg.setHeader("path", path);
            URL url = ex.getContext().getClassResolver().loadResourceAsURL(path);
            String doc = new String(Files.readAllBytes(Paths.get(url.toURI())));
            msg.setBody(doc);

            Map sourceMap = (Map) ex.getProperty(SOURCE_MAP_NAME);
            if (sourceMap == null) {
                sourceMap = new HashMap();
                ex.setProperty(SOURCE_MAP_NAME, sourceMap);
            }
            sourceMap.put(id, msg);
            ex.setMessage(msg);
        }
    }

}