org.neo4j.jdbc.bolt.cache.BoltDriverCached Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-jdbc-bolt Show documentation
Bolt implementation for the Neo4j JDBC Driver
There is a newer version: 5.0.0
package org.neo4j.jdbc.bolt.cache;

import org.neo4j.driver.Driver;
import org.neo4j.driver.Metrics;
import org.neo4j.driver.Session;
import org.neo4j.driver.SessionConfig;
import org.neo4j.driver.async.AsyncSession;
import org.neo4j.driver.reactive.RxSession;
import org.neo4j.driver.types.TypeSystem;

import java.util.concurrent.CompletionStage;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * A driver wrapper to keep connected the instance with the cache
 */
public class BoltDriverCached implements Driver {

    private static final UnsupportedOperationException UNSUPPORTED_OPERATION_EXCEPTION = new UnsupportedOperationException("Async operation are not supported over JDBC");
    private Driver internal;
    private BoltDriverCache cache;
    private BoltDriverCacheKey key;

    private final AtomicInteger sessionCounter = new AtomicInteger(0);

    /**
     *
     * @param internal
     * @param cache
     * @param key
     */
    public BoltDriverCached(Driver internal, BoltDriverCache cache, BoltDriverCacheKey key) {
        this.internal = internal;
        this.cache = cache;
        this.key = key;
    }

    @Override
    public boolean isEncrypted() {
        return internal.isEncrypted();
    }

    @Override
    public Session session() {
        sessionCounter.incrementAndGet();
        return internal.session();
    }

    @Override
    public Session session(SessionConfig sessionConfig) {
        sessionCounter.incrementAndGet();
        return internal.session(sessionConfig);
    }

    @Override
    public RxSession rxSession() {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }

    @Override
    public RxSession rxSession(SessionConfig sessionConfig) {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }

    @Override
    public AsyncSession asyncSession() {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }

    @Override
    public AsyncSession asyncSession(SessionConfig sessionConfig) {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }

    @Override
    public void close() {
        if(sessionCounter.decrementAndGet() <= 0){
            cache.removeFromCache(key);
            internal.close();
        }
    }

    @Override
    public CompletionStage closeAsync() {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }

    @Override
    public Metrics metrics() {
        return internal.metrics();
    }

    @Override
    public boolean isMetricsEnabled() {
        return internal.isMetricsEnabled();
    }

    @Override
    public TypeSystem defaultTypeSystem() {
        return internal.defaultTypeSystem();
    }

    @Override
    public void verifyConnectivity() {
        internal.verifyConnectivity();
    }

    @Override
    public CompletionStage verifyConnectivityAsync() {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }

    @Override
    public boolean supportsMultiDb() {
        return internal.supportsMultiDb();
    }

    @Override
    public CompletionStage supportsMultiDbAsync() {
        throw UNSUPPORTED_OPERATION_EXCEPTION;
    }
}