com.bitranger.parknshop.common.fulltext.BuildIndexForCustomer.java Source code

Introduction

Here is the source code for com.bitranger.parknshop.common.fulltext.BuildIndexForCustomer.java

Source

/*******************************************************************************
 * Copyright (c) 2014 BitRangers (Team C1).
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *      BitRangers (Team C1) - initial API and implementation
 ******************************************************************************/
/**
 * 
 */
package com.bitranger.parknshop.common.fulltext;

import java.io.File;
import java.io.IOException;
import java.util.List;

import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.IndexWriterConfig;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.SimpleFSDirectory;
import org.apache.lucene.util.Version;

import com.bitranger.parknshop.buyer.dao.IPsCustomerDAO;
import com.bitranger.parknshop.buyer.model.PsCustomer;
import com.bitranger.parknshop.common.dao.FetchOption;

/**
 * @author GaoJin
 * 
 */
public class BuildIndexForCustomer {

    // use the IPsItemDAO to get the information.
    private IPsCustomerDAO customerDao;

    public IPsCustomerDAO getCustomerDao() {
        return customerDao;
    }

    public void setCustomerDao(IPsCustomerDAO customerDao) {
        this.customerDao = customerDao;
    }

    // the index will be stored in the given way
    private static File indexFile = new File("D:\\luceneIndex");

    public static File getIndexFile() {
        return indexFile;
    }

    public static void setIndexFile(File indexFile) {
        BuildIndexForCustomer.indexFile = indexFile;
    }

    // define a analyzer
    private static Analyzer analyzer = new StandardAnalyzer(Version.LUCENE_45);
    // get the customerlist
    private static List<PsCustomer> customerlist;

    // create the index
    @SuppressWarnings("deprecation")
    public void createIndex() throws IOException {
        // store in the directory
        Directory directory = new SimpleFSDirectory(indexFile);
        IndexWriterConfig indexWriterConfig = new IndexWriterConfig(Version.LUCENE_45, analyzer);
        indexWriterConfig.setOpenMode(IndexWriterConfig.OpenMode.CREATE);
        IndexWriter writer = new IndexWriter(directory, indexWriterConfig);// 
        FetchOption option = new FetchOption();
        customerlist = customerDao.findAll(option);
        if (customerlist.size() > 0) {
            long time1 = System.currentTimeMillis();
            for (int i = 0; i < customerlist.size(); i++) {
                Document document = new Document();
                document.add(new Field("nickname", customerlist.get(i).getNickname(), Field.Store.YES,
                        Field.Index.ANALYZED));
                document.add(
                        new Field("email", customerlist.get(i).getEmail(), Field.Store.YES, Field.Index.ANALYZED));
                document.add(new Field("password", customerlist.get(i).getPassword(), Field.Store.YES,
                        Field.Index.ANALYZED));
                document.add(new Field("gender", String.valueOf(customerlist.get(i).getGender()), Field.Store.YES,
                        Field.Index.ANALYZED));
                document.add(
                        new Field("name", customerlist.get(i).getName(), Field.Store.YES, Field.Index.ANALYZED));

                writer.addDocument(document);
            }
            long time2 = System.currentTimeMillis();
            System.out.println("" + writer.numDocs() + "");
            System.out.println("" + (time2 - time1) + "");
        }
        writer.close();
    }

}