com.mongodb.internal.binding.SingleServerBinding Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongodb-driver-core Show documentation
Show all versions of mongodb-driver-core Show documentation
The Java operations layer for the MongoDB Java Driver.
Third parties can wrap this layer to provide custom higher-level APIs
/*
* 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.binding;
import com.mongodb.ReadPreference;
import com.mongodb.RequestContext;
import com.mongodb.ServerAddress;
import com.mongodb.ServerApi;
import com.mongodb.internal.connection.OperationContext;
import com.mongodb.connection.ServerDescription;
import com.mongodb.internal.connection.Cluster;
import com.mongodb.internal.connection.Connection;
import com.mongodb.internal.connection.NoOpSessionContext;
import com.mongodb.internal.connection.ServerTuple;
import com.mongodb.internal.selector.ServerAddressSelector;
import com.mongodb.internal.session.SessionContext;
import com.mongodb.lang.Nullable;
import static com.mongodb.assertions.Assertions.notNull;
/**
* A simple binding where all connection sources are bound to the server specified in the constructor.
*
* This class is not part of the public API and may be removed or changed at any time
*/
public class SingleServerBinding extends AbstractReferenceCounted implements ReadWriteBinding {
private final Cluster cluster;
private final ServerAddress serverAddress;
@Nullable
private final ServerApi serverApi;
private final RequestContext requestContext;
private final OperationContext operationContext;
/**
* Creates an instance, defaulting to {@link com.mongodb.ReadPreference#primary()} for reads.
* @param cluster a non-null Cluster which will be used to select a server to bind to
* @param serverAddress a non-null address of the server to bind to
* @param serverApi the server API, which may be null
* @param requestContext the request context, which may not be null
*/
public SingleServerBinding(final Cluster cluster, final ServerAddress serverAddress, @Nullable final ServerApi serverApi,
final RequestContext requestContext) {
this.cluster = notNull("cluster", cluster);
this.serverAddress = notNull("serverAddress", serverAddress);
this.serverApi = serverApi;
this.requestContext = notNull("requestContext", requestContext);
operationContext = new OperationContext();
}
@Override
public ConnectionSource getWriteConnectionSource() {
return new SingleServerBindingConnectionSource();
}
@Override
public ReadPreference getReadPreference() {
return ReadPreference.primary();
}
@Override
public ConnectionSource getReadConnectionSource() {
return new SingleServerBindingConnectionSource();
}
@Override
public ConnectionSource getReadConnectionSource(final int minWireVersion, final ReadPreference fallbackReadPreference) {
throw new UnsupportedOperationException();
}
@Override
public SessionContext getSessionContext() {
return NoOpSessionContext.INSTANCE;
}
@Override
@Nullable
public ServerApi getServerApi() {
return serverApi;
}
@Override
public RequestContext getRequestContext() {
return requestContext;
}
@Override
public OperationContext getOperationContext() {
return operationContext;
}
@Override
public SingleServerBinding retain() {
super.retain();
return this;
}
private final class SingleServerBindingConnectionSource extends AbstractReferenceCounted implements ConnectionSource {
private final ServerDescription serverDescription;
private SingleServerBindingConnectionSource() {
SingleServerBinding.this.retain();
ServerTuple serverTuple = cluster.selectServer(new ServerAddressSelector(serverAddress), operationContext);
serverDescription = serverTuple.getServerDescription();
}
@Override
public ServerDescription getServerDescription() {
return serverDescription;
}
@Override
public SessionContext getSessionContext() {
return NoOpSessionContext.INSTANCE;
}
@Override
public OperationContext getOperationContext() {
return operationContext;
}
@Override
public ServerApi getServerApi() {
return serverApi;
}
@Override
public RequestContext getRequestContext() {
return requestContext;
}
@Override
public ReadPreference getReadPreference() {
return ReadPreference.primary();
}
@Override
public Connection getConnection() {
return cluster.selectServer(new ServerAddressSelector(serverAddress), operationContext)
.getServer().getConnection(operationContext);
}
@Override
public ConnectionSource retain() {
super.retain();
return this;
}
@Override
public int release() {
int count = super.release();
if (count == 0) {
SingleServerBinding.this.release();
}
return count;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy