lucene-java-3.5.0/lucene/contrib/benchmark/src/java/org/apache/lucene/benchmark/byTask/tasks/ForceMergeTask.java

   1 package org.apache.lucene.benchmark.byTask.tasks;
   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
  20 import org.apache.lucene.benchmark.byTask.PerfRunData;
  21 import org.apache.lucene.index.IndexWriter;
  22
  23 /**
  24  * Runs forceMerge on the index.
  25  * <br>Other side effects: none.
  26  */
  27 public class ForceMergeTask extends PerfTask {
  28
  29   public ForceMergeTask(PerfRunData runData) {
  30     super(runData);
  31   }
  32
  33   int maxNumSegments = -1;
  34
  35   @Override
  36   public int doLogic() throws Exception {
  37     if (maxNumSegments == -1) {
  38       throw new IllegalStateException("required argument (maxNumSegments) was not specified");
  39     }
  40     IndexWriter iw = getRunData().getIndexWriter();
  41     iw.forceMerge(maxNumSegments);
  42     //System.out.println("forceMerge called");
  43     return 1;
  44   }
  45
  46   @Override
  47   public void setParams(String params) {
  48     super.setParams(params);
  49     maxNumSegments = Double.valueOf(params).intValue();
  50   }
  51
  52   @Override
  53   public boolean supportsParams() {
  54     return true;
  55   }
  56 }