TestSalesmanSearch.java

package com.manning.hsia.dvdstore.ex12_1;

import org.apache.lucene.index.Term;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TermQuery;
import org.hibernate.Transaction;
import org.hibernate.search.FullTextQuery;
import org.hibernate.search.FullTextSession;
import org.hibernate.search.Search;
import org.hibernate.search.store.FSDirectoryProvider;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

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

import com.manning.hsia.test.ch12.SearchTestCase;
import com.manning.hsia.test.Product;

public class TestSalesmanSearch extends SearchTestCase {

	@Test(groups="ch12")
	public void searchProduct() throws Exception {
		FullTextSession session = Search.getFullTextSession(openSession());
		Transaction tx = session.beginTransaction();

		try {
			Query query = new TermQuery(new Term("description", "salesman"));
			System.out.println(query.toString());

			org.hibernate.search.FullTextQuery hibQuery =
				session.createFullTextQuery(query,
					Product.class);

			hibQuery.setProjection(FullTextQuery.DOCUMENT,
				FullTextQuery.SCORE,
				FullTextQuery.DOCUMENT_ID,
				FullTextQuery.EXPLANATION);

			List<Object[]> results = hibQuery.list();

			assert results.size() > 0 : "no results returned";
			for (Object[] result : results) {
				System.out.println("score => " + result[1]);

				System.out.println(hibQuery
					.explain((Integer) result[2]));
			}
			for (Object element :
				session.createQuery("from " + Product.class.getName()).list())
				session.delete(element);
			tx.commit();
		}
		finally {
			session.close();
		}
	}

	protected Class[] getMappings() {
		return new Class[]{
			Product.class,
		};
	}

	@BeforeClass(groups="ch12", alwaysRun=true)
	protected void setUp() throws Exception {
		File sub = locateBaseDir();
		File[] files = sub.listFiles();
		if (files != null) {
			for (File file : files) {
				if (file.isDirectory()) {
					delete(file);
				}
			}
		}
		copyIndexes();
		buildSessionFactory(getMappings(), getAnnotatedPackages(), getXmlFiles());
	}

	protected void configure(org.hibernate.cfg.Configuration cfg) {
		super.configure(cfg);
		cfg.setProperty("hibernate.search.default.directory_provider", FSDirectoryProvider.class.getName());
		cfg.setProperty("hibernate.search.default.indexBase", locateBaseDir().getAbsolutePath());
//		cfg.setProperty("hibernate.search.use_query_cache", "true");
	}
}