org.neo4j.driver.internal.bolt.pooledimpl.PooledBoltConnection Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-java-driver Show documentation
Show all versions of neo4j-java-driver Show documentation
Access to the Neo4j graph database through Java
/*
* Copyright (c) "Neo4j"
* Neo4j Sweden AB [https://neo4j.com]
*
* 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 org.neo4j.driver.internal.bolt.pooledimpl;
import java.time.Duration;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.function.Function;
import org.neo4j.driver.Value;
import org.neo4j.driver.exceptions.AuthorizationExpiredException;
import org.neo4j.driver.internal.bolt.api.AccessMode;
import org.neo4j.driver.internal.bolt.api.AuthData;
import org.neo4j.driver.internal.bolt.api.BasicResponseHandler;
import org.neo4j.driver.internal.bolt.api.BoltConnection;
import org.neo4j.driver.internal.bolt.api.BoltConnectionState;
import org.neo4j.driver.internal.bolt.api.BoltProtocolVersion;
import org.neo4j.driver.internal.bolt.api.BoltServerAddress;
import org.neo4j.driver.internal.bolt.api.DatabaseName;
import org.neo4j.driver.internal.bolt.api.NotificationConfig;
import org.neo4j.driver.internal.bolt.api.ResponseHandler;
import org.neo4j.driver.internal.bolt.api.TelemetryApi;
import org.neo4j.driver.internal.bolt.api.TransactionType;
import org.neo4j.driver.internal.bolt.api.summary.BeginSummary;
import org.neo4j.driver.internal.bolt.api.summary.CommitSummary;
import org.neo4j.driver.internal.bolt.api.summary.DiscardSummary;
import org.neo4j.driver.internal.bolt.api.summary.LogoffSummary;
import org.neo4j.driver.internal.bolt.api.summary.LogonSummary;
import org.neo4j.driver.internal.bolt.api.summary.PullSummary;
import org.neo4j.driver.internal.bolt.api.summary.ResetSummary;
import org.neo4j.driver.internal.bolt.api.summary.RollbackSummary;
import org.neo4j.driver.internal.bolt.api.summary.RouteSummary;
import org.neo4j.driver.internal.bolt.api.summary.RunSummary;
import org.neo4j.driver.internal.bolt.api.summary.TelemetrySummary;
public class PooledBoltConnection implements BoltConnection {
private final BoltConnection delegate;
private final PooledBoltConnectionProvider provider;
private final Runnable releaseRunnable;
private final Runnable purgeRunnable;
private CompletableFuture closeFuture;
public PooledBoltConnection(
BoltConnection delegate,
PooledBoltConnectionProvider provider,
Runnable releaseRunnable,
Runnable purgeRunnable) {
this.delegate = Objects.requireNonNull(delegate);
this.provider = Objects.requireNonNull(provider);
this.releaseRunnable = Objects.requireNonNull(releaseRunnable);
this.purgeRunnable = Objects.requireNonNull(purgeRunnable);
}
@Override
public CompletionStage route(
DatabaseName databaseName, String impersonatedUser, Set bookmarks) {
return delegate.route(databaseName, impersonatedUser, bookmarks).thenApply(ignored -> this);
}
@Override
public CompletionStage beginTransaction(
DatabaseName databaseName,
AccessMode accessMode,
String impersonatedUser,
Set bookmarks,
TransactionType transactionType,
Duration txTimeout,
Map txMetadata,
String txType,
NotificationConfig notificationConfig) {
return delegate.beginTransaction(
databaseName,
accessMode,
impersonatedUser,
bookmarks,
transactionType,
txTimeout,
txMetadata,
txType,
notificationConfig)
.thenApply(ignored -> this);
}
@Override
public CompletionStage runInAutoCommitTransaction(
DatabaseName databaseName,
AccessMode accessMode,
String impersonatedUser,
Set bookmarks,
String query,
Map parameters,
Duration txTimeout,
Map txMetadata,
NotificationConfig notificationConfig) {
return delegate.runInAutoCommitTransaction(
databaseName,
accessMode,
impersonatedUser,
bookmarks,
query,
parameters,
txTimeout,
txMetadata,
notificationConfig)
.thenApply(ignored -> this);
}
@Override
public CompletionStage run(String query, Map parameters) {
return delegate.run(query, parameters).thenApply(ignored -> this);
}
@Override
public CompletionStage pull(long qid, long request) {
return delegate.pull(qid, request).thenApply(ignored -> this);
}
@Override
public CompletionStage discard(long qid, long number) {
return delegate.discard(qid, number).thenApply(ignored -> this);
}
@Override
public CompletionStage commit() {
return delegate.commit().thenApply(ignored -> this);
}
@Override
public CompletionStage rollback() {
return delegate.rollback().thenApply(ignored -> this);
}
@Override
public CompletionStage reset() {
return delegate.reset().thenApply(ignored -> this);
}
@Override
public CompletionStage logoff() {
return delegate.logoff().thenApply(ignored -> this);
}
@Override
public CompletionStage logon(Map authMap) {
return delegate.logon(authMap).thenApply(ignored -> this);
}
@Override
public CompletionStage telemetry(TelemetryApi telemetryApi) {
return delegate.telemetry(telemetryApi).thenApply(ignored -> this);
}
@Override
public CompletionStage clear() {
return delegate.clear();
}
@Override
public CompletionStage flush(ResponseHandler handler) {
return delegate.flush(new ResponseHandler() {
@Override
public void onError(Throwable throwable) {
if (throwable instanceof AuthorizationExpiredException) {
provider.onExpired();
}
handler.onError(throwable);
}
@Override
public void onBeginSummary(BeginSummary summary) {
handler.onBeginSummary(summary);
}
@Override
public void onRunSummary(RunSummary summary) {
handler.onRunSummary(summary);
}
@Override
public void onRecord(Value[] fields) {
handler.onRecord(fields);
}
@Override
public void onPullSummary(PullSummary summary) {
handler.onPullSummary(summary);
}
@Override
public void onDiscardSummary(DiscardSummary summary) {
handler.onDiscardSummary(summary);
}
@Override
public void onCommitSummary(CommitSummary summary) {
handler.onCommitSummary(summary);
}
@Override
public void onRollbackSummary(RollbackSummary summary) {
handler.onRollbackSummary(summary);
}
@Override
public void onResetSummary(ResetSummary summary) {
handler.onResetSummary(summary);
}
@Override
public void onRouteSummary(RouteSummary summary) {
handler.onRouteSummary(summary);
}
@Override
public void onLogoffSummary(LogoffSummary summary) {
handler.onLogoffSummary(summary);
}
@Override
public void onLogonSummary(LogonSummary summary) {
handler.onLogonSummary(summary);
}
@Override
public void onTelemetrySummary(TelemetrySummary summary) {
handler.onTelemetrySummary(summary);
}
@Override
public void onIgnored() {
handler.onIgnored();
}
@Override
public void onComplete() {
handler.onComplete();
}
})
.whenComplete((ignored, throwable) -> {
if (throwable != null) {
if (delegate.state() == BoltConnectionState.CLOSED) {
purgeRunnable.run();
}
}
});
}
@Override
public CompletionStage forceClose(String reason) {
return delegate.forceClose(reason).whenComplete((closeResult, closeThrowable) -> purgeRunnable.run());
}
@Override
public CompletionStage close() {
CompletableFuture closeFuture;
var close = false;
synchronized (this) {
if (this.closeFuture == null) {
this.closeFuture = new CompletableFuture<>();
close = true;
}
closeFuture = this.closeFuture;
}
if (close) {
if (delegate.state() == BoltConnectionState.CLOSED) {
purgeRunnable.run();
closeFuture.complete(null);
return closeFuture;
} else if (delegate.state() == BoltConnectionState.ERROR) {
purgeRunnable.run();
closeFuture.complete(null);
return closeFuture;
}
var resetHandler = new BasicResponseHandler();
delegate.reset()
.thenCompose(boltConnection -> boltConnection.flush(resetHandler))
.thenCompose(ignored -> resetHandler.summaries())
.handle((ignored, throwable) -> {
if (throwable != null) {
return delegate()
.close()
.whenComplete((closeResult, closeThrowable) -> purgeRunnable.run());
} else {
return CompletableFuture.completedStage(null)
.whenComplete((ignoredResult, nothing) -> releaseRunnable.run());
}
})
.thenCompose(Function.identity())
.whenComplete((ignored, throwable) -> {
if (throwable != null) {
closeFuture.completeExceptionally(throwable);
} else {
closeFuture.complete(null);
}
});
}
return closeFuture;
}
@Override
public BoltConnectionState state() {
return delegate.state();
}
@Override
public CompletionStage authData() {
return delegate.authData();
}
@Override
public String serverAgent() {
return delegate().serverAgent();
}
@Override
public BoltServerAddress serverAddress() {
return delegate.serverAddress();
}
@Override
public BoltProtocolVersion protocolVersion() {
return delegate.protocolVersion();
}
@Override
public boolean telemetrySupported() {
return delegate.telemetrySupported();
}
// internal use only
BoltConnection delegate() {
return delegate;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy