net.jforum.controllers.LuceneAdminController.java Source code

Introduction

Here is the source code for net.jforum.controllers.LuceneAdminController.java

Source

/*
 * Copyright (c) JForum Team. All rights reserved.
 *
 * The software in this package is published under the terms of the LGPL
 * license a copy of which has been included with this distribution in the
 * license.txt file.
 *
 * The JForum Project
 * http://www.jforum.net
 */
package net.jforum.controllers;

import java.io.IOException;
import java.util.Date;

import net.jforum.actions.helpers.Domain;
import net.jforum.core.exceptions.ForumException;
import net.jforum.entities.Post;
import net.jforum.repository.ForumRepository;
import net.jforum.util.ConfigKeys;
import net.jforum.util.JForumConfig;

import org.apache.lucene.index.IndexReader;
import org.hibernate.CacheMode;
import org.hibernate.FlushMode;
import org.hibernate.ScrollMode;
import org.hibernate.ScrollableResults;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.search.FullTextSession;
import org.hibernate.search.Search;
import org.hibernate.search.SearchFactory;
import org.hibernate.search.reader.ReaderProvider;
import org.hibernate.search.store.DirectoryProvider;

import br.com.caelum.vraptor.Path;
import br.com.caelum.vraptor.Resource;
import br.com.caelum.vraptor.Result;

/**
 * @author Rafael Steil
 */
@Resource
@Path(Domain.ADMIN_SEARCH)
public class LuceneAdminController {
    private JForumConfig config;
    private ForumRepository forumRepository;
    private SessionFactory sessionFactory;
    private final Result result;

    public LuceneAdminController(JForumConfig config, ForumRepository forumRepository,
            SessionFactory sessionFactory, Result result) {
        this.config = config;
        this.forumRepository = forumRepository;
        this.sessionFactory = sessionFactory;
        this.result = result;
    }

    public void rebuildIndex() {

        Runnable indexingJob = new Runnable() {
            public void run() {
                Session session = null;

                try {
                    session = sessionFactory.openSession();

                    FullTextSession fullTextSession = Search.createFullTextSession(session);
                    fullTextSession.setFlushMode(FlushMode.MANUAL);
                    fullTextSession.setCacheMode(CacheMode.IGNORE);

                    session.beginTransaction();

                    int index = 0;
                    int batchSize = config.getInt(ConfigKeys.LUCENE_BATCH_SIZE);

                    ScrollableResults results = fullTextSession.createCriteria(Post.class).createAlias("topic", "t")
                            .scroll(ScrollMode.FORWARD_ONLY);

                    while (results.next() && "1".equals(config.getValue(ConfigKeys.LUCENE_CURRENTLY_INDEXING))) {
                        index++;

                        fullTextSession.index(results.get(0));

                        if (index % batchSize == 0) {
                            session.clear();
                        }
                    }

                    session.getTransaction().commit();
                } catch (Exception e) {
                    if (session.getTransaction().isActive()) {
                        session.getTransaction().rollback();
                    }
                } finally {
                    if (session.isOpen() && session.isConnected()) {
                        session.close();
                    }
                }
            }
        };

        this.config.addProperty(ConfigKeys.LUCENE_CURRENTLY_INDEXING, "1");

        Thread thread = new Thread(indexingJob);
        thread.start();

        this.result.redirectTo(this).list();
    }

    public void cancelIndexing() {
        this.config.addProperty(ConfigKeys.LUCENE_CURRENTLY_INDEXING, "0");
        this.result.redirectTo(this).list();
    }

    /**
     * Shows the main statistics page
     */
    public void list() {
        IndexReader indexReader = null;
        ReaderProvider readerProvider = null;

        try {
            SearchFactory searchFactory = Search.createFullTextSession(this.sessionFactory.getCurrentSession())
                    .getSearchFactory();

            DirectoryProvider<?> directoryProvider = searchFactory.getDirectoryProviders(Post.class)[0];
            readerProvider = searchFactory.getReaderProvider();
            indexReader = readerProvider.openReader(directoryProvider);

            String indexDirectory = directoryProvider.getDirectory().toString();
            indexDirectory = indexDirectory.substring(indexDirectory.indexOf('@') + 1);

            boolean indexExists = IndexReader.indexExists(indexDirectory);

            this.result.include("indexExists", indexExists);

            if (indexExists) {
                this.result.include("numberOfDocs", indexReader.numDocs());
                this.result.include("indexLocation", indexDirectory);
                this.result.include("totalMessages", this.forumRepository.getTotalMessages());
                this.result.include("isLocked", IndexReader.isLocked(indexDirectory));
                this.result.include("lastModified", new Date(IndexReader.lastModified(indexDirectory)));
            }
        } catch (IOException e) {
            throw new ForumException(e);
        } finally {
            if (readerProvider != null && indexReader != null) {
                readerProvider.closeReader(indexReader);
            }
        }
    }
}