org.apache.nutch.webui.NutchUiServer.java Source code

Introduction

Here is the source code for org.apache.nutch.webui.NutchUiServer.java

Source

/**
 * 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.nutch.webui;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.GnuParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.hadoop.util.StringUtils;
import org.apache.wicket.protocol.http.WicketFilter;
import org.apache.wicket.spring.SpringWebApplicationFactory;
import org.mortbay.jetty.Handler;
import org.mortbay.jetty.Server;
import org.mortbay.jetty.servlet.Context;
import org.mortbay.jetty.servlet.DefaultServlet;
import org.mortbay.jetty.servlet.FilterHolder;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.request.RequestContextListener;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;

public class NutchUiServer {
    private static final String APP_FACTORY_NAME = SpringWebApplicationFactory.class.getName();
    private static final String CONFIG_LOCATION = "org.apache.nutch.webui";
    private static final String CMD_PORT = "port";
    private static Integer port = 8080;

    public static void main(String[] args) throws Exception {
        CommandLineParser parser = new GnuParser();
        Options options = createWebAppOptions();
        CommandLine commandLine = null;
        HelpFormatter formatter = new HelpFormatter();
        try {
            commandLine = parser.parse(options, args);
        } catch (Exception e) {
            formatter.printHelp("NutchUiServer", options, true);
            StringUtils.stringifyException(e);
        }

        if (commandLine.hasOption("help")) {
            formatter.printHelp("NutchUiServer", options, true);
            return;
        }
        if (commandLine.hasOption(CMD_PORT)) {
            port = Integer.parseInt(commandLine.getOptionValue(CMD_PORT));
        }
        startServer();
    }

    private static void startServer() throws Exception, InterruptedException {
        Server server = new Server(port);
        Context context = new Context(server, "/", Context.SESSIONS);
        context.addServlet(DefaultServlet.class, "/*");

        context.addEventListener(new ContextLoaderListener(getContext()));
        context.addEventListener(new RequestContextListener());

        WicketFilter filter = new WicketFilter();
        filter.setFilterPath("/");
        FilterHolder holder = new FilterHolder(filter);
        holder.setInitParameter("applicationFactoryClassName", APP_FACTORY_NAME);
        context.addFilter(holder, "/*", Handler.DEFAULT);

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

    private static WebApplicationContext getContext() {
        AnnotationConfigWebApplicationContext context = new AnnotationConfigWebApplicationContext();
        context.setConfigLocation(CONFIG_LOCATION);
        return context;
    }

    private static Options createWebAppOptions() {
        Options options = new Options();
        Option helpOpt = new Option("h", "help", false, "show this help message");
        OptionBuilder.withDescription("Port to run the WebApplication on.");
        OptionBuilder.hasOptionalArg();
        OptionBuilder.withArgName("port number");
        options.addOption(OptionBuilder.create(CMD_PORT));
        options.addOption(helpOpt);
        return options;
    }

}