nifi-registry-core/nifi-registry-framework/src/main/java/org/apache/nifi/registry/db/migration/LegacyDataSourceFactory.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.nifi.registry.db.migration;

import org.apache.commons.lang3.StringUtils;
import org.apache.nifi.registry.properties.NiFiRegistryProperties;
import org.h2.jdbcx.JdbcConnectionPool;

import javax.sql.DataSource;
import java.io.File;

/**
 * NOTE: This DataSource factory was used in the original 0.1.0 release and remains to migrate data from the old database.
 * This class is intentionally not a Spring bean, and will be used manually in the custom Flyway migration.
 */
public class LegacyDataSourceFactory {

    private static final String DB_USERNAME_PASSWORD = "nifireg";
    private static final int MAX_CONNECTIONS = 5;

    // database file name
    private static final String DATABASE_FILE_NAME = "nifi-registry";

    private final NiFiRegistryProperties properties;

    private JdbcConnectionPool connectionPool;

    public LegacyDataSourceFactory(final NiFiRegistryProperties properties) {
        this.properties = properties;
    }

    public DataSource getDataSource() {
        if (connectionPool == null) {
            final String databaseUrl = getDatabaseUrl(properties);
            connectionPool = JdbcConnectionPool.create(databaseUrl, DB_USERNAME_PASSWORD, DB_USERNAME_PASSWORD);
            connectionPool.setMaxConnections(MAX_CONNECTIONS);
        }

        return connectionPool;
    }

    public static String getDatabaseUrl(final NiFiRegistryProperties properties) {
        // locate the repository directory
        final String repositoryDirectoryPath = properties.getLegacyDatabaseDirectory();

        // ensure the repository directory is specified
        if (repositoryDirectoryPath == null) {
            throw new NullPointerException("Database directory must be specified.");
        }

        // create a handle to the repository directory
        final File repositoryDirectory = new File(repositoryDirectoryPath);

        // get a handle to the database file
        final File databaseFile = new File(repositoryDirectory, DATABASE_FILE_NAME);

        // format the database url
        String databaseUrl = "jdbc:h2:" + databaseFile + ";AUTOCOMMIT=OFF;DB_CLOSE_ON_EXIT=FALSE;LOCK_MODE=3";
        String databaseUrlAppend = properties.getLegacyDatabaseUrlAppend();
        if (StringUtils.isNotBlank(databaseUrlAppend)) {
            databaseUrl += databaseUrlAppend;
        }

        return databaseUrl;
    }

}