com.palantir.atlasdb.keyvalue.dbkvs.impl.ConnectionManagerAwareDbKvs Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of atlasdb-dbkvs Show documentation
Palantir open source project
There is a newer version: 0.1152.0
/*
 * (c) Copyright 2018 Palantir Technologies Inc. All rights reserved.
 *
 * 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.palantir.atlasdb.keyvalue.dbkvs.impl;

import com.palantir.atlasdb.AtlasDbConstants;
import com.palantir.atlasdb.keyvalue.api.KeyValueService;
import com.palantir.atlasdb.keyvalue.dbkvs.DbKeyValueServiceConfig;
import com.palantir.atlasdb.keyvalue.dbkvs.util.SqlConnectionSuppliers;
import com.palantir.atlasdb.keyvalue.impl.ForwardingKeyValueService;
import com.palantir.atlasdb.spi.KeyValueServiceRuntimeConfig;
import com.palantir.atlasdb.spi.LocalConnectionConfig;
import com.palantir.nexus.db.pool.ConnectionManager;
import com.palantir.nexus.db.pool.HikariClientPoolConnectionManagers;
import com.palantir.refreshable.Refreshable;
import java.util.Optional;

// This class should be removed and replaced by DbKvs when InDbTimestampStore depends directly on DbKvs
public final class ConnectionManagerAwareDbKvs extends ForwardingKeyValueService {
    private final DbKeyValueService kvs;
    private final ConnectionManager connManager;
    private final SqlConnectionSupplier sqlConnectionSupplier;

    /**
     * @deprecated This method does not support live reloading the DB password. Use
     * {@link #create(DbKeyValueServiceConfig, Refreshable, boolean)} instead.
     */
    @Deprecated
    public static ConnectionManagerAwareDbKvs create(DbKeyValueServiceConfig config) {
        return create(config, AtlasDbConstants.DEFAULT_INITIALIZE_ASYNC);
    }

    /**
     * @deprecated This method does not support live reloading the DB password. Use
     * {@link #create(DbKeyValueServiceConfig, Refreshable, boolean)} instead.
     */
    @Deprecated
    @SuppressWarnings("InlineMeSuggester")
    public static ConnectionManagerAwareDbKvs create(DbKeyValueServiceConfig config, boolean initializeAsync) {
        return create(config, Refreshable.only(Optional.empty()), initializeAsync);
    }

    public static ConnectionManagerAwareDbKvs create(
            DbKeyValueServiceConfig config,
            Refreshable> runtimeConfig,
            boolean initializeAsync) {
        ConnectionManager connManager;
        if (config.sharedResourcesConfig().isPresent()) {
            LocalConnectionConfig localConnectionConfig =
                    config.sharedResourcesConfig().get().connectionConfig();
            connManager = HikariClientPoolConnectionManagers.createShared(
                    config.connection(),
                    localConnectionConfig.poolSize(),
                    localConnectionConfig.acquireFromSharedPoolTimeoutInSeconds());
        } else {
            connManager = HikariClientPoolConnectionManagers.create(config.connection());
        }

        SqlConnectionSupplier sqlConnSupplier =
                SqlConnectionSuppliers.createSimpleConnectionSupplier(connManager, config, runtimeConfig);
        return new ConnectionManagerAwareDbKvs(
                DbKvs.create(config, sqlConnSupplier, initializeAsync), connManager, sqlConnSupplier);
    }

    private ConnectionManagerAwareDbKvs(
            DbKeyValueService kvs, ConnectionManager connManager, SqlConnectionSupplier sqlConnectionSupplier) {
        this.kvs = kvs;
        this.connManager = connManager;
        this.sqlConnectionSupplier = sqlConnectionSupplier;
    }

    @Override
    public KeyValueService delegate() {
        return kvs;
    }

    public ConnectionManager getConnectionManager() {
        return connManager;
    }

    public SqlConnectionSupplier getSqlConnectionSupplier() {
        return sqlConnectionSupplier;
    }

    public String getTablePrefix() {
        return kvs.getTablePrefix();
    }
}