Search in sources :

Example 1 with ConnectionPoolClosedEvent

use of com.mongodb.event.ConnectionPoolClosedEvent in project mongo-java-driver by mongodb.

the class DefaultConnectionPool method close.

@Override
public void close() {
    if (!closed) {
        pool.close();
        if (sizeMaintenanceTimer != null) {
            sizeMaintenanceTimer.shutdownNow();
        }
        shutdownAsyncGetter();
        closed = true;
        connectionPoolListener.connectionPoolClosed(new ConnectionPoolClosedEvent(serverId));
    }
}
Also used : ConnectionPoolClosedEvent(com.mongodb.event.ConnectionPoolClosedEvent)

Example 2 with ConnectionPoolClosedEvent

use of com.mongodb.event.ConnectionPoolClosedEvent in project mongo-java-driver by mongodb.

the class DefaultConnectionPool method close.

@Override
public void close() {
    if (stateAndGeneration.close()) {
        pool.close();
        backgroundMaintenance.close();
        asyncWorkManager.close();
        openConcurrencyLimiter.signalClosedOrPaused();
        connectionPoolListener.connectionPoolClosed(new ConnectionPoolClosedEvent(serverId));
    }
}
Also used : ConnectionPoolClosedEvent(com.mongodb.event.ConnectionPoolClosedEvent)

Example 3 with ConnectionPoolClosedEvent

use of com.mongodb.event.ConnectionPoolClosedEvent in project mongo-java-driver by mongodb.

the class AbstractConnectionPoolTest method shouldPassAllOutcomes.

@Test
public void shouldPassAllOutcomes() throws Exception {
    try {
        for (BsonValue cur : definition.getArray("operations")) {
            final BsonDocument operation = cur.asDocument();
            String name = operation.getString("name").getValue();
            if (name.equals("start")) {
                String target = operation.getString("target", new BsonString("")).getValue();
                executorServiceMap.put(target, Executors.newSingleThreadExecutor(r -> {
                    Thread result = Executors.defaultThreadFactory().newThread(r);
                    result.setName(target);
                    return result;
                }));
            } else if (name.equals("wait")) {
                Thread.sleep(operation.getNumber("ms").intValue());
            } else if (name.equals("waitForThread")) {
                String target = operation.getString("target", new BsonString("")).getValue();
                Exception exceptionFromFuture = futureMap.remove(target).get(5, TimeUnit.SECONDS);
                if (exceptionFromFuture != null) {
                    throw exceptionFromFuture;
                }
            } else if (name.equals("waitForEvent")) {
                Class<?> eventClass = getEventClass(operation.getString("event").getValue());
                assumeNotNull(eventClass);
                long timeoutMillis = operation.getNumber("timeout", new BsonInt64(TimeUnit.SECONDS.toMillis(5))).longValue();
                listener.waitForEvent(eventClass, operation.getNumber("count").intValue(), timeoutMillis, TimeUnit.MILLISECONDS);
            } else if (name.equals("clear")) {
                pool.invalidate(null);
            } else if (name.equals("ready")) {
                pool.ready();
            } else if (name.equals("close")) {
                pool.close();
            } else if (name.equals("checkOut") || name.equals("checkIn")) {
                Callable<Exception> callable = createCallable(operation);
                if (operation.containsKey("thread")) {
                    String threadTarget = operation.getString("thread").getValue();
                    ExecutorService executorService = executorServiceMap.get(threadTarget);
                    futureMap.put(threadTarget, executorService.submit(callable));
                } else {
                    callable.call();
                }
            } else {
                throw new UnsupportedOperationException("No support for " + name);
            }
        }
    } catch (Exception e) {
        if (!definition.containsKey("error")) {
            throw e;
        }
        BsonDocument errorDocument = definition.getDocument("error");
        String exceptionType = errorDocument.getString("type").getValue();
        if (exceptionType.equals("PoolClosedError")) {
            assertEquals(IllegalStateException.class, e.getClass());
        } else if (exceptionType.equals("WaitQueueTimeoutError")) {
            if (e.getClass() != MongoTimeoutException.class) {
                throw e;
            }
        } else {
            throw e;
        }
    }
    if (definition.containsKey("events")) {
        Iterator<Object> actualEventsIterator = getNonIgnoredActualEvents().iterator();
        BsonArray expectedEvents = definition.getArray("events");
        for (BsonValue cur : expectedEvents) {
            BsonDocument expectedEvent = cur.asDocument();
            String type = expectedEvent.getString("type").getValue();
            if (type.equals("ConnectionPoolCreated")) {
                ConnectionPoolCreatedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionPoolCreatedEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getServerId().getAddress());
                assertEquals(settings, actualEvent.getSettings());
            } else if (type.equals("ConnectionPoolCleared")) {
                ConnectionPoolClearedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionPoolClearedEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getServerId().getAddress());
            } else if (type.equals("ConnectionPoolReady")) {
                ConnectionPoolReadyEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionPoolReadyEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getServerId().getAddress());
            } else if (type.equals("ConnectionPoolClosed")) {
                ConnectionPoolClosedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionPoolClosedEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getServerId().getAddress());
            } else if (type.equals("ConnectionCreated")) {
                ConnectionCreatedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionCreatedEvent.class);
                assertConnectionIdMatch(expectedEvent, actualEvent.getConnectionId());
            } else if (type.equals("ConnectionReady")) {
                ConnectionReadyEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionReadyEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getConnectionId().getServerId().getAddress());
            } else if (type.equals("ConnectionClosed")) {
                ConnectionClosedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionClosedEvent.class);
                assertConnectionIdMatch(expectedEvent, actualEvent.getConnectionId());
                assertReasonMatch(expectedEvent, actualEvent);
            } else if (type.equals("ConnectionCheckOutStarted")) {
                ConnectionCheckOutStartedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionCheckOutStartedEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getServerId().getAddress());
            } else if (type.equals("ConnectionCheckOutFailed")) {
                ConnectionCheckOutFailedEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionCheckOutFailedEvent.class);
                assertAddressMatch(expectedEvent, actualEvent.getServerId().getAddress());
                assertReasonMatch(expectedEvent, actualEvent);
            } else if (type.equals("ConnectionCheckedOut")) {
                ConnectionCheckedOutEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionCheckedOutEvent.class);
                assertConnectionIdMatch(expectedEvent, actualEvent.getConnectionId());
            } else if (type.equals("ConnectionCheckedIn")) {
                ConnectionCheckedInEvent actualEvent = getNextEvent(actualEventsIterator, ConnectionCheckedInEvent.class);
                assertConnectionIdMatch(expectedEvent, actualEvent.getConnectionId());
            } else {
                throw new UnsupportedOperationException("Unsupported event type " + type);
            }
        }
    }
}
Also used : URISyntaxException(java.net.URISyntaxException) JsonTestServerVersionChecker(com.mongodb.JsonTestServerVersionChecker) ConnectionPoolCreatedEvent(com.mongodb.event.ConnectionPoolCreatedEvent) BsonValue(org.bson.BsonValue) ClusterId(com.mongodb.connection.ClusterId) Future(java.util.concurrent.Future) ServerListener(com.mongodb.event.ServerListener) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) SocketSettings(com.mongodb.connection.SocketSettings) BsonArray(org.bson.BsonArray) After(org.junit.After) Map(java.util.Map) Assert.fail(org.junit.Assert.fail) ConnectionCheckedOutEvent(com.mongodb.event.ConnectionCheckedOutEvent) Parameterized(org.junit.runners.Parameterized) StreamFactory(com.mongodb.connection.StreamFactory) Collection(java.util.Collection) Set(java.util.Set) Assertions.assertFalse(com.mongodb.assertions.Assertions.assertFalse) ConnectionPoolClearedEvent(com.mongodb.event.ConnectionPoolClearedEvent) Executors(java.util.concurrent.Executors) String.format(java.lang.String.format) SingleResultCallback(com.mongodb.internal.async.SingleResultCallback) List(java.util.List) ConnectionClosedEvent(com.mongodb.event.ConnectionClosedEvent) Assume.assumeNotNull(org.junit.Assume.assumeNotNull) Mockito.mock(org.mockito.Mockito.mock) ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) ConnectionPoolSettings(com.mongodb.connection.ConnectionPoolSettings) JsonPoweredTestHelper(util.JsonPoweredTestHelper) ConnectionCheckedInEvent(com.mongodb.event.ConnectionCheckedInEvent) Assume.assumeFalse(org.junit.Assume.assumeFalse) RunWith(org.junit.runner.RunWith) HashMap(java.util.HashMap) Callable(java.util.concurrent.Callable) SameObjectProvider(com.mongodb.internal.inject.SameObjectProvider) BsonString(org.bson.BsonString) BsonDocument(org.bson.BsonDocument) ArrayList(java.util.ArrayList) HashSet(java.util.HashSet) ClusterFixture(com.mongodb.ClusterFixture) ClusterConnectionMode(com.mongodb.connection.ClusterConnectionMode) BsonInt64(org.bson.BsonInt64) ConnectionCreatedEvent(com.mongodb.event.ConnectionCreatedEvent) MongoDriverInformation(com.mongodb.MongoDriverInformation) ExecutorService(java.util.concurrent.ExecutorService) Before(org.junit.Before) ServerAddress(com.mongodb.ServerAddress) SslSettings(com.mongodb.connection.SslSettings) ConnectionReadyEvent(com.mongodb.event.ConnectionReadyEvent) MongoInterruptedException(com.mongodb.MongoInterruptedException) Iterator(java.util.Iterator) ServerId(com.mongodb.connection.ServerId) CommandReadOperation(com.mongodb.internal.operation.CommandReadOperation) Test(org.junit.Test) IOException(java.io.IOException) BsonDocumentCodec(org.bson.codecs.BsonDocumentCodec) File(java.io.File) MongoTimeoutException(com.mongodb.MongoTimeoutException) TimeUnit(java.util.concurrent.TimeUnit) ConnectionPoolClosedEvent(com.mongodb.event.ConnectionPoolClosedEvent) Mockito(org.mockito.Mockito) ConnectionPoolReadyEvent(com.mongodb.event.ConnectionPoolReadyEvent) ConnectionCheckOutStartedEvent(com.mongodb.event.ConnectionCheckOutStartedEvent) ObjectId(org.bson.types.ObjectId) Nullable(com.mongodb.lang.Nullable) ConnectionId(com.mongodb.connection.ConnectionId) ConnectionCheckOutFailedEvent(com.mongodb.event.ConnectionCheckOutFailedEvent) Collections(java.util.Collections) Assert.assertEquals(org.junit.Assert.assertEquals) ConnectionClosedEvent(com.mongodb.event.ConnectionClosedEvent) ConnectionReadyEvent(com.mongodb.event.ConnectionReadyEvent) BsonString(org.bson.BsonString) ConnectionCheckOutStartedEvent(com.mongodb.event.ConnectionCheckOutStartedEvent) Callable(java.util.concurrent.Callable) ConnectionCheckedOutEvent(com.mongodb.event.ConnectionCheckedOutEvent) ConnectionPoolClearedEvent(com.mongodb.event.ConnectionPoolClearedEvent) ConnectionCreatedEvent(com.mongodb.event.ConnectionCreatedEvent) ConnectionCheckedInEvent(com.mongodb.event.ConnectionCheckedInEvent) ConnectionPoolReadyEvent(com.mongodb.event.ConnectionPoolReadyEvent) ConnectionCheckOutFailedEvent(com.mongodb.event.ConnectionCheckOutFailedEvent) URISyntaxException(java.net.URISyntaxException) MongoInterruptedException(com.mongodb.MongoInterruptedException) IOException(java.io.IOException) MongoTimeoutException(com.mongodb.MongoTimeoutException) BsonInt64(org.bson.BsonInt64) BsonDocument(org.bson.BsonDocument) ConnectionPoolCreatedEvent(com.mongodb.event.ConnectionPoolCreatedEvent) BsonString(org.bson.BsonString) BsonArray(org.bson.BsonArray) ExecutorService(java.util.concurrent.ExecutorService) ConnectionPoolClosedEvent(com.mongodb.event.ConnectionPoolClosedEvent) BsonValue(org.bson.BsonValue) Test(org.junit.Test)

Aggregations

ConnectionPoolClosedEvent (com.mongodb.event.ConnectionPoolClosedEvent)3 ClusterFixture (com.mongodb.ClusterFixture)1 JsonTestServerVersionChecker (com.mongodb.JsonTestServerVersionChecker)1 MongoDriverInformation (com.mongodb.MongoDriverInformation)1 MongoInterruptedException (com.mongodb.MongoInterruptedException)1 MongoTimeoutException (com.mongodb.MongoTimeoutException)1 ServerAddress (com.mongodb.ServerAddress)1 Assertions.assertFalse (com.mongodb.assertions.Assertions.assertFalse)1 ClusterConnectionMode (com.mongodb.connection.ClusterConnectionMode)1 ClusterId (com.mongodb.connection.ClusterId)1 ConnectionId (com.mongodb.connection.ConnectionId)1 ConnectionPoolSettings (com.mongodb.connection.ConnectionPoolSettings)1 ServerId (com.mongodb.connection.ServerId)1 SocketSettings (com.mongodb.connection.SocketSettings)1 SslSettings (com.mongodb.connection.SslSettings)1 StreamFactory (com.mongodb.connection.StreamFactory)1 ConnectionCheckOutFailedEvent (com.mongodb.event.ConnectionCheckOutFailedEvent)1 ConnectionCheckOutStartedEvent (com.mongodb.event.ConnectionCheckOutStartedEvent)1 ConnectionCheckedInEvent (com.mongodb.event.ConnectionCheckedInEvent)1 ConnectionCheckedOutEvent (com.mongodb.event.ConnectionCheckedOutEvent)1