FSDirectoryTest.java example

Explorer
hibernate-search-master
/*
 * Hibernate Search, full-text search for your domain model
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.search.test.directoryProvider;

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

import org.apache.lucene.index.DirectoryReader;
import org.apache.lucene.index.DocsEnum;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.LeafReader;
import org.apache.lucene.index.LeafReaderContext;
import org.apache.lucene.index.Term;
import org.apache.lucene.queryparser.classic.QueryParser;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TermQuery;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.FSDirectory;
import org.hibernate.Session;
import org.hibernate.search.test.Document;
import org.hibernate.search.test.SearchTestBase;
import org.hibernate.search.testsupport.TestConstants;
import org.hibernate.search.testsupport.junit.SkipOnElasticsearch;
import org.hibernate.search.util.impl.FileHelper;
import org.junit.Assert;
import org.junit.Test;
import org.junit.experimental.categories.Category;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

/**
 * @author Gavin King
 * @author Sanne Grinovero
 */
@Category(SkipOnElasticsearch.class) // Directories are specific to the Lucene backend
public class FSDirectoryTest extends SearchTestBase {

	@Test
	public void testEventIntegration() throws Exception {
		try ( Session s = getSessionFactory().openSession() ) {
			s.getTransaction().begin();
			s.persist( new Document( "Hibernate in Action", "Object/relational mapping with Hibernate", "blah blah blah" ) );
			s.getTransaction().commit();
		}

		Document entity;

		try ( Directory dir = FSDirectory.open( getBaseIndexDir().resolve( "Documents" ) ) ) {
			try ( IndexReader reader = DirectoryReader.open( dir ) ) {
				int num = reader.numDocs();
				assertEquals( 1, num );
				assertEquals( 1, reader.docFreq( new Term( "Abstract", "hibernate" ) ) );
				assertEquals( 1, reader.docFreq( new Term( "title", "action" ) ) );
				assertEquals( "1", projectSingleField( reader, "id", new Term( "title", "action" ) ) );
			}

			try ( Session s = getSessionFactory().openSession() ) {
				s.getTransaction().begin();
				entity = s.get( Document.class, Long.valueOf( 1 ) );
				entity.setSummary( "Object/relational mapping with EJB3" );
				s.persist( new Document( "Seam in Action", "", "blah blah blah blah" ) );
				s.getTransaction().commit();
			}

			try ( IndexReader reader = DirectoryReader.open( dir ) ) {
				int num = reader.numDocs();
				assertEquals( 2, num );
				assertEquals( 1, reader.docFreq( new Term( "Abstract", "ejb" ) ) );
			}

			try ( Session s = getSessionFactory().openSession() ) {
				s.getTransaction().begin();
				s.delete( entity );
				s.getTransaction().commit();
			}

			try ( IndexReader reader = DirectoryReader.open( dir ) ) {
				int num = reader.numDocs();
				assertEquals( 1, num );
				assertEquals( 1, reader.docFreq( new Term( "title", "seam" ) ) );
				assertEquals( "2", projectSingleField( reader, "id", new Term( "title", "seam" ) ) );
			}
		}

		try ( Session s = getSessionFactory().openSession() ) {
			s.getTransaction().begin();
			s.delete( s.createCriteria( Document.class ).uniqueResult() );
			s.getTransaction().commit();
		}
	}

	/**
	 * Project a field as a String from a Lucene Document matching the provided term.
	 * The method asserts that one match is found, and no more.
	 */
	private String projectSingleField(IndexReader reader, String fieldName, Term term) throws IOException {
		String projection = null;
		for ( LeafReaderContext leaf : reader.leaves() ) {
			final LeafReader atomicReader = leaf.reader();
			final DocsEnum termDocsEnum = atomicReader.termDocsEnum( term );
			while ( termDocsEnum.nextDoc() != DocsEnum.NO_MORE_DOCS ) {
				final int docID = termDocsEnum.docID();
				org.apache.lucene.document.Document document = reader.document( docID );
				String value = document.get( fieldName );
				Assert.assertNull( "duplicate matches found! This method assumes a single document will match the Term.", projection );
				projection = value;
			}
		}
		Assert.assertNotNull( projection );
		return projection;
	}

	@SuppressWarnings("unchecked")
	@Test
	public void testBoost() throws Exception {
		try ( Session s = getSessionFactory().openSession() ) {
			s.getTransaction().begin();
			s.persist( new Document( "Hibernate in Action", "Object and Relational", "blah blah blah" ) );
			s.persist( new Document( "Object and Relational", "Hibernate in Action", "blah blah blah" ) );
			s.getTransaction().commit();
		}

		try ( FSDirectory dir = FSDirectory.open( getBaseIndexDir().resolve( "Documents" ) ) ) {
			try ( IndexReader indexReader = DirectoryReader.open( dir ) ) {
				IndexSearcher searcher = new IndexSearcher( indexReader );
				QueryParser qp = new QueryParser( "id", TestConstants.standardAnalyzer );
				Query query = qp.parse( "title:Action OR Abstract:Action" );
				TopDocs hits = searcher.search( query, 1000 );
				assertEquals( 2, hits.totalHits );
				assertTrue( hits.scoreDocs[0].score == 2 * hits.scoreDocs[1].score );
				org.apache.lucene.document.Document doc = searcher.doc( 0 );
				assertEquals( "Hibernate in Action", doc.get( "title" ) );
			}
		}

		try ( Session s = getSessionFactory().openSession() ) {
			s.getTransaction().begin();
			List list = s.createQuery( "from Document" ).list();
			for ( Document document : (List<Document>) list ) {
				s.delete( document );
			}
			s.getTransaction().commit();
		}
	}

	@Test
	public void testSearchOnDeletedIndex() throws Exception {
		try ( Session s = getSessionFactory().openSession() ) {
			s.getTransaction().begin();
			s.persist( new Document( "Hibernate Search in Action", "", "" ) );
			s.getTransaction().commit();
		}

		try ( Directory dir = FSDirectory.open( getBaseIndexDir().resolve( "Documents" ) ) ) {
			try ( IndexReader indexReader = DirectoryReader.open( dir ) ) {
				IndexSearcher searcher = new IndexSearcher( indexReader );
				// deleting before search, but after IndexSearcher creation:
				// ( fails when deleting -concurrently- to IndexSearcher initialization! )
				// Use the "tryDelete" form to ignore exceptions on Windows:
				FileHelper.tryDelete( getBaseIndexDir() );
				TermQuery query = new TermQuery( new Term( "title", "action" ) );
				TopDocs hits = searcher.search( query, 1000 );
				assertEquals( 1, hits.totalHits );
				org.apache.lucene.document.Document doc = searcher.doc( 0 );
				assertEquals( "Hibernate Search in Action", doc.get( "title" ) );
			}
		}
	}

	@Override
	public Class<?>[] getAnnotatedClasses() {
		return new Class[] { Document.class };
	}

	@Override
	public void configure(Map<String,Object> cfg) {
		cfg.put( "hibernate.search.default.directory_provider", "filesystem" );
	}

}