JettyJWebContainerServiceImpl.java

package org.objectweb.jonas.web.jetty50;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.Map;
import java.util.StringTokenizer;
import java.util.Vector;

import javax.management.ObjectName;
import javax.naming.Context;
import javax.naming.NamingException;

import org.objectweb.util.monolog.api.BasicLevel;

import org.objectweb.jonas.jmx.JmxService;
import org.objectweb.jonas.jmx.JonasObjectName;
import org.objectweb.jonas.service.ServiceException;
import org.objectweb.jonas.service.ServiceManager;
import org.objectweb.jonas.web.AbsJWebContainerServiceImpl;
import org.objectweb.jonas.web.JWebContainerServiceException;

import org.mortbay.http.HttpContext;
import org.mortbay.http.HttpListener;
import org.mortbay.jetty.Server;
import org.mortbay.jetty.servlet.WebApplicationContext;

public class JettyJWebContainerServiceImpl extends AbsJWebContainerServiceImpl {

    private static final String JETTY_DEFAULT_WEB_XML_FILE = AbsJWebContainerServiceImpl.JONAS_BASE + File.separator + "conf" + File.separator + "jetty5-webdefault.xml";

    private static String config = null;

    private static final String JETTY_CONFIG = "config";

    private Server jettyServer = null;

    protected void doInit(javax.naming.Context ctx) throws ServiceException {

        super.doInit(ctx);

        String strJettyHome = System.getProperty("jetty.home");
        if (strJettyHome != null) {
            getLogger().log(BasicLevel.DEBUG, "");

            try {
                config = (String) ctx.lookup(JETTY_CONFIG);
            } catch (NamingException neExc) {
                // by default we take jetty5.xml under JONAS_BASE/conf
                String jonasbase = System.getProperties().getProperty("jonas.base");
                config = jonasbase + "/conf/jetty5.xml";
            }
            getLogger().log(BasicLevel.LEVEL_DEBUG, "using configuration file " + config);

            jettyServer = new Server();
        }

    }

    public void doStart() throws ServiceException {
        getLogger().log(BasicLevel.DEBUG, "");

        if (jettyServer != null) {
            JmxService srvcJMX = null;

            // Start Jetty directly and just pass it to JMX if available
            if (config != null) {
                try {
                    jettyServer.configure(config);
                    jettyServer.start();
                } catch (Exception eExc) {
                    getLogger().log(BasicLevel.LEVEL_ERROR,
                            "error has occured while starting Jetty server using configuration file " + config, eExc);
                }
            }

            ServiceManager sm = null;
            try {
                sm = ServiceManager.getInstance();
            } catch (Exception e) {
                String err = "Cannot get ServiceManager instance.";
                getLogger().log(BasicLevel.ERROR, err);
                throw new ServiceException(err, e);
            }

            try {
                srvcJMX = (JmxService) sm.getJmxService();
                if (srvcJMX != null) {
                    // Register JettyService MBean : JettyService
                    Object obj;
                    ObjectName objname;
                    // Once we register this bean, it will start Jetty using
                    // given configuration
                    // file. That's why we have to do it in run() instead in
                    // doStart()
                    obj = new JettyJonasServerMBean(jettyServer);
                    objname = JonasObjectName.wwwService();
                    srvcJMX.getJmxServer().registerMBean(obj, objname);
                }
            } catch (ServiceException seExc) {
                // Jmx Service not available, do nothing
                getLogger().log(
                        BasicLevel.LEVEL_DEBUG,
                        "cannot start Jetty server using configuration file " + config
                                + " using JMX. Will start without JMX.", seExc);
            } catch (Exception eExc) {
                getLogger().log(BasicLevel.LEVEL_ERROR, "cannot start Jetty server using configuration file " + config,
                        eExc);
                throw new ServiceException("Cannot start Jetty server using configuration file " + config, eExc);
            }

            // ... and deploy wars of the jonas.properties
            super.doStart();
        } else {
            throw new ServiceException("Cannot start Jetty server.");
        }
    }

    protected void doStop() throws ServiceException {
        // Undeploy the wars ...
        super.doStop();

        // ... and shut down embedded jetty
        getLogger().log(BasicLevel.DEBUG, "");
        if (isStarted()) {
            JmxService srvcJMX = null;

            ServiceManager sm = null;
            try {
                sm = ServiceManager.getInstance();
            } catch (Exception e) {
                String err = "Cannot get ServiceManager instance.";
                getLogger().log(BasicLevel.ERROR, err);
                throw new ServiceException(err, e);
            }

            try {
                srvcJMX = (JmxService) sm.getJmxService();
                if (srvcJMX != null) {
                    ObjectName objname;

                    objname = new ObjectName("jonas:type=service,name=jetty");
                    srvcJMX.getJmxServer().unregisterMBean(objname);
                    jettyServer = null;
                    objname = new ObjectName("jonas:type=jetty,name=jettylog");
                    srvcJMX.getJmxServer().unregisterMBean(objname);
                    objname = new ObjectName("jonas:type=jetty,name=jettycode");
                    srvcJMX.getJmxServer().unregisterMBean(objname);
                }
            } catch (ServiceException seExc) {
                getLogger().log(BasicLevel.LEVEL_ERROR, "JMX Service not available", seExc);
            } catch (Exception eExc) {
                getLogger().log(BasicLevel.LEVEL_ERROR, "Cannot stop Jetty server", eExc);
                throw new ServiceException("Cannot stop Jetty server", eExc);
            }

            if (jettyServer != null) {
                try {
                    jettyServer.stop();
                    jettyServer.destroy();
                    jettyServer = null;
                } catch (Exception eExc) {
                    getLogger().log(BasicLevel.LEVEL_ERROR,
                            "error has occured while stopping Jetty server using configuration file " + config, eExc);
                }
            }
        }
    }

    protected void doRegisterWar(Context ctx) throws JWebContainerServiceException {
        // Get the 5 parameters :
        //   - warURL is the URL of the war to register (required param).
        //   - contextRoot is the context root to which this application
        // should be installed (must be unique) (required param).
        //   - hostName is the name of the host on which deploy the war
        // (optional param taken into account only if no <context> element
        //  was declared in server.xml for this web application) .
        //   - java2DelegationModel the compliance to java2 delegation model
        //   - parentCL the war classloader of this war.
        //URL warURL = null;
        URL unpackedWarURL = null;
        String contextRoot = null;
        boolean java2DelegationModel = true;
        try {
            unpackedWarURL = (URL) ctx.lookup("unpackedWarURL");
            contextRoot = (String) ctx.lookup("contextRoot");
            Boolean bool = (Boolean) ctx.lookup("java2DelegationModel");
            java2DelegationModel = bool.booleanValue();
        } catch (NamingException e) {
            String err = "Error while getting parameter from context param ";
            getLogger().log(BasicLevel.ERROR, err + e.getMessage());
            throw new JWebContainerServiceException(err, e);
        }

        ClassLoader webClassLoader = null;
        try {
            webClassLoader = (ClassLoader) ctx.lookup("parentCL");
        } catch (NamingException e) {
            String err = "error while getting parameter from context param ";
            getLogger().log(BasicLevel.ERROR, err + e.getMessage());
            throw new JWebContainerServiceException(err, e);
        }

        String hostName = null;
        try {
            hostName = (String) ctx.lookup("hostName");
        } catch (NamingException e) {
            hostName = "";
        }

        // Install a new web application, whose web application archive is
        // at the specified URL, into this container with the specified
        // context root.
        // A context root of "" (the empty string) should be used for the root
        // application for this container. Otherwise, the context root must
        // start with a slash.

        if (contextRoot.equals("/")) {
            contextRoot = "";
        } else if (contextRoot.equalsIgnoreCase("ROOT")) {
            // Jetty uses ROOT.war and ROOT directory to as root context
            contextRoot = "";
        }

        // install the war.
        File fWar = new File(unpackedWarURL.getFile());
        String fileName = fWar.getAbsolutePath();

        if (jettyServer != null) {
            try {
                WebApplicationContext contextWebApp;

                if ((hostName == null) || (hostName.length() == 0)) {
                    // There is no host
                    getLogger().log(BasicLevel.DEBUG,
                            "Jetty server installing web app " + fileName + " and context " + contextRoot);
                    contextWebApp = jettyServer.addWebApplication(contextRoot, fileName);
                } else {
                    // Host was specified
                    getLogger().log(
                            BasicLevel.DEBUG,
                            "Jetty server installing web app " + fileName + " on host " + hostName + " and context "
                                    + contextRoot);
                    contextWebApp = jettyServer.addWebApplication(hostName, contextRoot, fileName);
                }

                // Add default xml descriptor
                if (new File(JETTY_DEFAULT_WEB_XML_FILE).exists()) {
                    contextWebApp.setDefaultsDescriptor(JETTY_DEFAULT_WEB_XML_FILE);
                } else {
                    getLogger().log(BasicLevel.WARN, "The file '" + JETTY_DEFAULT_WEB_XML_FILE + "' is not present. Check that your JONAS_BASE is up-to-date.");
                }

                // Specifying the jsp class path used by jasper
                contextWebApp.setAttribute("org.apache.catalina.jsp_classpath", getJOnASClassPath(webClassLoader));

                // Set the parent class loader
                contextWebApp.setParentClassLoader(webClassLoader);

                // Set this classloader to the Java2 compliant mode ?
                contextWebApp.setClassLoaderJava2Compliant(java2DelegationModel);

                getLogger().log(BasicLevel.DEBUG,
                        "Webapp class loader java 2 delegation model set to " + java2DelegationModel);

                getLogger().log(BasicLevel.DEBUG, "Jetty server starting web app " + fileName);
                contextWebApp.start();
                getLogger().log(BasicLevel.DEBUG, "Jetty server is running web app " + fileName);

            } catch (IOException ioeExc) {
                String err = "Cannot install this web application " + ioeExc;
                getLogger().log(BasicLevel.ERROR, err);
                throw new JWebContainerServiceException(err, ioeExc);
            } catch (Exception eExc) {
                String err = "Cannot start this web application " + eExc;
                getLogger().log(BasicLevel.ERROR, err);
                throw new JWebContainerServiceException(err, eExc);
            }
        } else {
            getLogger().log(BasicLevel.DEBUG, "No Jetty server to install web app " + fileName);
        }
    }

    public String getJOnASClassPath(ClassLoader webClassLoader) {

        StringBuffer classpath = new StringBuffer();
        int n = 0;
        while (webClassLoader != null) {
            if (!(webClassLoader instanceof URLClassLoader)) {
                break;
            }
            URL[] repositories = ((URLClassLoader) webClassLoader).getURLs();
            for (int i = 0; i < repositories.length; i++) {
                String repository = repositories[i].toString();
                if (repository.startsWith("file://")) {
                    repository = repository.substring("file://".length());
                } else if (repository.startsWith("file:")) {
                    repository = repository.substring("file:".length());
                } else {
                    continue;
                }
                if (repository == null) {
                    continue;
                }
                if (n > 0) {
                    classpath.append(File.pathSeparator);
                }
                classpath.append(repository);
                n++;
            }
            webClassLoader = webClassLoader.getParent();
        }

        return classpath.toString();
    }

    protected void doUnRegisterWar(Context ctx) throws JWebContainerServiceException {
        // Get the 2 parameters :
        //   - contextRoot is the context root to be removed (required param).
        //   - hostName is the name of the host to remove the war (optional).
        String contextRoot = null;
        try {
            contextRoot = (String) ctx.lookup("contextRoot");
        } catch (NamingException e) {
            String err = "Error while getting parameter from context param ";
            getLogger().log(BasicLevel.ERROR, err + e.getMessage());
            throw new JWebContainerServiceException(err, e);
        }

        String hostName = null;
        try {
            hostName = (String) ctx.lookup("hostName");
        } catch (NamingException e) {
            hostName = "";
        }
        // A context root of "" (the empty string) should be used for the root
        // application for this container. Otherwise, the context root must
        // start with a slash.

        if (contextRoot.equals("/")) {
            contextRoot = "";
        } else if (contextRoot.equalsIgnoreCase("ROOT")) {
            // Jetty uses ROOT.war and ROOT directory to as root context
            contextRoot = "";
        }

        if (jettyServer != null) {
            HttpContext contextWebApp;

            if ((hostName == null) || (hostName.length() == 0)) {
                // There is no host
                getLogger().log(BasicLevel.DEBUG, "Jetty server looking upweb app " + " from context " + contextRoot);
                contextWebApp = jettyServer.getContext(contextRoot);
            } else {
                // Host was specified
                getLogger().log(BasicLevel.DEBUG,
                        "Jetty server looking up web app " + " on host " + hostName + " and context " + contextRoot);
                contextWebApp = jettyServer.getContext(hostName, contextRoot);
            }

            if (contextWebApp != null) {
                getLogger().log(BasicLevel.DEBUG,
                        "Jetty server found and is stopping web app at context " + contextRoot);

                // Stop it gracefully
                try {
                    contextWebApp.stop(true);
                } catch (InterruptedException ieExc) {
                    getLogger().log(BasicLevel.LEVEL_DEBUG,
                            "Jetty server encoutered exception while stopping web application ", ieExc);
                }

                getLogger().log(BasicLevel.DEBUG,
                        "Jetty server stopped and is removing web app at context " + contextRoot);

                jettyServer.removeContext(contextWebApp);

                getLogger().log(BasicLevel.DEBUG,
                        "Jetty server removed and is destroying web app at context " + contextRoot);

                contextWebApp.destroy();

                getLogger().log(BasicLevel.DEBUG, "Jetty server unloaded web app at context " + contextRoot);

            } else {
                getLogger().log(BasicLevel.DEBUG, "Jetty server didn't find web app at context " + contextRoot);
            }

        } else {
            getLogger().log(BasicLevel.DEBUG, "No Jetty server to install web app at context " + contextRoot);
        }
    }

    protected void updateServerInfos() {
        String infos = org.mortbay.http.Version.getImplVersion();

        StringTokenizer st = new StringTokenizer(infos, "/");
        if (st.countTokens() != 2) {
            setServerName(infos);
            setServerVersion("");
        } else {
            setServerName(st.nextToken());
            setServerVersion(st.nextToken());
        }
    }

    public String getDefaultHost() throws JWebContainerServiceException {
        Map hosts = jettyServer.getHostMap();
        // If we have more than 1 host, we cannot determine default host!
        if (hosts.size() != 1) {
            String err = "Cannot determine default host :" + " Jetty server has more than 1 host";
            throw new JWebContainerServiceException(err);
        }
        String vHost = (String) hosts.keySet().iterator().next();
        if (vHost == null) {
            return "localhost";
        } else {
            return vHost;
        }
    }

    public String getDefaultHttpPort() throws JWebContainerServiceException {
        HttpListener[] listeners = jettyServer.getListeners();
        Vector http = new Vector();
        for (int i = 0; i < listeners.length; i++) {
            String scheme = listeners[i].getDefaultScheme();
            if (scheme.equalsIgnoreCase("http")) {
                http.add(listeners[i]);
            }
        }

        if (http.size() != 1) {
            String err = "Cannot determine default http port :" + " Jetty server has more (or zero) HTTP Listener";
            throw new JWebContainerServiceException(err);
        }

        return new String("" + ((HttpListener) http.get(0)).getPort());
    }

    public String getDefaultHttpsPort() throws JWebContainerServiceException {
        HttpListener[] listeners = jettyServer.getListeners();
        Vector http = new Vector();
        for (int i = 0; i < listeners.length; i++) {
            String scheme = listeners[i].getDefaultScheme();
            if (scheme.equalsIgnoreCase("https")) {
                http.add(listeners[i]);
            }
        }

        if (http.size() != 1) {
            String err = "Cannot determine default http port :" + " Jetty server has more (or zero) HTTPS Listener";
            throw new JWebContainerServiceException(err);
        }

        return new String("" + ((HttpListener) http.get(0)).getPort());
    }

}

---