com.mongodb.client.internal.ClientSessionBinding Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongodb-driver Show documentation
Show all versions of mongodb-driver Show documentation
The MongoDB Driver uber-artifact that combines mongodb-driver-sync and the legacy driver
/*
* 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.client.internal;
import com.mongodb.ReadConcern;
import com.mongodb.ReadPreference;
import com.mongodb.binding.ConnectionSource;
import com.mongodb.binding.ReadWriteBinding;
import com.mongodb.client.ClientSession;
import com.mongodb.connection.Connection;
import com.mongodb.connection.ServerDescription;
import com.mongodb.internal.session.ClientSessionContext;
import com.mongodb.session.SessionContext;
import static org.bson.assertions.Assertions.notNull;
/**
* This class is not part of the public API and may be removed or changed at any time.
*/
public class ClientSessionBinding implements ReadWriteBinding {
private final ReadWriteBinding wrapped;
private final ClientSession session;
private final boolean ownsSession;
private final ClientSessionContext sessionContext;
public ClientSessionBinding(final ClientSession session, final boolean ownsSession, final ReadWriteBinding wrapped) {
this.wrapped = notNull("wrapped", wrapped);
this.ownsSession = ownsSession;
this.session = notNull("session", session);
this.sessionContext = new SyncClientSessionContext(session);
}
@Override
public ReadPreference getReadPreference() {
return wrapped.getReadPreference();
}
@Override
public int getCount() {
return wrapped.getCount();
}
@Override
public ReadWriteBinding retain() {
wrapped.retain();
return this;
}
@Override
public void release() {
wrapped.release();
closeSessionIfCountIsZero();
}
private void closeSessionIfCountIsZero() {
if (getCount() == 0 && ownsSession) {
session.close();
}
}
@Override
public ConnectionSource getReadConnectionSource() {
ConnectionSource readConnectionSource = wrapped.getReadConnectionSource();
return new SessionBindingConnectionSource(readConnectionSource);
}
@Override
public SessionContext getSessionContext() {
return sessionContext;
}
@Override
public ConnectionSource getWriteConnectionSource() {
ConnectionSource writeConnectionSource = wrapped.getWriteConnectionSource();
return new SessionBindingConnectionSource(writeConnectionSource);
}
private class SessionBindingConnectionSource implements ConnectionSource {
private ConnectionSource wrapped;
SessionBindingConnectionSource(final ConnectionSource wrapped) {
this.wrapped = wrapped;
}
@Override
public ServerDescription getServerDescription() {
return wrapped.getServerDescription();
}
@Override
public SessionContext getSessionContext() {
return sessionContext;
}
@Override
public Connection getConnection() {
return wrapped.getConnection();
}
@Override
@SuppressWarnings("checkstyle:methodlength")
public ConnectionSource retain() {
wrapped = wrapped.retain();
return this;
}
@Override
public int getCount() {
return wrapped.getCount();
}
@Override
public void release() {
wrapped.release();
closeSessionIfCountIsZero();
}
}
private final class SyncClientSessionContext extends ClientSessionContext implements SessionContext {
private final ClientSession clientSession;
SyncClientSessionContext(final ClientSession clientSession) {
super(clientSession);
this.clientSession = clientSession;
}
@Override
public boolean isImplicitSession() {
return ownsSession;
}
@Override
public boolean notifyMessageSent() {
return clientSession.notifyMessageSent();
}
@Override
public boolean hasActiveTransaction() {
return clientSession.hasActiveTransaction();
}
@Override
public ReadConcern getReadConcern() {
if (clientSession.hasActiveTransaction()) {
return clientSession.getTransactionOptions().getReadConcern();
} else {
return wrapped.getSessionContext().getReadConcern();
}
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy