io.rxmicro.data.sql.r2dbc.postgresql.PostgreSQLConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rxmicro-data-sql-r2dbc-postgresql Show documentation
The module to work with dynamic repositories for interaction with PostgreSQL DB using the Reactive Relational Database Connectivity (R2DBC).
There is a newer version: 0.11
Show newest version
/*
 * Copyright (c) 2020. https://rxmicro.io
 *
 * 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 io.rxmicro.data.sql.r2dbc.postgresql;

import io.r2dbc.spi.Connection;
import io.rxmicro.common.meta.BuilderMethod;
import io.rxmicro.data.sql.SQLPooledDatabaseConfig;

import java.time.Duration;
import java.util.Optional;
import java.util.function.Function;

import static io.rxmicro.common.util.Formats.format;

/**
 * Allows configuring Postgre SQL DB pool options.
 *
 * @author nedis
 * @see PostgreSQLRepository
 * @see PostgreSQLClientFactory
 * @since 0.1
 */
@SuppressWarnings("UnusedReturnValue")
public final class PostgreSQLConfig extends SQLPooledDatabaseConfig {

    /**
     * Default PostgreSQL port.
     */
    public static final int DEFAULT_POSTGRESQL_PORT = 5432;

    private Function connectionDecorator;

    /**
     * Creates a Postgre SQL config instance with default settings.
     */
    public PostgreSQLConfig() {
        setPort(DEFAULT_POSTGRESQL_PORT);
        setUser("rxmicro");
    }

    /**
     * Returns the {@link Optional} connection decorator function.
     *
     * 
     * This features is useful for testing purposes.
     *
     * @return the {@link Optional} connection decorator function.
     */
    public Optional> getConnectionDecorator() {
        return Optional.ofNullable(connectionDecorator);
    }

    /**
     * Sets the connection decorator function.
     *
     * 
     * This features is useful for testing purposes.
     *
     * @param connectionDecorator the connection decorator function
     * @return the reference to this {@link PostgreSQLConfig} instance
     */
    @BuilderMethod
    public PostgreSQLConfig setConnectionDecorator(final Function connectionDecorator) {
        this.connectionDecorator = connectionDecorator;
        return this;
    }

    @Override
    public String getConnectionString() {
        return format("r2dbc:postgresql://?:?/?", getHost(), getPort(), getDatabase());
    }

    @Override
    public PostgreSQLConfig setAcquireRetry(final int acquireRetry) {
        return (PostgreSQLConfig) super.setAcquireRetry(acquireRetry);
    }

    @Override
    public PostgreSQLConfig setInitialSize(final int initialSize) {
        return (PostgreSQLConfig) super.setInitialSize(initialSize);
    }

    @Override
    public PostgreSQLConfig setMaxSize(final int maxSize) {
        return (PostgreSQLConfig) super.setMaxSize(maxSize);
    }

    @Override
    public PostgreSQLConfig setValidationQuery(final String validationQuery) {
        return (PostgreSQLConfig) super.setValidationQuery(validationQuery);
    }

    @Override
    public PostgreSQLConfig setMaxIdleTime(final Duration maxIdleTime) {
        return (PostgreSQLConfig) super.setMaxIdleTime(maxIdleTime);
    }

    @Override
    public PostgreSQLConfig setMaxCreateConnectionTime(final Duration maxCreateConnectionTime) {
        return (PostgreSQLConfig) super.setMaxCreateConnectionTime(maxCreateConnectionTime);
    }

    @Override
    public PostgreSQLConfig setMaxAcquireTime(final Duration maxAcquireTime) {
        return (PostgreSQLConfig) super.setMaxAcquireTime(maxAcquireTime);
    }

    @Override
    public PostgreSQLConfig setMaxLifeTime(final Duration maxLifeTime) {
        return (PostgreSQLConfig) super.setMaxLifeTime(maxLifeTime);
    }

    @Override
    public PostgreSQLConfig setHost(final String host) {
        return (PostgreSQLConfig) super.setHost(host);
    }

    @Override
    public PostgreSQLConfig setPort(final int port) {
        return (PostgreSQLConfig) super.setPort(port);
    }

    @Override
    public PostgreSQLConfig setUser(final String user) {
        return (PostgreSQLConfig) super.setUser(user);
    }

    @Override
    public PostgreSQLConfig setPassword(final CharSequence password) {
        return (PostgreSQLConfig) super.setPassword(password);
    }

    @Override
    public PostgreSQLConfig setDatabase(final String database) {
        return (PostgreSQLConfig) super.setDatabase(database);
    }

    @Override
    public PostgreSQLConfig addOption(final String name, final String value) {
        return (PostgreSQLConfig) super.addOption(name, value);
    }

    @Override
    public PostgreSQLConfig setConnectTimeout(final Duration connectTimeout) {
        return (PostgreSQLConfig) super.setConnectTimeout(connectTimeout);
    }
}