org.odata4j.cxf.producer.server.CxfJettyServer Maven / Gradle / Ivy

Go to download
package org.odata4j.cxf.producer.server;

import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

import javax.ws.rs.core.Application;

import org.apache.cxf.jaxrs.servlet.CXFNonSpringJaxrsServlet;
import org.core4j.Enumerable;
import org.eclipse.jetty.server.Handler;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.HandlerCollection;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.odata4j.core.Throwables;
import org.odata4j.producer.server.ODataServer;

/**
 * OData server using the CXF JAX-RS implementation and Jetty as HTTP server.
 */
public class CxfJettyServer implements ODataServer {

  private String appBaseUri;
  private Class odataApp;
  private Class rootApp;
  private final List jettyRequestHandlers = new ArrayList();
  private Server server;

  public CxfJettyServer(String appBaseUri) {
    this.appBaseUri = appBaseUri;
  }

  public CxfJettyServer(String appBaseUri, Class odataApp, Class rootApp) {
    this.appBaseUri = appBaseUri;
    this.odataApp = odataApp;
    this.rootApp = rootApp;
  }

  @Override
  public ODataServer setODataApplication(Class odataApp) {
    this.odataApp = odataApp;
    return this;
  }

  @Override
  public ODataServer setRootApplication(Class rootApp) {
    this.rootApp = rootApp;
    return this;
  }

  public CxfJettyServer addJettyRequestHandler(Handler handler) {
    jettyRequestHandlers.add(handler);
    return this;
  }

  @Override
  public ODataServer start() {
    if (odataApp == null)
      throw new RuntimeException("ODataApplication not set");

    URL url;
    try {
      url = new URL(appBaseUri);
    } catch (MalformedURLException e) {
      throw Throwables.propagate(e);
    }

    CXFNonSpringJaxrsServlet odataServlet = new CXFNonSpringJaxrsServlet();
    ServletHolder odataServletHolder = new ServletHolder(odataServlet);
    odataServletHolder.setInitParameter("javax.ws.rs.Application", odataApp.getCanonicalName());

    ServletContextHandler contextHandler = new ServletContextHandler(ServletContextHandler.SESSIONS);
    contextHandler.addServlet(odataServletHolder, normalizePath(url.getPath()) + "/*");

    if (rootApp != null) {
      CXFNonSpringJaxrsServlet rootServlet = new CXFNonSpringJaxrsServlet();
      ServletHolder rootServletHolder = new ServletHolder(rootServlet);
      rootServletHolder.setInitParameter("javax.ws.rs.Application", rootApp.getCanonicalName());

      contextHandler.addServlet(rootServletHolder, "/*");
    }

    server = new Server(url.getPort());
    server.setHandler(getHandlerCollection(contextHandler));

    try {

      server.start();
      return this;
    } catch (Exception e) {
      throw Throwables.propagate(e);
    }
  }

  @Override
  public ODataServer stop() {
    try {
      server.stop();
      return this;
    } catch (Exception e) {
      throw Throwables.propagate(e);
    }
  }

  private static String normalizePath(String path) {
    if (path.endsWith("/"))
      return path.substring(0, path.length() - 1);
    return path;
  }

  private HandlerCollection getHandlerCollection(ServletContextHandler contextHandler) {
    List handlers = jettyRequestHandlers;
    handlers.add(contextHandler);

    HandlerCollection handlerCollection = new HandlerCollection();
    handlerCollection.setHandlers(Enumerable.create(handlers).toArray(Handler.class));
    return handlerCollection;
  }
}