Example usage for org.apache.solr.common.util ObjectReleaseTracker OBJECTS

Introduction

In this page you can find the example usage for org.apache.solr.common.util ObjectReleaseTracker OBJECTS.

Prototype

Map OBJECTS

To view the source code for org.apache.solr.common.util ObjectReleaseTracker OBJECTS.

Click Source Link

Usage

From source file:org.apache.beam.sdk.io.solr.SolrIOTest.java

License:Apache License

/**
 * Test that retries are invoked when Solr returns error. We invoke this by calling a non existing
 * collection, and use a strategy that will retry on any SolrException. The logger is used to
 * verify expected behavior./* w  w w .ja v  a  2  s.co  m*/
 */
@Test
public void testWriteRetry() throws Throwable {
    thrown.expect(IOException.class);
    thrown.expectMessage("Error writing to Solr");

    // entry state of the release tracker to ensure we only unregister newly created objects
    @SuppressWarnings("unchecked")
    Set<Object> entryState = ImmutableSet.copyOf(ObjectReleaseTracker.OBJECTS.keySet());

    SolrIO.Write write = SolrIO.write().withConnectionConfiguration(connectionConfiguration)
            .withRetryConfiguration(SolrIO.RetryConfiguration.create(3, Duration.standardMinutes(3))
                    .withRetryPredicate(new LenientRetryPredicate()))
            .to("wrong-collection");

    List<SolrInputDocument> data = SolrIOTestUtils.createDocuments(NUM_DOCS);
    pipeline.apply(Create.of(data)).apply(write);

    try {
        pipeline.run();
    } catch (final Pipeline.PipelineExecutionException e) {
        // Hack: await all worker threads completing (BEAM-4040)
        int waitAttempts = 30; // defensive coding
        while (namedThreadIsAlive("direct-runner-worker") && waitAttempts-- >= 0) {
            LOG.info("Pausing to allow direct-runner-worker threads to finish");
            Thread.sleep(1000);
        }

        // remove solrClients created by us as there are no guarantees on Teardown here
        for (Object o : ObjectReleaseTracker.OBJECTS.keySet()) {
            if (o instanceof SolrZkClient && !entryState.contains(o)) {
                LOG.info("Removing unreleased SolrZkClient");
                ObjectReleaseTracker.release(o);
            }
        }

        // check 2 retries were initiated by inspecting the log before passing on the exception
        expectedLogs.verifyWarn(String.format(SolrIO.Write.WriteFn.RETRY_ATTEMPT_LOG, 1));
        expectedLogs.verifyWarn(String.format(SolrIO.Write.WriteFn.RETRY_ATTEMPT_LOG, 2));

        throw e.getCause();
    }

    fail("Pipeline should not have run to completion");
}

---