alvinalexander.com | career | drupal | java | mac | mysql | perl | scala | uml | unix  

What this is

This file is included in the DevDaily.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" TM.

Other links

The source code

/*
 * $Header: /cvsroot/mvnforum/mvnforum/src/com/mvnforum/search/member/MemberIndexer.java,v 1.8 2005/01/18 11:52:22 minhnn Exp $
 * $Author: minhnn $
 * $Revision: 1.8 $
 * $Date: 2005/01/18 11:52:22 $
 *
 * ====================================================================
 *
 * Copyright (C) 2002-2005 by MyVietnam.net
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or any later version.
 *
 * All copyright notices regarding mvnForum MUST remain intact
 * in the scripts and in the outputted HTML.
 * The "powered by" text/logo with a link back to
 * http://www.mvnForum.com and http://www.MyVietnam.net in the
 * footer of the pages MUST remain visible when the pages
 * are viewed on the internet or intranet.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 * Support can be obtained from support forums at:
 * http://www.mvnForum.com/mvnforum/index
 *
 * Correspondence and Marketing Questions can be sent to:
 * info@MyVietnam.net
 *
 * @author: Minh Nguyen  minhnn@MyVietnam.net
 * @author: Dejan Krsmanovic dejan_krsmanovic@yahoo.com
 */
package com.mvnforum.search.member;

import java.io.IOException;

import com.mvnforum.MVNForumConfig;
import com.mvnforum.MVNForumFactoryConfig;
import com.mvnforum.db.MemberBean;
import net.myvietnam.mvncore.exception.SearchException;
import net.myvietnam.mvncore.util.DateUtil;
import net.myvietnam.mvncore.util.TimerUtil;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.document.*;
import org.apache.lucene.index.*;
import org.apache.lucene.search.IndexSearcher;

public class MemberIndexer
{
    private static Log log = LogFactory.getLog(MemberIndexer.class);

    //Field names (used for indexing)
    public static final String FIELD_MEMBER_ID      = "MemberID";
    public static final String FIELD_MEMBER_NAME    = "MemberName";
    public static final String FIELD_CREATION_DATE  = "MemberCreationDate";
    public static final String FIELD_MEMBER_EMAIL   = "MemberEmail";
    public static final String FIELD_MEMBER_LASTNAME    = "MemberLastName";
    public static final String FIELD_MEMBER_FIRSTNAME   = "MemberFirstName";
    public static final String FIELD_MEMBER_COUNTRY = "MemberCountry";

    //public static final String PROPERTY_SEARCH_PATH      = "search.path";
    //public static final String PROPERTY_SEARCH_AUTOINDEX = "search.autoindex";

    //Timer is used for scheduling jobs
    private static Analyzer analyzer;
    private static String searchMemberIndexDir;

    private static long lastOptimizeTime = 0;

    static {
        searchMemberIndexDir = MVNForumConfig.getSearchMemberIndexDir();
        initializeAnalyzer();
    }

    public static void scheduleAddMemberTask(MemberBean MemberBean) {
        AddUpdateMemberIndexTask task = new AddUpdateMemberIndexTask(MemberBean, AddUpdateMemberIndexTask.OPERATION_ADD);
        TimerUtil.getInstance().schedule(task, 0);
    }

    public static void scheduleUpdateMemberTask(MemberBean MemberBean) {
        AddUpdateMemberIndexTask task = new AddUpdateMemberIndexTask(MemberBean, AddUpdateMemberIndexTask.OPERATION_UPDATE);
        TimerUtil.getInstance().schedule(task, 0);
    }

    public static void scheduleDeleteMemberTask(int objectID) {
        DeleteMemberIndexTask task = new DeleteMemberIndexTask(objectID);
        TimerUtil.getInstance().schedule(task, 0);
    }

    public static void scheduleRebuildIndexTask() {
        int maxMemberID = 0;
        RebuildMemberIndexTask task = new RebuildMemberIndexTask(maxMemberID);
        TimerUtil.getInstance().schedule(task, 0);
    }

    static Analyzer getAnalyzer() {
        return analyzer;
    }

    /**
     * This class will load analyzer when starting. If specified analyzer class
     * cannot be loaded then default analyzer will be used.
     */
    private static void initializeAnalyzer() {
        String analyzerClassName = MVNForumFactoryConfig.getLuceneAnalyzerClassName();
        if ( (analyzerClassName == null) || (analyzerClassName.equals("")) ) {
            //create standard analyzer
            //String[] stopWords = this.loadStopWords();
            analyzer = new StandardAnalyzer();
            log.debug("Using StandardAnalyzer for indexing");
        } else {
            //try to create specified analyzer
            try {
                log.debug("About to load Analyzer [" + analyzerClassName + "] for indexing");
                analyzer = (Analyzer) Class.forName(analyzerClassName).newInstance();
            } catch (Exception e) {
                log.warn("Cannot load " + analyzerClassName + ". Loading StandardAnalyzer");
                analyzer = new StandardAnalyzer();
            }
        }
    }

    /**
     * This method is used for getting new IndexWriter. It can create new index
     * or add Member to existing index. Creating new index will delete previous so it
     * should be used for rebuilding index.
     * @param create - true if new index should be created.
     *               - false for adding companies to existing index
     * @return IndexWriter object that is used for adding companies to index
     */
    static IndexWriter getIndexWriter(boolean create) throws SearchException {
        IndexWriter writer = null;

        //If create = false, we will create IndexWriter with false argument
        if (create == false) {
            try {
                writer = new IndexWriter(searchMemberIndexDir, analyzer, false);
                writer.setUseCompoundFile(true);
                return writer;
            } catch (IOException e) {
                log.warn("Cannot open existed index. New index will be created.", e);
                //Ignore Exception. We will try to create index with true parameter
            }
        }
        // We are here in two cases: We wanted to create new index or because
        // index doesn't existed
        try {
            //This will create new index and delete existing
            writer = new IndexWriter(searchMemberIndexDir, analyzer, true);
            writer.setUseCompoundFile(true);
            return writer;
        } catch (IOException e) {
            throw new SearchException("Error while creating index writer");
        }
    }

    /**
     * This method is used for adding single Member to index
     * Note: this method doesnt close the writer
     * @param MemberBean A Member that should be indexed
     * @param writer IndexWriter that is used for storing
     * @throws SearchException
     */
    static void doIndexMember(MemberBean memberBean, IndexWriter writer) throws SearchException {
        if (memberBean == null) return;
        if ( memberBean.getMemberName() == null || memberBean.getMemberName().equals("") ||
             (memberBean.getMemberCreationDate() == null)) {
            return;
        }

        //Each Member will be represented as a document
        Document memberDocument = new Document();
        //Document has following fields that could be queried on
        memberDocument.add(Field.Keyword(FIELD_MEMBER_ID, Integer.toString(memberBean.getMemberID())));
        memberDocument.add(Field.UnStored(FIELD_MEMBER_NAME, memberBean.getMemberName()));
        memberDocument.add(Field.UnStored(FIELD_MEMBER_EMAIL, memberBean.getMemberEmail()));
        memberDocument.add(Field.UnStored(FIELD_MEMBER_LASTNAME, memberBean.getMemberLastname()));
        memberDocument.add(Field.UnStored(FIELD_MEMBER_FIRSTNAME, memberBean.getMemberFirstname()));
        memberDocument.add(Field.UnStored(FIELD_MEMBER_COUNTRY, memberBean.getMemberCountry()));
        memberDocument.add(Field.Keyword(FIELD_CREATION_DATE, DateField.dateToString(memberBean.getMemberCreationDate())));//MemberBean.getMemberCreationDate())); // DateField.dateToString(MemberBean.getMemberCreationDate())));
        //now we have created document with fields so we can store it
        try {
            writer.addDocument(memberDocument);
        } catch (IOException e) {
            log.error("MemberIndexer.doIndexMember failed", e);
            throw new SearchException("Error writing new Member to index");
        }
    }

    /**
     * Add single Member to index
     * @param MemberBean
     * @throws SearchException
     */
    static void addMemberToIndex(MemberBean memberBean) throws SearchException, IOException {
        IndexWriter writer = null;
        try {
            writer = getIndexWriter(false);
            if (writer == null) {
                log.warn("Cannot get the IndexWriter");
                return;
            }
            doIndexMember(memberBean, writer);

            // now check if we should optimize index (each hour)
            long now = System.currentTimeMillis();
            long timeFromLastOptimize = now - lastOptimizeTime;
            if (timeFromLastOptimize > DateUtil.HOUR) {
                log.debug("writer.optimize() called in addMemberToIndex");
                writer.optimize();
                lastOptimizeTime = now;
            }
        } catch (SearchException ex) {
            throw ex;
        } finally {
            try {
                if (writer != null) {
                    writer.close();
                }
            } catch (IOException e) {
            }
        }
    }

    /**
     * This method is used for deleting Member from index.
     * @param MemberID is id of the Member that should be deleted
     * @throws SearchException
     */
    static void deleteMemberFromIndex(int memberID) throws SearchException {
        IndexReader reader = null;
        try {
            reader = IndexReader.open(searchMemberIndexDir);
            if (reader == null) {
                log.warn("Cannot get the IndexReader");
                return;
            }

            Term term = new Term(FIELD_MEMBER_ID, String.valueOf(memberID));
            int deletedCount = reader.delete(term);
            log.debug("deleteMemberFromIndex: deleted member = " + deletedCount);
        } catch (IOException e) {
            throw new SearchException("Error trying to delete Member with memberID = " + memberID);
        } finally {
            try {
                if (reader != null) {
                    reader.close();
                }
            } catch (IOException e) {
            }
        }
    }

    public static IndexSearcher getSearcher() throws IOException {
        try {
            IndexSearcher searcher = new IndexSearcher(searchMemberIndexDir);
            return searcher;
        } catch (IOException ex) {
            // we throw new IOException because the original exception
            // contain sensitive directory information
            log.error("Cannot access the lucene search index for query. Please check if you have configed mvnForumHome properly. You can also go to Admin Zone to rebuild the Lucene index files.", ex);
            //@todo : localize me
            throw new IOException("Cannot access the lucene search index. Please report this error to web site Administrator (check mvnForumHome or rebuild Lucene index).");
        }
    }

    public static int getNumDocs() {
        int numDocs = -1;
        IndexReader reader = null;
        try {
            reader = IndexReader.open(searchMemberIndexDir);
            if (reader == null) {
                log.warn("Cannot get the IndexReader");
                return -1;
            }
            numDocs = reader.numDocs();
        } catch ( IOException ioe) {
            //ignore
        } finally {
            try {
                if (reader != null) reader.close();
            } catch (IOException e) {
                log.debug("Error closing Lucene IndexReader", e);
            }
        }
        return numDocs;
    }

}
... this post is sponsored by my books ...

#1 New Release!

FP Best Seller

 

new blog posts

 

Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.

A percentage of advertising revenue from
pages under the /java/jwarehouse URI on this website is
paid back to open source projects.