com.github.rnewson.couchdb.lucene.Main.java Source code

Introduction

Here is the source code for com.github.rnewson.couchdb.lucene.Main.java

Source

package com.github.rnewson.couchdb.lucene;

/**
 * Copyright 2009 Robert Newson
 * 
 * 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.
 */

import java.io.File;

import org.apache.commons.configuration.HierarchicalINIConfiguration;
import org.apache.commons.configuration.reloading.FileChangedReloadingStrategy;
import org.apache.http.client.HttpClient;
import org.apache.log4j.Logger;
import org.mortbay.jetty.Connector;
import org.mortbay.jetty.Handler;
import org.mortbay.jetty.Server;
import org.mortbay.jetty.nio.SelectChannelConnector;
import org.mortbay.jetty.servlet.Context;
import org.mortbay.jetty.servlet.FilterHolder;
import org.mortbay.jetty.servlet.ServletHolder;
import org.mortbay.servlet.GzipFilter;

public class Main {

    private static final Logger LOG = Logger.getLogger(Main.class);

    /**
     * Run couchdb-lucene.
     */
    public static void main(String[] args) throws Exception {
        final HierarchicalINIConfiguration configuration = new HierarchicalINIConfiguration(
                Main.class.getClassLoader().getResource("couchdb-lucene.ini"));
        configuration.setReloadingStrategy(new FileChangedReloadingStrategy());

        final File dir = new File(configuration.getString("lucene.dir", "indexes"));

        if (dir == null) {
            LOG.error("lucene.dir not set.");
            System.exit(1);
        }
        if (!dir.exists() && !dir.mkdir()) {
            LOG.error("Could not create " + dir.getCanonicalPath());
            System.exit(1);
        }
        if (!dir.canRead()) {
            LOG.error(dir + " is not readable.");
            System.exit(1);
        }
        if (!dir.canWrite()) {
            LOG.error(dir + " is not writable.");
            System.exit(1);
        }
        LOG.info("Index output goes to: " + dir.getCanonicalPath());

        final Server server = new Server();
        final SelectChannelConnector connector = new SelectChannelConnector();
        connector.setHost(configuration.getString("lucene.host", "localhost"));
        connector.setPort(configuration.getInt("lucene.port", 5985));

        LOG.info("Accepting connections with " + connector);

        server.setConnectors(new Connector[] { connector });
        server.setStopAtShutdown(true);
        server.setSendServerVersion(false);

        HttpClientFactory.setIni(configuration);
        final HttpClient httpClient = HttpClientFactory.getInstance();

        final LuceneServlet servlet = new LuceneServlet(httpClient, dir, configuration);

        final Context context = new Context(server, "/", Context.NO_SESSIONS | Context.NO_SECURITY);
        context.addServlet(new ServletHolder(servlet), "/*");
        context.addFilter(new FilterHolder(new GzipFilter()), "/*", Handler.DEFAULT);
        context.setErrorHandler(new JSONErrorHandler());
        server.setHandler(context);

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

}