Example usage for org.apache.lucene.store BaseDirectoryWrapper close

Introduction

In this page you can find the example usage for org.apache.lucene.store BaseDirectoryWrapper close.

Prototype

@Override
    public void close() throws IOException 

Source Link

Usage

From source file:com.meizu.nlp.classification.utils.DataSplitterTest.java

License:Apache License

public static void assertSplit(LeafReader originalIndex, double testRatio, double crossValidationRatio,
        String... fieldNames) throws Exception {

    BaseDirectoryWrapper trainingIndex = newDirectory();
    BaseDirectoryWrapper testIndex = newDirectory();
    BaseDirectoryWrapper crossValidationIndex = newDirectory();

    try {//  w ww. j a  va 2s .  c  o  m
        DatasetSplitter datasetSplitter = new DatasetSplitter(testRatio, crossValidationRatio);
        datasetSplitter.split(originalIndex, trainingIndex, testIndex, crossValidationIndex,
                new MockAnalyzer(random()), fieldNames);

        assertNotNull(trainingIndex);
        assertNotNull(testIndex);
        assertNotNull(crossValidationIndex);

        DirectoryReader trainingReader = DirectoryReader.open(trainingIndex);
        assertTrue((int) (originalIndex.maxDoc() * (1d - testRatio - crossValidationRatio)) == trainingReader
                .maxDoc());
        DirectoryReader testReader = DirectoryReader.open(testIndex);
        assertTrue((int) (originalIndex.maxDoc() * testRatio) == testReader.maxDoc());
        DirectoryReader cvReader = DirectoryReader.open(crossValidationIndex);
        assertTrue((int) (originalIndex.maxDoc() * crossValidationRatio) == cvReader.maxDoc());

        trainingReader.close();
        testReader.close();
        cvReader.close();
        closeQuietly(trainingReader);
        closeQuietly(testReader);
        closeQuietly(cvReader);
    } finally {
        if (trainingIndex != null) {
            trainingIndex.close();
        }
        if (testIndex != null) {
            testIndex.close();
        }
        if (crossValidationIndex != null) {
            crossValidationIndex.close();
        }
    }
}

---

From source file:org.elasticsearch.index.store.DirectoryUtilsTest.java

License:Apache License

@Test
public void testGetLeave() throws IOException {
    File file = ElasticsearchTestCase.newTempDir(LifecycleScope.TEST);
    final int iters = scaledRandomIntBetween(10, 100);
    for (int i = 0; i < iters; i++) {
        {/*from   w  ww  .j  a  va 2s  .c om*/
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(new FilterDirectory(dir) {
            }, FSDirectory.class, null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(dir, FSDirectory.class, null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            Set<String> stringSet = Collections.emptySet();
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(
                    new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean()), FSDirectory.class,
                    null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            Set<String> stringSet = Collections.emptySet();
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(
                    new FilterDirectory(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean())) {
                    }, FSDirectory.class, null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            Set<String> stringSet = Collections.emptySet();
            BaseDirectoryWrapper dir = newFSDirectory(file);
            RAMDirectory directory = DirectoryUtils.getLeaf(
                    new FilterDirectory(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean())) {
                    }, RAMDirectory.class, null);
            assertThat(directory, nullValue());
            dir.close();
        }

    }
}

---

From source file:org.elasticsearch.index.store.DirectoryUtilsTests.java

License:Apache License

@Test
public void testGetLeave() throws IOException {
    Path file = createTempDir();// www  .j  a v a 2 s.  co m
    final int iters = scaledRandomIntBetween(10, 100);
    for (int i = 0; i < iters; i++) {
        {
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(new FilterDirectory(dir) {
            }, FSDirectory.class, null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(dir, FSDirectory.class, null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            Set<String> stringSet = Collections.emptySet();
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(
                    new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean()), FSDirectory.class,
                    null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            Set<String> stringSet = Collections.emptySet();
            BaseDirectoryWrapper dir = newFSDirectory(file);
            FSDirectory directory = DirectoryUtils.getLeaf(
                    new FilterDirectory(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean())) {
                    }, FSDirectory.class, null);
            assertThat(directory, notNullValue());
            assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir, null)));
            dir.close();
        }

        {
            Set<String> stringSet = Collections.emptySet();
            BaseDirectoryWrapper dir = newFSDirectory(file);
            RAMDirectory directory = DirectoryUtils.getLeaf(
                    new FilterDirectory(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean())) {
                    }, RAMDirectory.class, null);
            assertThat(directory, nullValue());
            dir.close();
        }

    }
}

---