+++ /dev/null
-package org.apache.lucene.search;
-/**
- * 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.
- */
-
-import org.apache.lucene.util.LuceneTestCase;
-import org.apache.lucene.store.Directory;
-import org.apache.lucene.index.IndexReader;
-import org.apache.lucene.index.IndexWriter;
-import org.apache.lucene.index.IndexWriterConfig;
-import org.apache.lucene.index.IndexWriterConfig.OpenMode;
-import org.apache.lucene.analysis.MockAnalyzer;
-import org.apache.lucene.document.*;
-
-import java.util.Random;
-import java.util.List;
-import java.util.concurrent.atomic.AtomicBoolean;
-import java.io.IOException;
-
-public class TestThreadSafe extends LuceneTestCase {
- Directory dir1;
-
- IndexReader ir1;
-
- class Thr extends Thread {
- final int iter;
- final Random rand;
- final AtomicBoolean failed;
-
- // pass in random in case we want to make things reproducable
- public Thr(int iter, Random rand, AtomicBoolean failed) {
- this.iter = iter;
- this.rand = rand;
- this.failed = failed;
- }
-
- @Override
- public void run() {
- try {
- for (int i=0; i<iter; i++) {
- /*** future
- // pick a random index reader... a shared one, or create your own
- IndexReader ir;
- ***/
-
- switch(rand.nextInt(1)) {
- case 0: loadDoc(ir1); break;
- }
-
- }
- } catch (Throwable th) {
- failed.set(true);
- throw new RuntimeException(th);
- }
- }
-
-
- void loadDoc(IndexReader ir) throws IOException {
- // beware of deleted docs in the future
- Document doc = ir.document(rand.nextInt(ir.maxDoc()),
- new FieldSelector() {
- public FieldSelectorResult accept(String fieldName) {
- switch(rand.nextInt(2)) {
- case 0: return FieldSelectorResult.LAZY_LOAD;
- case 1: return FieldSelectorResult.LOAD;
- // TODO: add other options
- default: return FieldSelectorResult.LOAD;
- }
- }
- }
- );
-
- List<Fieldable> fields = doc.getFields();
- for (final Fieldable f : fields ) {
- validateField(f);
- }
-
- }
-
- }
-
-
- void validateField(Fieldable f) {
- String val = f.stringValue();
- if (!val.startsWith("^") || !val.endsWith("$")) {
- throw new RuntimeException("Invalid field:" + f.toString() + " val=" +val);
- }
- }
-
- String[] words = "now is the time for all good men to come to the aid of their country".split(" ");
-
- void buildDir(Directory dir, int nDocs, int maxFields, int maxFieldLen) throws IOException {
- IndexWriter iw = new IndexWriter(dir, new IndexWriterConfig(
- TEST_VERSION_CURRENT, new MockAnalyzer(random)).setOpenMode(OpenMode.CREATE).setMaxBufferedDocs(10));
- for (int j=0; j<nDocs; j++) {
- Document d = new Document();
- int nFields = random.nextInt(maxFields);
- for (int i=0; i<nFields; i++) {
- int flen = random.nextInt(maxFieldLen);
- StringBuilder sb = new StringBuilder("^ ");
- while (sb.length() < flen) sb.append(' ').append(words[random.nextInt(words.length)]);
- sb.append(" $");
- Field.Store store = Field.Store.YES; // make random later
- Field.Index index = Field.Index.ANALYZED; // make random later
- d.add(newField("f"+i, sb.toString(), store, index));
- }
- iw.addDocument(d);
- }
- iw.close();
- }
-
-
- void doTest(int iter, int nThreads) throws Exception {
- Thr[] tarr = new Thr[nThreads];
- AtomicBoolean failed = new AtomicBoolean();
- for (int i=0; i<nThreads; i++) {
- tarr[i] = new Thr(iter, new Random(random.nextLong()), failed);
- tarr[i].start();
- }
- for (int i=0; i<nThreads; i++) {
- tarr[i].join();
- }
- assertFalse(failed.get());
- }
-
- public void testLazyLoadThreadSafety() throws Exception{
- dir1 = newDirectory();
- // test w/ field sizes bigger than the buffer of an index input
- buildDir(dir1, 15, 5, 2000);
-
- // do many small tests so the thread locals go away inbetween
- int num = atLeast(10);
- for (int i = 0; i < num; i++) {
- ir1 = IndexReader.open(dir1, false);
- doTest(10,10);
- ir1.close();
- }
- dir1.close();
- }
-
-}