lucene-java-3.4.0/lucene/contrib/benchmark/src/java/org/apache/lucene/benchmark/byTask/tasks/CommitIndexTask.java

   1 package org.apache.lucene.benchmark.byTask.tasks;
   2 /**
   3  * Licensed to the Apache Software Foundation (ASF) under one or more
   4  * contributor license agreements.  See the NOTICE file distributed with
   5  * this work for additional information regarding copyright ownership.
   6  * The ASF licenses this file to You under the Apache License, Version 2.0
   7  * (the "License"); you may not use this file except in compliance with
   8  * the License.  You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 import java.util.HashMap;
  20 import java.util.Map;
  21
  22 import org.apache.lucene.benchmark.byTask.PerfRunData;
  23 import org.apache.lucene.index.IndexWriter;
  24 import org.apache.lucene.index.IndexReader;
  25
  26 /**
  27  * Commits the IndexWriter.
  28  *
  29  */
  30 public class CommitIndexTask extends PerfTask {
  31   Map<String,String> commitUserData;
  32
  33   public CommitIndexTask(PerfRunData runData) {
  34     super(runData);
  35   }
  36
  37   @Override
  38   public boolean supportsParams() {
  39     return true;
  40   }
  41
  42   @Override
  43   public void setParams(String params) {
  44     super.setParams(params);
  45     commitUserData = new HashMap<String,String>();
  46     commitUserData.put(OpenReaderTask.USER_DATA, params);
  47   }
  48
  49   @Override
  50   public int doLogic() throws Exception {
  51     IndexWriter iw = getRunData().getIndexWriter();
  52     if (iw != null) {
  53       iw.commit(commitUserData);
  54     } else {
  55       IndexReader r = getRunData().getIndexReader();
  56       if (r != null) {
  57         r.commit(commitUserData);
  58         r.decRef();
  59       } else {
  60         throw new IllegalStateException("neither IndexWriter nor IndexReader is currently open");
  61       }
  62     }
  63
  64     return 1;
  65   }
  66 }