lucene.lab.ReaderClass.java Source code

Introduction

Here is the source code for lucene.lab.ReaderClass.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 lucene.lab;

import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.core.KeywordAnalyzer;
import org.apache.lucene.analysis.core.WhitespaceAnalyzer;
import org.apache.lucene.analysis.miscellaneous.PerFieldAnalyzerWrapper;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.index.DirectoryReader;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.queryparser.classic.MultiFieldQueryParser;
import org.apache.lucene.queryparser.classic.ParseException;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.TopScoreDocCollector;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.FSDirectory;

/**
 *
 * @author Rodrigo
 */
public class ReaderClass {

    //PerFieldAnalyzerWrapper analyzerWrapper;
    //IndexReader reader;
    //IndexSearcher searcher;
    public void ReaderClass() throws IOException {

        //Path indexPath = Paths.get("C:\\index\\");
        //Directory dir = FSDirectory.open(indexPath);
        //reader = DirectoryReader.open(dir);
        //searcher = new IndexSearcher(reader);
        //analyzerWrapper= this.createAnalyzer();
    }

    public List<Album> search(String consulta) throws ParseException, IOException {

        Path indexPath = Paths.get("C:\\index\\");
        System.out.println("Entr! fuck");
        //File indexPath = new File("C:\\index\\");
        Directory dir = FSDirectory.open(indexPath);

        IndexReader reader = DirectoryReader.open(dir);
        IndexSearcher searcher = new IndexSearcher(reader);

        PerFieldAnalyzerWrapper analyzerWrapper = this.createAnalyzer();

        Analyzer analyzer = new WhitespaceAnalyzer();
        String[] fields = { "title", "titleMin", "artist", "artistMin", "text", "summary", "tags", "tagsMin" };
        //QueryParser parser = new QueryParser(field, analyzerWrapper);
        MultiFieldQueryParser parser = new MultiFieldQueryParser(fields, analyzerWrapper);

        Query query = parser.parse(consulta);
        int hitsPerPage = 100;
        //IndexReader reader = IndexReader.open(index);
        //IndexSearcher searcher = new IndexSearcher(reader);
        TopScoreDocCollector collector = TopScoreDocCollector.create(hitsPerPage);
        searcher.search(query, collector);
        ScoreDoc[] hits = collector.topDocs().scoreDocs;

        int max_cant_comentarios = 0;
        List<Integer> lista_cant_comentarios = new ArrayList<>();
        List<Album> lista_album = new ArrayList<>();

        System.out.println("Encontr " + hits.length + " hits.");
        if (hits.length > 0) {

            float maxLuceneScore = hits[0].score;

            for (int i = 0; i < hits.length; ++i) {

                int docId = hits[i].doc;
                Document d = searcher.doc(docId);

                Album album = new Album();
                album.d = d;
                album.luceneScore = hits[i].score;

                lista_cant_comentarios.add(album.cant_comentarios());
                lista_album.add(album);

            }

            Collections.sort(lista_cant_comentarios);
            max_cant_comentarios = lista_cant_comentarios.get(lista_cant_comentarios.size() - 1);

            for (Album album : lista_album) {

                album.formulaRanking(maxLuceneScore, max_cant_comentarios);
                album.title = album.d.get("title");
                album.artist = album.d.get("artist");
                //System.out.println(album.score);
            }

            Collections.sort(lista_album);

            int i = 0;
            for (Album album : lista_album) {

                System.out.println((i + 1) + ". Ttulo: " + album.d.get("title") + "\t" + "Artista: "
                        + album.d.get("artist") + "\t" + "Score" + album.score);
                i++;
            }
        }
        return lista_album;

    }

    public PerFieldAnalyzerWrapper createAnalyzer() {

        Map<String, Analyzer> analyzerPerField = new HashMap<>();

        analyzerPerField.put("productId", new KeywordAnalyzer());
        analyzerPerField.put("title", new WhitespaceAnalyzer());
        analyzerPerField.put("titleMin", new WhitespaceAnalyzer());

        //analyzerPerField.put("userID", new KeywordAnalyzer());
        //analyzerPerField.put("profileName", new WhitespaceAnalyzer() );
        //analyzerPerField.put("score", new KeywordAnalyzer());
        analyzerPerField.put("summary", new StandardAnalyzer());
        analyzerPerField.put("text", new StandardAnalyzer());
        analyzerPerField.put("artist", new WhitespaceAnalyzer());
        analyzerPerField.put("artistMin", new WhitespaceAnalyzer());
        analyzerPerField.put("tags", new WhitespaceAnalyzer());
        analyzerPerField.put("tagsMin", new WhitespaceAnalyzer());
        // analyzerPerField.put("date",new KeywordAnalyzer());
        //analyzerPerField.put("country",new KeywordAnalyzer());
        //analyzerPerField.put("barcode", new KeywordAnalyzer());
        //analyzerPerField.put("trackCount", new KeywordAnalyzer());
        //analyzerPerField.put("label", new WhitespaceAnalyzer());
        //analyzerPerField.put("language", new KeywordAnalyzer());

        PerFieldAnalyzerWrapper analyzerW = new PerFieldAnalyzerWrapper(new StandardAnalyzer(), analyzerPerField);

        return analyzerW;
    }

}