com.collective.celos.JettyServer.java Source code

Introduction

Here is the source code for com.collective.celos.JettyServer.java

Source

/*
 * Copyright 2015 Collective, Inc.
 *
 * Licensed 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 com.collective.celos;

import java.net.URL;
import java.nio.file.Paths;
import java.util.Map;

import javax.servlet.ServletContext;

import org.apache.http.client.utils.URIBuilder;
import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.webapp.WebAppContext;

/**
 * Common Jetty code for Celos server and Celos UI.
 */
public class JettyServer {

    private Server server;
    private WebAppContext context;

    public void start(int port) throws Exception {
        server = new Server(port);
        startServer();
    }

    public int start() throws Exception {
        server = new Server();
        ServerConnector serverConnector = new ServerConnector(server);
        server.setConnectors(new Connector[] { serverConnector });

        startServer();

        return serverConnector.getLocalPort();
    }

    private void startServer() throws Exception {
        URL url = Thread.currentThread().getContextClassLoader().getResource("WEB-INF");
        URIBuilder uriBuilder = new URIBuilder(url.toURI());
        uriBuilder.setPath(Paths.get(url.getPath()).getParent().toString());

        context = new WebAppContext(uriBuilder.toString() + "/", "/");
        context.setInitParameter("org.eclipse.jetty.servlet.Default.dirAllowed", "false");

        context.setExtractWAR(false);

        server.setHandler(context);
        server.start();
    }

    public void stop() throws Exception {
        if (server == null) {
            throw new IllegalStateException("JettyServer was not started");
        }
        server.stop();
        server.destroy();
        server.join();
    }

    public void setupContext(Map<String, Object> attributes, Map<String, String> initParam) {
        for (Map.Entry<String, Object> entry : attributes.entrySet()) {
            context.setAttribute(entry.getKey(), entry.getValue());
        }
        for (Map.Entry<String, String> entry : initParam.entrySet()) {
            context.setInitParameter(entry.getKey(), entry.getValue());
        }
    }

    public ServletContext getContext() {
        return context.getServletHandler().getServletContext();
    }

}