| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.lucene.benchmark.byTask.tasks; |
| |
| |
| import java.io.IOException; |
| import java.util.Collection; |
| import java.util.Map; |
| |
| import org.apache.lucene.benchmark.byTask.PerfRunData; |
| import org.apache.lucene.index.DirectoryReader; |
| import org.apache.lucene.index.IndexCommit; |
| import org.apache.lucene.store.Directory; |
| |
| /** |
| * Open an index reader. |
| * <br>Other side effects: index reader object in perfRunData is set. |
| * <br> Optional params commitUserData eg. OpenReader(false,commit1) |
| */ |
| public class OpenReaderTask extends PerfTask { |
| public static final String USER_DATA = "userData"; |
| private String commitUserData = null; |
| |
| public OpenReaderTask(PerfRunData runData) { |
| super(runData); |
| } |
| |
| @Override |
| public int doLogic() throws IOException { |
| Directory dir = getRunData().getDirectory(); |
| DirectoryReader r = null; |
| if (commitUserData != null) { |
| r = DirectoryReader.open(OpenReaderTask.findIndexCommit(dir, commitUserData)); |
| } else { |
| r = DirectoryReader.open(dir); |
| } |
| getRunData().setIndexReader(r); |
| // We transfer reference to the run data |
| r.decRef(); |
| return 1; |
| } |
| |
| @Override |
| public void setParams(String params) { |
| super.setParams(params); |
| if (params != null) { |
| String[] split = params.split(","); |
| if (split.length > 0) { |
| commitUserData = split[0]; |
| } |
| } |
| } |
| |
| @Override |
| public boolean supportsParams() { |
| return true; |
| } |
| |
| public static IndexCommit findIndexCommit(Directory dir, String userData) throws IOException { |
| Collection<IndexCommit> commits = DirectoryReader.listCommits(dir); |
| for (final IndexCommit ic : commits) { |
| Map<String,String> map = ic.getUserData(); |
| String ud = null; |
| if (map != null) { |
| ud = map.get(USER_DATA); |
| } |
| if (ud != null && ud.equals(userData)) { |
| return ic; |
| } |
| } |
| |
| throw new IOException("index does not contain commit with userData: " + userData); |
| } |
| } |