lucene-java-3.5.0/lucene/src/test/org/apache/lucene/TestMergeSchedulerExternal.java

   1 package org.apache.lucene;
   2
   3 /**
   4  * Licensed to the Apache Software Foundation (ASF) under one or more
   5  * contributor license agreements.  See the NOTICE file distributed with
   6  * this work for additional information regarding copyright ownership.
   7  * The ASF licenses this file to You under the Apache License, Version 2.0
   8  * (the "License"); you may not use this file except in compliance with
   9  * the License.  You may obtain a copy of the License at
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  */
  19 import java.io.IOException;
  20
  21 import org.apache.lucene.util.LuceneTestCase;
  22 import org.apache.lucene.store.Directory;
  23 import org.apache.lucene.store.MockDirectoryWrapper;
  24 import org.apache.lucene.store.RAMDirectory;
  25 import org.apache.lucene.index.CorruptIndexException;
  26 import org.apache.lucene.index.IndexWriter;
  27 import org.apache.lucene.index.IndexWriterConfig;
  28 import org.apache.lucene.index.LogMergePolicy;
  29 import org.apache.lucene.index.MergePolicy;
  30 import org.apache.lucene.index.ConcurrentMergeScheduler;
  31 import org.apache.lucene.index.MergeScheduler;
  32 import org.apache.lucene.index.MergePolicy.OneMerge;
  33 import org.apache.lucene.analysis.MockAnalyzer;
  34 import org.apache.lucene.document.Document;
  35 import org.apache.lucene.document.Field;
  36
  37 /**
  38  * Holds tests cases to verify external APIs are accessible
  39  * while not being in org.apache.lucene.index package.
  40  */
  41 public class TestMergeSchedulerExternal extends LuceneTestCase {
  42
  43   volatile boolean mergeCalled;
  44   volatile boolean mergeThreadCreated;
  45   volatile boolean excCalled;
  46
  47   private class MyMergeScheduler extends ConcurrentMergeScheduler {
  48
  49     private class MyMergeThread extends ConcurrentMergeScheduler.MergeThread {
  50       public MyMergeThread(IndexWriter writer, MergePolicy.OneMerge merge) throws IOException {
  51         super(writer, merge);
  52         mergeThreadCreated = true;
  53       }
  54     }
  55
  56     @Override
  57     protected MergeThread getMergeThread(IndexWriter writer, MergePolicy.OneMerge merge) throws IOException {
  58       MergeThread thread = new MyMergeThread(writer, merge);
  59       thread.setThreadPriority(getMergeThreadPriority());
  60       thread.setDaemon(true);
  61       thread.setName("MyMergeThread");
  62       return thread;
  63     }
  64
  65     @Override
  66     protected void handleMergeException(Throwable t) {
  67       excCalled = true;
  68     }
  69
  70     @Override
  71     protected void doMerge(MergePolicy.OneMerge merge) throws IOException {
  72       mergeCalled = true;
  73       super.doMerge(merge);
  74     }
  75   }
  76
  77   private static class FailOnlyOnMerge extends MockDirectoryWrapper.Failure {
  78     @Override
  79     public void eval(MockDirectoryWrapper dir)  throws IOException {
  80       StackTraceElement[] trace = new Exception().getStackTrace();
  81       for (int i = 0; i < trace.length; i++) {
  82         if ("doMerge".equals(trace[i].getMethodName()))
  83           throw new IOException("now failing during merge");
  84       }
  85     }
  86   }
  87
  88   public void testSubclassConcurrentMergeScheduler() throws IOException {
  89     MockDirectoryWrapper dir = newDirectory();
  90     dir.failOn(new FailOnlyOnMerge());
  91
  92     Document doc = new Document();
  93     Field idField = newField("id", "", Field.Store.YES, Field.Index.NOT_ANALYZED);
  94     doc.add(idField);
  95
  96     IndexWriter writer = new IndexWriter(dir, newIndexWriterConfig(
  97         TEST_VERSION_CURRENT, new MockAnalyzer(random)).setMergeScheduler(new MyMergeScheduler())
  98         .setMaxBufferedDocs(2).setRAMBufferSizeMB(IndexWriterConfig.DISABLE_AUTO_FLUSH)
  99         .setMergePolicy(newLogMergePolicy()));
 100     LogMergePolicy logMP = (LogMergePolicy) writer.getConfig().getMergePolicy();
 101     logMP.setMergeFactor(10);
 102     for(int i=0;i<20;i++)
 103       writer.addDocument(doc);
 104
 105     ((MyMergeScheduler) writer.getConfig().getMergeScheduler()).sync();
 106     writer.close();
 107
 108     assertTrue(mergeThreadCreated);
 109     assertTrue(mergeCalled);
 110     assertTrue(excCalled);
 111     dir.close();
 112   }
 113
 114   private static class ReportingMergeScheduler extends MergeScheduler {
 115
 116     @Override
 117     public void merge(IndexWriter writer) throws CorruptIndexException, IOException {
 118       OneMerge merge = null;
 119       while ((merge = writer.getNextMerge()) != null) {
 120         if (VERBOSE) {
 121           System.out.println("executing merge " + merge.segString(writer.getDirectory()));
 122         }
 123         writer.merge(merge);
 124       }
 125     }
 126
 127     @Override
 128     public void close() throws CorruptIndexException, IOException {}
 129
 130   }
 131
 132   public void testCustomMergeScheduler() throws Exception {
 133     // we don't really need to execute anything, just to make sure the custom MS
 134     // compiles. But ensure that it can be used as well, e.g., no other hidden
 135     // dependencies or something. Therefore, don't use any random API !
 136     Directory dir = new RAMDirectory();
 137     IndexWriterConfig conf = new IndexWriterConfig(TEST_VERSION_CURRENT, null);
 138     conf.setMergeScheduler(new ReportingMergeScheduler());
 139     IndexWriter writer = new IndexWriter(dir, conf);
 140     writer.addDocument(new Document());
 141     writer.commit(); // trigger flush
 142     writer.addDocument(new Document());
 143     writer.commit(); // trigger flush
 144     writer.forceMerge(1);
 145     writer.close();
 146     dir.close();
 147   }
 148
 149 }