org.simplericity.jettyconsole.DefaultJettyManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jetty-console-core Show documentation
There is a newer version: 1.61
package org.simplericity.jettyconsole;

import org.eclipse.jetty.io.EndPoint;
import org.eclipse.jetty.server.Handler;
import org.eclipse.jetty.server.Request;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.ContextHandlerCollection;
import org.eclipse.jetty.server.handler.HandlerCollection;
import org.eclipse.jetty.server.nio.SelectChannelConnector;
import org.eclipse.jetty.webapp.WebAppContext;
import org.kantega.jexmec.PluginManager;
import org.simplericity.jettyconsole.api.Configuration;
import org.simplericity.jettyconsole.api.JettyConsolePlugin;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * Created by IntelliJ IDEA.
 * User: bjorsnos
 * Date: Nov 20, 2007
 * Time: 8:46:11 PM
 * To change this template use File | Settings | File Templates.
 */
public class DefaultJettyManager implements JettyManager {

    private Server server;
    private Logger log = LoggerFactory.getLogger(getClass());
    private Properties settings;
    private String name;
    private Runnable shutdownHook;
    private List listenerList = new ArrayList();
    private PluginManager pluginManager;
    private WebAppContext webapp;

    public DefaultJettyManager(Properties settings, PluginManager pluginManager) {
        this.settings = settings;
        this.pluginManager = pluginManager;
        this.name = DefaultJettyManager.this.settings.getProperty("name");

        shutdownHook = new Runnable() {
            public void run() {
                shutdown();
            }
        };
    }


    public void shutdown() {
        log.info("Shutting down " + name + "..");
        try {
            if (server != null && (server.isStarted() || server.isStarting())) {
                server.stop();
                log.info("Shutdown of " + name + " complete.");
                for (JettyListener listener : listenerList) {
                    listener.serverStopped();
                }
            }
        } catch (Exception e) {
            log.info("Exception shutting down " + name + ": " + e.getMessage(), e);
        }
    }

    public void startServer(Configuration configuration) {
        server = new Server();



        SelectChannelConnector connector = new SelectChannelConnector() {

            public void customize(EndPoint endpoint, Request request) throws IOException {
                super.customize(endpoint, request);
                for (JettyConsolePlugin plugin : pluginManager.getPlugins()) {
                    plugin.customizeRequest(endpoint, request);
                }
            }
        };

        for (JettyConsolePlugin plugin : pluginManager.getPlugins()) {
            plugin.customizeConnector(connector);
        }

        server.addConnector(connector);

        HandlerCollection handlers = new HandlerCollection();

        ContextHandlerCollection contexts = new ContextHandlerCollection();

        handlers.setHandlers(new Handler[] {contexts});

        server.setHandler(handlers);

        for (JettyConsolePlugin plugin : pluginManager.getPlugins()) {
            plugin.customizeServer(server);
        }

        File war = getWarLocation();

        webapp = new WebAppContext(contexts, war.getAbsolutePath(), configuration.getContextPath());

        for (JettyConsolePlugin plugin : pluginManager.getPlugins()) {
            plugin.beforeStart(webapp);
        }
        webapp.setAttribute("shutdownhook", shutdownHook);
        
        log.info("Added web application on path " + configuration.getContextPath() + " from war " + war.getAbsolutePath());

        log.info("Starting web application on port " + connector.getPort());

        final ClassLoader loader = Thread.currentThread().getContextClassLoader();

        try {
            Thread.currentThread().setContextClassLoader(JettyConsoleStarter.class.getClassLoader());

            try {
                server.start();
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        } finally {
            Thread.currentThread().setContextClassLoader(loader);
        }


    }

    public void stopServer() {
        if(webapp != null) {
            for (JettyConsolePlugin plugin : pluginManager.getPlugins()) {
                plugin.beforeStop(webapp);
            }
        }
        try {
            if (server != null && (server.isStarted() || server.isStarting())) {
                server.stop();
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public File getWarLocation() {
        URL resource = getClass().getResource("/WEB-INF/web.xml");
        String file = resource.getFile();
        file = file.substring("file:".length(), file.indexOf("!"));
        try {
            file = URLDecoder.decode(file, "utf-8");
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException(e);
        }
        return new File(file);
    }

    public void addListener(JettyListener listener) {
        listenerList.add(listener);
    }
}