//$Id: ReaderPerfTestCase.java 20946 2011-05-17 12:28:26Z stliu $
package org.hibernate.search.test.reader;
import java.io.File;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.Random;
import java.util.List;
import org.hibernate.search.test.SearchTestCase;
import org.hibernate.search.Environment;
import org.hibernate.search.Search;
import org.hibernate.search.FullTextQuery;
import org.hibernate.search.event.FullTextIndexEventListener;
import org.hibernate.search.store.FSDirectoryProvider;
import org.hibernate.SessionFactory;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.event.PostDeleteEventListener;
import org.hibernate.event.PostUpdateEventListener;
import org.hibernate.event.PostInsertEventListener;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.queryParser.MultiFieldQueryParser;
import org.apache.lucene.analysis.StopAnalyzer;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.search.Query;
/**
* @author Emmanuel Bernard
*/
public class ReaderPerfTestCase extends SearchTestCase {
protected void setUp() throws Exception {
File sub = getBaseIndexDir();
sub.mkdir();
File[] files = sub.listFiles();
for ( File file : files ) {
if ( file.isDirectory() ) {
delete( file );
}
}
//super.setUp(); //we need a fresh session factory each time for index set up
buildSessionFactory( getMappings(), getAnnotatedPackages(), getXmlFiles() );
}
protected Class[] getMappings() {
return new Class[] {
Detective.class,
Suspect.class
};
}
private File getBaseIndexDir() {
File current = new File( "." );
File sub = new File( current, "indextemp" );
return sub;
}
protected void tearDown() throws Exception {
super.tearDown();
File sub = getBaseIndexDir();
delete( sub );
}
private void delete(File sub) {
if ( sub.isDirectory() ) {
for ( File file : sub.listFiles() ) {
delete( file );
}
sub.delete();
}
else {
sub.delete();
}
}
public boolean insert = true;
public void testConcurrency() throws Exception {
Session s = openSession( );
Transaction tx = s.beginTransaction();
for ( int index = 0 ; index < 5000 ; index++ ) {
Detective detective = new Detective();
detective.setName( "John Doe " + index );
detective.setBadge( "123455" + index );
detective.setPhysicalDescription( "Blond green eye etc etc");
s.persist( detective );
Suspect suspect = new Suspect();
suspect.setName( "Jane Doe " + index );
suspect.setPhysicalDescription( "brunette, short, 30-ish");
if ( index % 20 == 0 ) {
suspect.setSuspectCharge( "thief liar " );
}
else {
suspect.setSuspectCharge( " It's 1875 in London. The police have captured career criminal Montmorency. In the process he has been grievously wounded and it is up to a young surgeon to treat his wounds. During his recovery Montmorency learns of the city's new sewer system.");
}
s.persist( suspect );
}
tx.commit();
s.close();
Thread.sleep( 1000 );
int nThreads = 15;
ExecutorService es = Executors.newFixedThreadPool( nThreads );
Work work = new Work( getSessions() );
ReverseWork reverseWork = new ReverseWork( getSessions() );
long start = System.currentTimeMillis();
int iteration = 100;
for ( int i = 0; i < iteration; i++ ) {
es.execute( work );
es.execute( reverseWork );
}
while ( work.count.get() < iteration - 1 ) {
Thread.sleep( 20 );
}
System.out.println( iteration + " iterations in " + nThreads + " threads: " + ( System
.currentTimeMillis() - start ) );
}
protected class Work implements Runnable {
private Random random = new Random( );
private SessionFactory sf;
// public volatile int count = 0;
public AtomicInteger count = new AtomicInteger(0);
public Work(SessionFactory sf) {
this.sf = sf;
}
public void run() {
Session s = sf.openSession();
Transaction tx = s.beginTransaction();
QueryParser parser = new MultiFieldQueryParser(
new String[] {"name", "physicalDescription", "suspectCharge"},
new StandardAnalyzer() );
FullTextQuery query = getQuery( "John Doe", parser, s );
assertTrue( query.getResultSize() != 0 );
query = getQuery( "green", parser, s );
random.nextInt( query.getResultSize() - 15 );
query.setFirstResult( random.nextInt( query.getResultSize() - 15 ) );
query.setMaxResults( 10 );
query.list();
tx.commit();
s.close();
s = sf.openSession();
tx = s.beginTransaction();
query = getQuery( "John Doe", parser, s );
assertTrue( query.getResultSize() != 0 );
query = getQuery( "thief", parser, s );
int firstResult = random.nextInt( query.getResultSize() - 15 );
query.setFirstResult( firstResult );
query.setMaxResults( 10 );
List result = query.list();
Object object = result.get(0);
if (insert && object instanceof Detective) {
Detective detective = (Detective) object;
detective.setPhysicalDescription( detective.getPhysicalDescription() + " Eye" + firstResult );
}
else if (insert && object instanceof Suspect) {
Suspect suspect = (Suspect) object;
suspect.setPhysicalDescription( suspect.getPhysicalDescription() + " Eye" + firstResult );
}
tx.commit();
s.close();
count.incrementAndGet();
}
private FullTextQuery getQuery(String queryString, QueryParser parser, Session s) {
Query luceneQuery = null;
try {
luceneQuery = parser.parse(queryString);
}
catch (ParseException e) {
e.printStackTrace();
}
return Search.createFullTextSession( s ).createFullTextQuery( luceneQuery );
}
}
protected class ReverseWork implements Runnable {
private SessionFactory sf;
private Random random = new Random();
public ReverseWork(SessionFactory sf) {
this.sf = sf;
}
public void run() {
Session s = sf.openSession();
Transaction tx = s.beginTransaction();
QueryParser parser = new MultiFieldQueryParser(
new String[] {"name", "physicalDescription", "suspectCharge"},
new StandardAnalyzer() );
FullTextQuery query = getQuery( "John Doe", parser, s );
assertTrue( query.getResultSize() != 0 );
query = getQuery( "london", parser, s );
random.nextInt( query.getResultSize() - 15 );
query.setFirstResult( random.nextInt( query.getResultSize() - 15 ) );
query.setMaxResults( 10 );
query.list();
tx.commit();
s.close();
s = sf.openSession();
tx = s.beginTransaction();
getQuery( "John Doe", parser, s );
assertTrue( query.getResultSize() != 0 );
query = getQuery( "green", parser, s );
random.nextInt( query.getResultSize() - 15 );
query.setFirstResult( random.nextInt( query.getResultSize() - 15 ) );
query.setMaxResults( 10 );
query.list();
tx.commit();
s.close();
}
private FullTextQuery getQuery(String queryString, QueryParser parser, Session s) {
Query luceneQuery = null;
try {
luceneQuery = parser.parse(queryString);
}
catch (ParseException e) {
e.printStackTrace();
}
return Search.createFullTextSession( s ).createFullTextQuery( luceneQuery );
}
}
protected void configure(org.hibernate.cfg.Configuration cfg) {
super.configure( cfg );
File sub = getBaseIndexDir();
cfg.setProperty( "hibernate.search.default.indexBase", sub.getAbsolutePath() );
cfg.setProperty( "hibernate.search.Clock.directory_provider", FSDirectoryProvider.class.getName() );
cfg.setProperty( Environment.ANALYZER_CLASS, StopAnalyzer.class.getName() );
FullTextIndexEventListener del = new FullTextIndexEventListener();
cfg.getEventListeners().setPostDeleteEventListeners( new PostDeleteEventListener[]{del} );
cfg.getEventListeners().setPostUpdateEventListeners( new PostUpdateEventListener[]{del} );
cfg.getEventListeners().setPostInsertEventListeners( new PostInsertEventListener[]{del} );
}
}