org.ow2.bonita.util.IndexTool.java Source code

Introduction

Here is the source code for org.ow2.bonita.util.IndexTool.java

Source

/**
 * Copyright (C) 2010  BonitaSoft S.A.
 * BonitaSoft, 31 rue Gustave Eiffel - 38000 Grenoble
 * This library is free software; you can redistribute it and/or modify it under the terms
 * of the GNU Lesser General Public License as published by the Free Software Foundation
 * version 2.1 of the License.
 * This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License along with this
 * program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth
 * Floor, Boston, MA  02110-1301, USA.
 **/
package org.ow2.bonita.util;

import org.hibernate.CacheMode;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.engine.SessionFactoryImplementor;
import org.hibernate.search.FullTextSession;
import org.hibernate.search.MassIndexer;
import org.hibernate.search.Search;
import org.ow2.bonita.env.EnvConstants;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 
 * @author Matthieu Chaffotte, Elias Ricken de Medeiros
 *
 */
public final class IndexTool {

    private static final Logger LOG = LoggerFactory.getLogger(IndexTool.class);
    private static final String REINDEX_DOMAIN_PROPERTY = "org.ow2.bonita.reindex.domain";

    public static void main(final String[] args) throws Exception {
        if (args == null || args.length != 1) {
            final String message = ExceptionManager.getInstance().getFullMessage("bh_DBM_1");
            throw new IllegalArgumentException(message);
        }
        final String domain = args[0];

        LOG.info("Indexing History DB");
        populateIndexes(domain, EnvConstants.HB_CONFIG_HISTORY);
        LOG.info("Indexing Core DB");
        populateIndexes(domain, EnvConstants.HB_CONFIG_CORE);

    }

    private static void populateIndexes(SessionFactoryImplementor sessionFactory) throws InterruptedException {
        FullTextSession fullTextSession = null;
        try {
            Session session = sessionFactory.openSession();
            fullTextSession = Search.getFullTextSession(session);
            Transaction tx = fullTextSession.beginTransaction();
            MassIndexer indexer = fullTextSession.createIndexer();
            indexer.cacheMode(CacheMode.IGNORE);
            indexer.startAndWait();
            fullTextSession.createIndexer();
            tx.commit();
        } finally {
            if (fullTextSession != null) {
                fullTextSession.close();
            }
        }
    }

    public static void populateIndexes(String domain, String configurationName) throws Exception {
        System.setProperty(REINDEX_DOMAIN_PROPERTY, domain);
        SessionFactoryImplementor sessionFactory = null;
        try {
            BonitaConstants.getBonitaHomeFolder();
            Configuration cfg = DbTool.getConfiguration(domain, configurationName);
            if ("true".equals(cfg.getProperty("bonita.search.use"))) {
                sessionFactory = DbTool.getSessionFactory(domain,
                        configurationName.replaceAll("-configuration", "-session-factory"));
                if (sessionFactory != null) {
                    populateIndexes(sessionFactory);
                }
            } else {
                LOG.info(configurationName + "does not support indexing");
            }
        } finally {
            System.clearProperty(REINDEX_DOMAIN_PROPERTY);
            closeSession(sessionFactory);

        }
    }

    private static void closeSession(final SessionFactoryImplementor sessionFactory) {
        if (sessionFactory != null) {
            try {
                sessionFactory.close();
            } catch (Exception e) {
                if (LOG.isErrorEnabled()) {
                    LOG.error("Problems while closing section factory: ", e);
                }
            }
        }
    }

}