Examples with IndexSearcherWrapper org.elasticsearch.index.shard.IndexSearcherWrapper used on opensource projects

Example 1 with IndexSearcherWrapper

use of org.elasticsearch.index.shard.IndexSearcherWrapper in project elasticsearch by elastic.

the class InternalEngineTests method testIndexSearcherWrapper.

public void testIndexSearcherWrapper() throws Exception {
    final AtomicInteger counter = new AtomicInteger();
    IndexSearcherWrapper wrapper = new IndexSearcherWrapper() {

        @Override
        public DirectoryReader wrap(DirectoryReader reader) {
            counter.incrementAndGet();
            return reader;
        }

        @Override
        public IndexSearcher wrap(IndexSearcher searcher) throws EngineException {
            counter.incrementAndGet();
            return searcher;
        }
    };
    Store store = createStore();
    Path translog = createTempDir("translog-test");
    InternalEngine engine = createEngine(store, translog);
    engine.close();
    engine = new InternalEngine(copy(engine.config(), EngineConfig.OpenMode.OPEN_INDEX_AND_TRANSLOG));
    assertTrue(engine.isRecovering());
    engine.recoverFromTranslog();
    Engine.Searcher searcher = wrapper.wrap(engine.acquireSearcher("test"));
    assertThat(counter.get(), equalTo(2));
    searcher.close();
    IOUtils.close(store, engine);
}

Example 2 with IndexSearcherWrapper

use of org.elasticsearch.index.shard.IndexSearcherWrapper in project elasticsearch by elastic.

the class IndexingMemoryControllerTests method testTranslogRecoveryWorksWithIMC.

public void testTranslogRecoveryWorksWithIMC() throws IOException {
    createIndex("test");
    ensureGreen();
    IndicesService indicesService = getInstanceFromNode(IndicesService.class);
    IndexService indexService = indicesService.indexService(resolveIndex("test"));
    IndexShard shard = indexService.getShardOrNull(0);
    for (int i = 0; i < 100; i++) {
        client().prepareIndex("test", "test", Integer.toString(i)).setSource("{\"foo\" : \"bar\"}", XContentType.JSON).get();
    }
    IndexSearcherWrapper wrapper = new IndexSearcherWrapper() {
    };
    shard.close("simon says", false);
    AtomicReference<IndexShard> shardRef = new AtomicReference<>();
    Settings settings = Settings.builder().put("indices.memory.index_buffer_size", "50kb").build();
    Iterable<IndexShard> iterable = () -> (shardRef.get() == null) ? Collections.<IndexShard>emptyList().iterator() : Collections.singleton(shardRef.get()).iterator();
    AtomicInteger flushes = new AtomicInteger();
    IndexingMemoryController imc = new IndexingMemoryController(settings, client().threadPool(), iterable) {

        @Override
        protected void writeIndexingBufferAsync(IndexShard shard) {
            assertEquals(shard, shardRef.get());
            flushes.incrementAndGet();
            shard.writeIndexingBuffer();
        }
    };
    final IndexShard newShard = IndexShardIT.newIndexShard(indexService, shard, wrapper, imc);
    shardRef.set(newShard);
    try {
        assertEquals(0, imc.availableShards().size());
        ShardRouting routing = newShard.routingEntry();
        DiscoveryNode localNode = new DiscoveryNode("foo", buildNewFakeTransportAddress(), emptyMap(), emptySet(), Version.CURRENT);
        newShard.markAsRecovering("store", new RecoveryState(routing, localNode, null));
        assertEquals(1, imc.availableShards().size());
        assertTrue(newShard.recoverFromStore());
        assertTrue("we should have flushed in IMC at least once but did: " + flushes.get(), flushes.get() >= 1);
        newShard.updateRoutingEntry(routing.moveToStarted());
    } finally {
        newShard.close("simon says", false);
    }
}