Search in sources :

Example 1 with ConnectionPoolCreatedEvent

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

the class DefaultConnectionPool method connectionPoolCreated.

/**
 * Send both current and deprecated events in order to preserve backwards compatibility.
 * Must not throw {@link Exception}s.
 */
private void connectionPoolCreated(final ConnectionPoolListener connectionPoolListener, final ServerId serverId, final ConnectionPoolSettings settings) {
    connectionPoolListener.connectionPoolCreated(new ConnectionPoolCreatedEvent(serverId, settings));
    connectionPoolListener.connectionPoolOpened(new com.mongodb.event.ConnectionPoolOpenedEvent(serverId, settings));
}
Also used : ConnectionPoolCreatedEvent(com.mongodb.event.ConnectionPoolCreatedEvent)

Example 2 with ConnectionPoolCreatedEvent

use of com.mongodb.event.ConnectionPoolCreatedEvent 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

ConnectionPoolCreatedEvent (com.mongodb.event.ConnectionPoolCreatedEvent)2 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