All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.mongodb.internal.connection.LoadBalancedServer Maven / Gradle / Ivy

Go to download

The Java operations layer for the MongoDB Java Driver. Third parties can wrap this layer to provide custom higher-level APIs

There is a newer version: 5.3.0-beta0
Show newest version
/*
 * Copyright 2008-present MongoDB, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.mongodb.internal.connection;

import com.mongodb.MongoCommandException;
import com.mongodb.MongoException;
import com.mongodb.MongoNodeIsRecoveringException;
import com.mongodb.MongoNotPrimaryException;
import com.mongodb.MongoSocketException;
import com.mongodb.MongoSocketReadTimeoutException;
import com.mongodb.annotations.ThreadSafe;
import com.mongodb.connection.ClusterConnectionMode;
import com.mongodb.connection.ServerConnectionState;
import com.mongodb.connection.ServerDescription;
import com.mongodb.connection.ServerId;
import com.mongodb.connection.ServerType;
import com.mongodb.event.ServerClosedEvent;
import com.mongodb.event.ServerDescriptionChangedEvent;
import com.mongodb.event.ServerListener;
import com.mongodb.event.ServerOpeningEvent;
import com.mongodb.internal.VisibleForTesting;
import com.mongodb.internal.async.SingleResultCallback;
import com.mongodb.internal.diagnostics.logging.Logger;
import com.mongodb.internal.diagnostics.logging.Loggers;
import com.mongodb.internal.session.SessionContext;
import com.mongodb.lang.Nullable;
import org.bson.types.ObjectId;

import java.util.concurrent.atomic.AtomicBoolean;

import static com.mongodb.assertions.Assertions.isTrue;
import static com.mongodb.connection.ServerConnectionState.CONNECTING;
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;

/**
 * 

This class is not part of the public API and may be removed or changed at any time

*/ @ThreadSafe public class LoadBalancedServer implements ClusterableServer { private static final Logger LOGGER = Loggers.getLogger("connection"); private final AtomicBoolean closed = new AtomicBoolean(); private final ServerId serverId; private final ConnectionPool connectionPool; private final ConnectionFactory connectionFactory; private final ServerListener serverListener; private final ClusterClock clusterClock; public LoadBalancedServer(final ServerId serverId, final ConnectionPool connectionPool, final ConnectionFactory connectionFactory, final ServerListener serverListener, final ClusterClock clusterClock) { this.serverId = serverId; this.connectionPool = connectionPool; this.connectionFactory = connectionFactory; this.serverListener = serverListener; this.clusterClock = clusterClock; serverListener.serverOpening(new ServerOpeningEvent(serverId)); serverListener.serverDescriptionChanged(new ServerDescriptionChangedEvent(serverId, ServerDescription.builder() .ok(true) .state(ServerConnectionState.CONNECTED) .type(ServerType.LOAD_BALANCER) .address(serverId.getAddress()) .build(), ServerDescription.builder().address(serverId.getAddress()).state(CONNECTING).build())); } @Override public void resetToConnecting() { // no op } @Override public void invalidate() { // no op } private void invalidate(final Throwable t, @Nullable final ObjectId serviceId, final int generation) { if (!isClosed()) { if (t instanceof MongoSocketException && !(t instanceof MongoSocketReadTimeoutException)) { if (serviceId != null) { connectionPool.invalidate(serviceId, generation); } } else if (t instanceof MongoNotPrimaryException || t instanceof MongoNodeIsRecoveringException) { if (SHUTDOWN_CODES.contains(((MongoCommandException) t).getErrorCode())) { if (serviceId != null) { connectionPool.invalidate(serviceId, generation); } } } } } @Override public void close() { if (!closed.getAndSet(true)) { connectionPool.close(); serverListener.serverClosed(new ServerClosedEvent(serverId)); } } @Override public boolean isClosed() { return closed.get(); } @Override public void connect() { // no op } @Override public Connection getConnection(final OperationContext operationContext) { isTrue("open", !isClosed()); return connectionFactory.create(connectionPool.get(operationContext), new LoadBalancedServerProtocolExecutor(), ClusterConnectionMode.LOAD_BALANCED); } @Override public void getConnectionAsync(final OperationContext operationContext, final SingleResultCallback callback) { isTrue("open", !isClosed()); connectionPool.getAsync(operationContext, (result, t) -> { if (t != null) { callback.onResult(null, t); } else { callback.onResult(connectionFactory.createAsync(result, new LoadBalancedServerProtocolExecutor(), ClusterConnectionMode.LOAD_BALANCED), null); } }); } @Override public int operationCount() { return -1; } @VisibleForTesting(otherwise = PRIVATE) ConnectionPool getConnectionPool() { return connectionPool; } private class LoadBalancedServerProtocolExecutor implements ProtocolExecutor { @SuppressWarnings("unchecked") @Override public T execute(final CommandProtocol protocol, final InternalConnection connection, final SessionContext sessionContext) { try { protocol.sessionContext(new ClusterClockAdvancingSessionContext(sessionContext, clusterClock)); return protocol.execute(connection); } catch (MongoWriteConcernWithResponseException e) { return (T) e.getResponse(); } catch (MongoException e) { handleExecutionException(connection, sessionContext, e); throw e; } } @SuppressWarnings("unchecked") @Override public void executeAsync(final CommandProtocol protocol, final InternalConnection connection, final SessionContext sessionContext, final SingleResultCallback callback) { protocol.sessionContext(new ClusterClockAdvancingSessionContext(sessionContext, clusterClock)); protocol.executeAsync(connection, errorHandlingCallback((result, t) -> { if (t != null) { if (t instanceof MongoWriteConcernWithResponseException) { callback.onResult((T) ((MongoWriteConcernWithResponseException) t).getResponse(), null); } else { handleExecutionException(connection, sessionContext, t); callback.onResult(null, t); } } else { callback.onResult(result, null); } }, LOGGER)); } private void handleExecutionException(final InternalConnection connection, final SessionContext sessionContext, final Throwable t) { invalidate(t, connection.getDescription().getServiceId(), connection.getGeneration()); if (t instanceof MongoSocketException && sessionContext.hasSession()) { sessionContext.markSessionDirty(); } } } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy