indexer.files.LuceneWriter.java Source code

Introduction

Here is the source code for indexer.files.LuceneWriter.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package indexer.files;

import java.io.File;
import java.io.IOException;
import indexer.data.Noticia;
import java.util.Arrays;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.es.SpanishAnalyzer;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.analysis.util.CharArraySet;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.IntField;
import org.apache.lucene.document.StringField;
import org.apache.lucene.document.TextField;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.IndexWriterConfig;
import org.apache.lucene.index.IndexWriterConfig.OpenMode;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.FSDirectory;
import org.apache.lucene.util.Version;

/**
 *
 * @author joseadiazg
 * 
 */
public class LuceneWriter {

    String pathToIndex = "";
    String pathToStopWords = "";
    IndexWriter indexWriter = null;

    private LuceneWriter() {

    }

    public LuceneWriter(String pathToIndex, String pathToStopWords) {
        this.pathToIndex = pathToIndex;
        this.pathToStopWords = pathToStopWords;
    }

    public boolean openIndex() {
        try {
            //Abrimos el directorio
            Directory dir = FSDirectory.open(new File(pathToIndex));

            //Elegimos un Analyzer . Y especificamos la versin de Lucene que usamos
            SpanishAnalyzer analyzer = new SpanishAnalyzer(Version.LUCENE_43, new CharArraySet(Version.LUCENE_43,
                    Arrays.asList(
                            StringUtils.split(FileUtils.readFileToString(new File(this.pathToStopWords), "UTF-8"))),
                    true));

            //Creamos un IndexWriterConfig 
            IndexWriterConfig iwc = new IndexWriterConfig(Version.LUCENE_43, analyzer);
            //Siempre vamos a sobreescribir el indice que tenemos en el directorio
            iwc.setOpenMode(OpenMode.CREATE);

            indexWriter = new IndexWriter(dir, iwc);

            return true;
        } catch (Exception e) {
            System.out.println("Ocurrio un problema abriendo el documento para escritura: " + e.getClass() + " :: "
                    + e.getMessage());
            return false;
        }
    }

    public void addNoticia(Noticia noticia) {
        Document doc = new Document();

        doc.add(new StringField("Title", noticia.getTitle(), Field.Store.YES));

        doc.add(new TextField("Text", noticia.getText(), Field.Store.YES));

        doc.add(new IntField("Date", noticia.getDate(), Field.Store.YES));
        try {
            indexWriter.addDocument(doc);
        } catch (IOException ex) {
            System.out.println(
                    "Ocurrio un problema al aadir el documento: " + ex.getClass() + " :: " + ex.getMessage());
        }
        System.out.println("Aadida noticia: " + noticia.getTitle());
    }

    public void finish() {
        try {
            //Realizamos un commit para guardar
            indexWriter.commit();
            //Cerramos el indexwriter
            indexWriter.close();
        } catch (IOException ex) {
            System.out.println(
                    "Ocurrio un problema cerrando el indice: " + ex.getClass() + " :: " + ex.getLocalizedMessage());
        }
    }
}