001 /** 002 * 003 * Licensed to the Apache Software Foundation (ASF) under one or more 004 * contributor license agreements. See the NOTICE file distributed with 005 * this work for additional information regarding copyright ownership. 006 * The ASF licenses this file to You under the Apache License, Version 2.0 007 * (the "License"); you may not use this file except in compliance with 008 * the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018 package org.apache.camel.view; 019 020 import org.apache.camel.CamelContext; 021 import org.apache.camel.Endpoint; 022 import org.apache.camel.Processor; 023 import org.apache.camel.Route; 024 import org.apache.camel.impl.EventDrivenConsumerRoute; 025 import org.apache.commons.logging.Log; 026 import org.apache.commons.logging.LogFactory; 027 028 import java.io.FileWriter; 029 import java.io.IOException; 030 import java.io.PrintWriter; 031 import java.util.List; 032 033 /** 034 * A <a href="http://www.graphviz.org/">DOT</a> file creator plugin which 035 * creates a DOT file showing the current routes 036 * 037 * @version $Revision: 523881 $ 038 */ 039 public class RouteDotGenerator { 040 private static final transient Log log = LogFactory.getLog(RouteDotGenerator.class); 041 private String file = "CamelRoutes.dot"; 042 043 public String getFile() { 044 return file; 045 } 046 047 /** 048 * Sets the destination file name to create the destination diagram 049 */ 050 public void setFile(String file) { 051 this.file = file; 052 } 053 054 public void drawRoutes(CamelContext context) throws IOException { 055 PrintWriter writer = new PrintWriter(new FileWriter(file)); 056 generateFile(writer, context); 057 } 058 059 protected void generateFile(PrintWriter writer, CamelContext context) { 060 writer.println("digraph \"Camel Routes\" {"); 061 writer.println(); 062 writer.println("label=\"Camel Container: " + context + "\"];"); 063 writer.println(); 064 writer.println("node [style = \"rounded,filled\", fillcolor = yellow, fontname=\"Helvetica-Oblique\"];"); 065 writer.println(); 066 printRoutes(writer, context.getRoutes()); 067 } 068 069 protected void printRoutes(PrintWriter writer, List<Route> routes) { 070 for (Route r : routes) { 071 Endpoint end = r.getEndpoint(); 072 writer.print(end.getEndpointUri()); 073 writer.print(" -> "); 074 writer.print(r); 075 writer.print(" -> "); 076 if (r instanceof EventDrivenConsumerRoute) { 077 EventDrivenConsumerRoute consumerRoute = (EventDrivenConsumerRoute) r; 078 Processor p = consumerRoute.getProcessor(); 079 writer.println(p); 080 } 081 } 082 } 083 }