com.mongodb.internal.connection.UsageTrackingInternalConnection 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.connection;
import com.mongodb.RequestContext;
import com.mongodb.connection.ConnectionDescription;
import com.mongodb.connection.ServerDescription;
import com.mongodb.event.ConnectionCreatedEvent;
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 org.bson.ByteBuf;
import org.bson.codecs.Decoder;
import java.util.List;
import static com.mongodb.assertions.Assertions.assertNotNull;
import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;
/**
* A connection that tracks when it was opened and when it was last used.
*/
class UsageTrackingInternalConnection implements InternalConnection {
private static final Logger LOGGER = Loggers.getLogger("connection");
private volatile long openedAt;
private volatile long lastUsedAt;
private volatile boolean closeSilently;
private final InternalConnection wrapped;
private final DefaultConnectionPool.ServiceStateManager serviceStateManager;
UsageTrackingInternalConnection(final InternalConnection wrapped, final DefaultConnectionPool.ServiceStateManager serviceStateManager) {
this.wrapped = wrapped;
this.serviceStateManager = serviceStateManager;
openedAt = Long.MAX_VALUE;
lastUsedAt = openedAt;
}
@Override
public void open() {
wrapped.open();
openedAt = System.currentTimeMillis();
lastUsedAt = openedAt;
if (getDescription().getServiceId() != null) {
serviceStateManager.addConnection(assertNotNull(getDescription().getServiceId()));
}
}
@Override
public void openAsync(final SingleResultCallback callback) {
wrapped.openAsync((result, t) -> {
if (t != null) {
callback.onResult(null, t);
} else {
openedAt = System.currentTimeMillis();
lastUsedAt = openedAt;
if (getDescription().getServiceId() != null) {
serviceStateManager.addConnection(getDescription().getServiceId());
}
callback.onResult(null, null);
}
});
}
@Override
public void close() {
try {
wrapped.close();
} finally {
if (openedAt != Long.MAX_VALUE && getDescription().getServiceId() != null) {
serviceStateManager.removeConnection(assertNotNull(getDescription().getServiceId()));
}
}
}
@Override
public boolean opened() {
return wrapped.opened();
}
@Override
public boolean isClosed() {
return wrapped.isClosed();
}
@Override
public ByteBuf getBuffer(final int size) {
return wrapped.getBuffer(size);
}
@Override
public void sendMessage(final List byteBuffers, final int lastRequestId) {
wrapped.sendMessage(byteBuffers, lastRequestId);
lastUsedAt = System.currentTimeMillis();
}
@Override
public T sendAndReceive(final CommandMessage message, final Decoder decoder, final SessionContext sessionContext,
final RequestContext requestContext, final OperationContext operationContext) {
T result = wrapped.sendAndReceive(message, decoder, sessionContext, requestContext, operationContext);
lastUsedAt = System.currentTimeMillis();
return result;
}
@Override
public void send(final CommandMessage message, final Decoder decoder, final SessionContext sessionContext) {
wrapped.send(message, decoder, sessionContext);
lastUsedAt = System.currentTimeMillis();
}
@Override
public T receive(final Decoder decoder, final SessionContext sessionContext) {
T result = wrapped.receive(decoder, sessionContext);
lastUsedAt = System.currentTimeMillis();
return result;
}
@Override
public boolean supportsAdditionalTimeout() {
return wrapped.supportsAdditionalTimeout();
}
@Override
public T receive(final Decoder decoder, final SessionContext sessionContext, final int additionalTimeout) {
T result = wrapped.receive(decoder, sessionContext, additionalTimeout);
lastUsedAt = System.currentTimeMillis();
return result;
}
@Override
public boolean hasMoreToCome() {
return wrapped.hasMoreToCome();
}
@Override
public void sendAndReceiveAsync(final CommandMessage message, final Decoder decoder,
final SessionContext sessionContext, final RequestContext requestContext, final OperationContext operationContext,
final SingleResultCallback callback) {
SingleResultCallback errHandlingCallback = errorHandlingCallback((result, t) -> {
lastUsedAt = System.currentTimeMillis();
callback.onResult(result, t);
}, LOGGER);
wrapped.sendAndReceiveAsync(message, decoder, sessionContext, requestContext, operationContext, errHandlingCallback);
}
@Override
public ResponseBuffers receiveMessage(final int responseTo) {
ResponseBuffers responseBuffers = wrapped.receiveMessage(responseTo);
lastUsedAt = System.currentTimeMillis();
return responseBuffers;
}
@Override
public void sendMessageAsync(final List byteBuffers, final int lastRequestId, final SingleResultCallback callback) {
SingleResultCallback errHandlingCallback = errorHandlingCallback((result, t) -> {
lastUsedAt = System.currentTimeMillis();
callback.onResult(result, t);
}, LOGGER);
wrapped.sendMessageAsync(byteBuffers, lastRequestId, errHandlingCallback);
}
@Override
public void receiveMessageAsync(final int responseTo, final SingleResultCallback callback) {
SingleResultCallback errHandlingCallback = errorHandlingCallback((result, t) -> {
lastUsedAt = System.currentTimeMillis();
callback.onResult(result, t);
}, LOGGER);
wrapped.receiveMessageAsync(responseTo, errHandlingCallback);
}
@Override
public ConnectionDescription getDescription() {
return wrapped.getDescription();
}
@Override
public ServerDescription getInitialServerDescription() {
return wrapped.getInitialServerDescription();
}
@Override
public int getGeneration() {
return wrapped.getGeneration();
}
/**
* Returns the time at which this connection was opened, or {@code Long.MAX_VALUE} if it has not yet been opened.
*
* @return the time when this connection was opened, in milliseconds since the epoch.
*/
long getOpenedAt() {
return openedAt;
}
/**
* Returns the time at which this connection was last used, or {@code Long.MAX_VALUE} if it has not yet been used.
*
* @return the time when this connection was last used, in milliseconds since the epoch.
*/
long getLastUsedAt() {
return lastUsedAt;
}
/**
* This method must be used if and only if {@link ConnectionCreatedEvent} was not sent for the connection.
* Must not throw {@link Exception}s.
*
* @see #isCloseSilently()
*/
void setCloseSilently() {
closeSilently = true;
}
/**
* Must not throw {@link Exception}s.
*
* @see #setCloseSilently()
*/
boolean isCloseSilently() {
return closeSilently;
}
}