IndexReplicator.java

package proj.zoie.perf.tools;

import java.io.File;
import java.io.IOException;

import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.IndexWriter.MaxFieldLength;
import org.apache.lucene.store.Directory;
import org.apache.lucene.util.Version;

import proj.zoie.impl.indexing.internal.DiskSearchIndex;

public class IndexReplicator {

	/**
	 * @param args
	 */
	public static void main(String[] args) throws IOException{
		File srcIndex = new File(args[0]);
		File targetIndex = new File(args[1]);
		int numReplicas = Integer.parseInt(args[2]);
		System.out.println("source index: "+srcIndex.getAbsolutePath());
		System.out.println("target index: "+targetIndex.getAbsolutePath());
		System.out.println("num replications: "+numReplicas);
		IndexReader reader = null;
		try
		{
			Directory dir = DiskSearchIndex.getIndexDir(srcIndex);
			reader=IndexReader.open(dir,true);
			System.out.println("source index, numdocs: "+reader.numDocs());
		}
		catch(IOException e)
		{
			e.printStackTrace();
		}
		finally
		{
			if (reader!=null)
			{
				reader.close();
			}
			reader = null;
		}
		targetIndex.mkdirs();
		Directory targetDir = DiskSearchIndex.getIndexDir(targetIndex);
		
		IndexWriter writer = null;
		try
		{
			writer = new IndexWriter(targetDir,new StandardAnalyzer(Version.LUCENE_CURRENT),true,MaxFieldLength.UNLIMITED);
			for (int i=0;i<numReplicas;++i)
			{
			  System.out.println("replicating "+(i+1)+" time(s)");
			  writer.addIndexesNoOptimize(new Directory[]{DiskSearchIndex.getIndexDir(srcIndex)});
			}
			System.out.println("optimizing....");
			writer.optimize();
			System.out.println("done optimizing....");
		}
		finally
		{
			if (writer!=null)
			{
				writer.close();
			}
		}
	
		try
		{
			reader=IndexReader.open(targetDir,true);
			System.out.println("target index, numdocs: "+reader.numDocs());
		}
		finally
		{
			if (reader!=null)
			{
				reader.close();
			}
		}
	}
}