trunk/components/camel-web/src/main/java/org/apache/camel/web/resources/RouteResource.java - camel - Git at Google

 /**
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You 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 org.apache.camel.web.resources;

 import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.io.StringReader;
 import java.io.StringWriter;
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.Collections;
 import java.util.List;

 import javax.ws.rs.Consumes;
 import javax.ws.rs.FormParam;
 import javax.ws.rs.GET;
 import javax.ws.rs.POST;
 import javax.ws.rs.Path;
 import javax.ws.rs.Produces;
 import javax.ws.rs.core.Context;
 import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.Response;
 import javax.ws.rs.core.UriInfo;
 import javax.xml.bind.JAXBContext;
 import javax.xml.bind.JAXBException;
 import javax.xml.bind.Marshaller;
 import javax.xml.bind.Unmarshaller;

 import com.sun.jersey.api.view.Viewable;

 import org.apache.camel.builder.RouteBuilder;
 import org.apache.camel.impl.DefaultCamelContext;
 import org.apache.camel.model.RouteDefinition;
 import org.apache.camel.util.ObjectHelper;
 import org.apache.camel.view.RouteDotGenerator;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;

 /**
  * A single Camel Route which is used to implement one or more
  * <a href="http://camel.apache.org/enterprise-integration-patterns.html">Enterprise Integration Patterns</a>
  *
  * @version
  */
 public class RouteResource extends CamelChildResourceSupport {
     public static final String LANGUAGE_XML = "Xml";
     private static final transient Logger LOG = LoggerFactory.getLogger(RouteResource.class);

     private RouteDefinition route;
     private String error = "";
     private String id;

     // what language is used to define this route
     private String language = LANGUAGE_XML;

     public RouteResource(RoutesResource routesResource, RouteDefinition route) {
         super(routesResource.getContextResource());
         this.route = route;
         this.id = route.idOrCreate(getCamelContext().getNodeIdFactory());
     }

     /**
      * Returns the XML or JSON representation of this route
      */
     @GET
     @Produces({MediaType.TEXT_XML, MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON})
     public RouteDefinition getRoute() {
         return route;
     }

     /**
      * Removes this route
      */
     @GET
     @Path("remove")
     @Produces(MediaType.TEXT_HTML)
     public Response removeRoute() {
         URI routesURI = null;
         try {
             routesURI = new URI("/routes");
             getCamelContext().removeRouteDefinition(route);
             return Response.seeOther(routesURI).build();
         } catch (Exception e) {
             LOG.error("failed to remove route " + id + ", error " + e.getMessage());
             return Response.seeOther(routesURI).build();
         }
     }

     /**
      * Returns the XML text
      */
     public String getRouteXml() throws JAXBException {
         JAXBContext context = JAXBContext.newInstance(Constants.JAXB_PACKAGES);
         Marshaller marshaller = context.createMarshaller();
         marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
         // TODO fix to use "" namespace prefix
         // using this
         // https://jaxb.dev.java.net/nonav/2.1.10/docs/vendorProperties.html#prefixmapper
         StringWriter buffer = new StringWriter();
         marshaller.marshal(route, buffer);
         return buffer.toString();
     }

     /**
      * Returns the content of the route definition class
      */
     public String getRouteDefinition() {
         if (language.equalsIgnoreCase(LANGUAGE_XML)) {
             try {
                 return getRouteXml();
             } catch (JAXBException e) {
                 return "Error on marshal the route definition!";
             }
         } else {
             return "Unsupported language!";
         }
     }

     /**
      * Returns the Graphviz DOT <a
      * href="http://camel.apache.org/visualisation.html">Visualisation</a> of
      * this route
      */
     @GET
     @Produces(Constants.DOT_MIMETYPE)
     public String getDot() throws IOException {
         RouteDotGenerator generator = new RouteDotGenerator("/tmp/camel");
         return generator.getRoutesText(getCamelContext());
     }

     /**
      * Allows a route definition to be updated
      */
     @POST
     @Consumes()
     @Produces({MediaType.TEXT_XML, MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON})
     public void postRoute(RouteDefinition routeDefinition) throws Exception {
         // lets preserve the ID
         routeDefinition.setId(id);

         // lets install the updated route
         getCamelContext().addRouteDefinitions(Collections.singletonList(routeDefinition));
     }

     /**
      * Allows a routes builder to be updated
      */
     public void postRoutes(RouteBuilder builder) throws Exception {
         DefaultCamelContext defaultCamelContext = (DefaultCamelContext)getCamelContext();
         // stop and remove the original route
         defaultCamelContext.stopRoute(id);
         defaultCamelContext.removeRoute(id);

         // add the routes in a route builder
         defaultCamelContext.addRoutes(builder);

         // reserve the id on the newest route
         List<RouteDefinition> routeDefinitions = defaultCamelContext.getRouteDefinitions();
         RouteDefinition route = routeDefinitions.get(routeDefinitions.size() - 1);
         route.setId(id);
         defaultCamelContext.startRoute(route);
     }

     /**
      * Updates a route definition using form encoded data from a web form
      *
      * @param language is the edited language used on this route
      * @param body is the route definition content POSTed typically from a HTML
      *            form with the <code>route</code> field
      * @param edited is a flag to show whether the route have been edited
      */
     @POST
     @Consumes("application/x-www-form-urlencoded")
     public Response postRouteForm(@Context UriInfo uriInfo, @FormParam("language") String language,
                                   @FormParam("route") String body, @FormParam("edited") String edited)
         throws URISyntaxException {

         if (edited.equals("false")) {
             return Response.seeOther(new URI("/routes")).build();
         }

         if (LOG.isDebugEnabled()) {
             LOG.debug("New Route is: " + body);
         }

         LOG.info(body);
         if (body == null) {
             error = "No Route submitted!";
         } else if (language.equals(LANGUAGE_XML)) {
             return parseXml(body);
         }
         error = "Not supproted language!";
         return Response.ok(new Viewable("edit", this)).build();
     }

     /**
      * process the route configuration defined in Xml
      */
     private Response parseXml(String xml) {
         try {
             JAXBContext context = JAXBContext.newInstance(Constants.JAXB_PACKAGES);
             Unmarshaller unmarshaller = context.createUnmarshaller();
             Object value = unmarshaller.unmarshal(new StringReader(xml));
             if (value instanceof RouteDefinition) {
                 RouteDefinition routeDefinition = (RouteDefinition)value;
                 postRoute(routeDefinition);
                 return Response.seeOther(new URI("/routes")).build();
             } else {
                 error = "Posted XML is not a route but is of type " + ObjectHelper.className(value);
             }
         } catch (JAXBException e) {
             error = "Failed to parse XML: " + e.getMessage();
         } catch (Exception e) {
             error = "Failed to install route: " + e.getMessage();
         }
         // lets re-render the form
         return Response.ok(new Viewable("edit", this)).build();
     }

     /**
      * Stores the route definition class into a file
      */
     private File storeRoute(String route, String language) throws IOException {
         // create a temporary file to store the route definition class
         File file = File.createTempFile("Route-", "." + language);
         FileWriter fw = new FileWriter(file);

         // write the route into the file
         fw.write(route);
         fw.flush();
         fw.close();
         return file;
     }

     /**
      * Returns the language
      */
     public String getLanguage() {
         return language;
     }

     public void setLanguage(String language) {
         this.language = LANGUAGE_XML;
     }

     /**
      * Looks up an individual route
      */
     @Path("status")
     public RouteStatusResource getRouteStatus() {
         return new RouteStatusResource(this);
     }

     public String getError() {
         return error;
     }

     public String getId() {
         return id;
     }
 }
	/**
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You 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 org.apache.camel.web.resources;

	import java.io.File;
	import java.io.FileWriter;
	import java.io.IOException;
	import java.io.StringReader;
	import java.io.StringWriter;
	import java.net.URI;
	import java.net.URISyntaxException;
	import java.util.Collections;
	import java.util.List;

	import javax.ws.rs.Consumes;
	import javax.ws.rs.FormParam;
	import javax.ws.rs.GET;
	import javax.ws.rs.POST;
	import javax.ws.rs.Path;
	import javax.ws.rs.Produces;
	import javax.ws.rs.core.Context;
	import javax.ws.rs.core.MediaType;
	import javax.ws.rs.core.Response;
	import javax.ws.rs.core.UriInfo;
	import javax.xml.bind.JAXBContext;
	import javax.xml.bind.JAXBException;
	import javax.xml.bind.Marshaller;
	import javax.xml.bind.Unmarshaller;

	import com.sun.jersey.api.view.Viewable;

	import org.apache.camel.builder.RouteBuilder;
	import org.apache.camel.impl.DefaultCamelContext;
	import org.apache.camel.model.RouteDefinition;
	import org.apache.camel.util.ObjectHelper;
	import org.apache.camel.view.RouteDotGenerator;
	import org.slf4j.Logger;
	import org.slf4j.LoggerFactory;

	/**
	* A single Camel Route which is used to implement one or more
	* <a href="http://camel.apache.org/enterprise-integration-patterns.html">Enterprise Integration Patterns</a>
	*
	* @version
	*/
	public class RouteResource extends CamelChildResourceSupport {
	public static final String LANGUAGE_XML = "Xml";
	private static final transient Logger LOG = LoggerFactory.getLogger(RouteResource.class);

	private RouteDefinition route;
	private String error = "";
	private String id;

	// what language is used to define this route
	private String language = LANGUAGE_XML;

	public RouteResource(RoutesResource routesResource, RouteDefinition route) {
	super(routesResource.getContextResource());
	this.route = route;
	this.id = route.idOrCreate(getCamelContext().getNodeIdFactory());
	}

	/**
	* Returns the XML or JSON representation of this route
	*/
	@GET
	@Produces({MediaType.TEXT_XML, MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON})
	public RouteDefinition getRoute() {
	return route;
	}

	/**
	* Removes this route
	*/
	@GET
	@Path("remove")
	@Produces(MediaType.TEXT_HTML)
	public Response removeRoute() {
	URI routesURI = null;
	try {
	routesURI = new URI("/routes");
	getCamelContext().removeRouteDefinition(route);
	return Response.seeOther(routesURI).build();
	} catch (Exception e) {
	LOG.error("failed to remove route " + id + ", error " + e.getMessage());
	return Response.seeOther(routesURI).build();
	}
	}

	/**
	* Returns the XML text
	*/
	public String getRouteXml() throws JAXBException {
	JAXBContext context = JAXBContext.newInstance(Constants.JAXB_PACKAGES);
	Marshaller marshaller = context.createMarshaller();
	marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
	// TODO fix to use "" namespace prefix
	// using this
	// https://jaxb.dev.java.net/nonav/2.1.10/docs/vendorProperties.html#prefixmapper
	StringWriter buffer = new StringWriter();
	marshaller.marshal(route, buffer);
	return buffer.toString();
	}

	/**
	* Returns the content of the route definition class
	*/
	public String getRouteDefinition() {
	if (language.equalsIgnoreCase(LANGUAGE_XML)) {
	try {
	return getRouteXml();
	} catch (JAXBException e) {
	return "Error on marshal the route definition!";
	}
	} else {
	return "Unsupported language!";
	}
	}

	/**
	* Returns the Graphviz DOT <a
	* href="http://camel.apache.org/visualisation.html">Visualisation</a> of
	* this route
	*/
	@GET
	@Produces(Constants.DOT_MIMETYPE)
	public String getDot() throws IOException {
	RouteDotGenerator generator = new RouteDotGenerator("/tmp/camel");
	return generator.getRoutesText(getCamelContext());
	}

	/**
	* Allows a route definition to be updated
	*/
	@POST
	@Consumes()
	@Produces({MediaType.TEXT_XML, MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON})
	public void postRoute(RouteDefinition routeDefinition) throws Exception {
	// lets preserve the ID
	routeDefinition.setId(id);

	// lets install the updated route
	getCamelContext().addRouteDefinitions(Collections.singletonList(routeDefinition));
	}

	/**
	* Allows a routes builder to be updated
	*/
	public void postRoutes(RouteBuilder builder) throws Exception {
	DefaultCamelContext defaultCamelContext = (DefaultCamelContext)getCamelContext();
	// stop and remove the original route
	defaultCamelContext.stopRoute(id);
	defaultCamelContext.removeRoute(id);

	// add the routes in a route builder
	defaultCamelContext.addRoutes(builder);

	// reserve the id on the newest route
	List<RouteDefinition> routeDefinitions = defaultCamelContext.getRouteDefinitions();
	RouteDefinition route = routeDefinitions.get(routeDefinitions.size() - 1);
	route.setId(id);
	defaultCamelContext.startRoute(route);
	}

	/**
	* Updates a route definition using form encoded data from a web form
	*
	* @param language is the edited language used on this route
	* @param body is the route definition content POSTed typically from a HTML
	* form with the <code>route</code> field
	* @param edited is a flag to show whether the route have been edited
	*/
	@POST
	@Consumes("application/x-www-form-urlencoded")
	public Response postRouteForm(@Context UriInfo uriInfo, @FormParam("language") String language,
	@FormParam("route") String body, @FormParam("edited") String edited)
	throws URISyntaxException {

	if (edited.equals("false")) {
	return Response.seeOther(new URI("/routes")).build();
	}

	if (LOG.isDebugEnabled()) {
	LOG.debug("New Route is: " + body);
	}

	LOG.info(body);
	if (body == null) {
	error = "No Route submitted!";
	} else if (language.equals(LANGUAGE_XML)) {
	return parseXml(body);
	}
	error = "Not supproted language!";
	return Response.ok(new Viewable("edit", this)).build();
	}

	/**
	* process the route configuration defined in Xml
	*/
	private Response parseXml(String xml) {
	try {
	JAXBContext context = JAXBContext.newInstance(Constants.JAXB_PACKAGES);
	Unmarshaller unmarshaller = context.createUnmarshaller();
	Object value = unmarshaller.unmarshal(new StringReader(xml));
	if (value instanceof RouteDefinition) {
	RouteDefinition routeDefinition = (RouteDefinition)value;
	postRoute(routeDefinition);
	return Response.seeOther(new URI("/routes")).build();
	} else {
	error = "Posted XML is not a route but is of type " + ObjectHelper.className(value);
	}
	} catch (JAXBException e) {
	error = "Failed to parse XML: " + e.getMessage();
	} catch (Exception e) {
	error = "Failed to install route: " + e.getMessage();
	}
	// lets re-render the form
	return Response.ok(new Viewable("edit", this)).build();
	}

	/**
	* Stores the route definition class into a file
	*/
	private File storeRoute(String route, String language) throws IOException {
	// create a temporary file to store the route definition class
	File file = File.createTempFile("Route-", "." + language);
	FileWriter fw = new FileWriter(file);

	// write the route into the file
	fw.write(route);
	fw.flush();
	fw.close();
	return file;
	}

	/**
	* Returns the language
	*/
	public String getLanguage() {
	return language;
	}

	public void setLanguage(String language) {
	this.language = LANGUAGE_XML;
	}

	/**
	* Looks up an individual route
	*/
	@Path("status")
	public RouteStatusResource getRouteStatus() {
	return new RouteStatusResource(this);
	}

	public String getError() {
	return error;
	}

	public String getId() {
	return id;
	}
	}