com.jn.sqlhelper.datasource.factory.hikaricp.HikariDataSources Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2020 the original author or authors.
 *
 * Licensed under the LGPL, Version 3.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.gnu.org/licenses/lgpl-3.0.html
 *
 * 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.jn.sqlhelper.datasource.factory.hikaricp;

import com.jn.langx.annotation.Nullable;
import com.jn.langx.util.Emptys;
import com.jn.langx.util.Strings;
import com.jn.langx.util.reflect.Reflects;
import com.jn.sqlhelper.common.security.JdbcSecuritys;
import com.jn.sqlhelper.common.transaction.utils.Isolation;
import com.jn.sqlhelper.common.transaction.utils.Transactions;
import com.jn.sqlhelper.datasource.DataSources;
import com.jn.sqlhelper.datasource.config.DataSourceProperties;
import com.jn.sqlhelper.common.security.DriverPropertiesCipher;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;

import javax.sql.DataSource;
import java.util.Properties;

/**
 * 提供基于 hikaricp 的DataSource 构建工具
 * @since 3.4.0
 */
public class HikariDataSources {

    private HikariDataSources() {
    }

    public static DataSource createDataSource(final DataSourceProperties props) {
        return createDataSource(props, JdbcSecuritys.getDefaultDriverPropertiesCipher());
    }

    /**
     * @since 3.4.5
     */
    public static DataSource createDataSource(final DataSourceProperties props, @Nullable DriverPropertiesCipher cipherer) {
        Properties driverProps = props.getDriverProps();
        HikariConfig config = null;
        if (Emptys.isNotEmpty(driverProps)) {
            config = new HikariConfig(driverProps);
        } else {
            config = new HikariConfig();
        }
        config.setDriverClassName(props.getDriverClassName());
        config.setJdbcUrl(props.getUrl());
        String username = props.getUsername();
        if (Strings.isNotBlank(username)) {
            username = DataSources.decrypt(cipherer, username);
            config.setUsername(username);
        }
        String password = props.getPassword();
        if (Strings.isNotBlank(password)) {
            password = DataSources.decrypt(cipherer, password);
            config.setPassword(password);
        }
        config.setPoolName(props.getName());
        config.setCatalog(props.getCatalog());
        Reflects.invokePublicMethod(config, "setSchema", new Class[]{String.class}, new Object[]{props.getSchema()}, true, false);
        config.setLeakDetectionThreshold(props.getLeakDetectionThresholdInMills());
        config.setConnectionTimeout(props.getConnectionTimeoutInMills());
        config.setValidationTimeout(props.getValidationTimeoutInMills());
        config.setConnectionInitSql(props.getValidationQuery());
        config.setConnectionTestQuery(props.getValidationQuery());
        config.setIdleTimeout(props.getIdleTimeoutInMills());
        config.setMaxLifetime(props.getMaxLifetimeInMills());
        config.setMaximumPoolSize(props.getMaxPoolSize());
        config.setMinimumIdle(props.getMinIdle());
        config.setAutoCommit(props.isAutoCommit());
        Isolation isolation = Transactions.getTransactionIsolation(props.getTransactionIsolation());
        if (Transactions.isValidIsolation(isolation)) {
            config.setTransactionIsolation(isolation.getDisplayText());
        }

        config.setReadOnly(props.isReadonly());
        return new HikariDataSource(config);
    }

    public static DataSource createDataSource(final Properties props) {
        return createDataSource(props, JdbcSecuritys.getDefaultDriverPropertiesCipher());
    }

    /**
     * @since 3.4.5
     */
    public static DataSource createDataSource(final Properties properties, DriverPropertiesCipher cipherer) {
        DataSources.decryptUsernamePassword(properties, cipherer);
        return new HikariDataSource(new HikariConfig(properties));
    }


    public static DataSourceProperties toDataSourceProperties(Properties properties) {
        DataSourceProperties dataSourceProperties = new DataSourceProperties();
        dataSourceProperties.setDriverProps(properties);
        return dataSourceProperties;
    }
}