no.skatteetaten.aurora.config.DatabaseConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aurora-oracle-spring-boot-starter Show documentation
Starter for oracle on Aurora Openshift
There is a newer version: 2.1.0
Show newest version
package no.skatteetaten.aurora.config;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.Properties;

import javax.sql.DataSource;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.PropertiesPropertySource;

/**
 * Configure database if the aurora.db property is set.
 * This should be moved to a repo of its own.
 */
@Configuration
@EnableConfigurationProperties(DatabaseConfig.AuroraProperties.class)
@ConditionalOnProperty(prefix = "aurora", value = "db")
public class DatabaseConfig {

    private static final Logger logger = LoggerFactory.getLogger(DatabaseConfig.class);
    private ConfigurableEnvironment env;
    private AuroraProperties auroraProperties;

    public DatabaseConfig(ConfigurableEnvironment env, AuroraProperties auroraProperties) {

        this.env = env;
        this.auroraProperties = auroraProperties;
    }

    @Bean
    @Profile("openshift")
    public DataSource dataSource() throws IOException {

        Properties props = getProperties();
        if (props == null) {
            return null;
        }

        logger.debug("Found database property with url:{}, username:{}, password:{}",
            props.getProperty("jdbc.url"), props.getProperty("jdbc.user"), props.getProperty("jdbc.password").length());

        return DataSourceBuilder.create()
            .url(props.getProperty("jdbc.url"))
            .username(props.getProperty("jdbc.user"))
            .password(props.getProperty("jdbc.password"))
            .build();
    }

    @Bean
    @Profile("openshift")
    public PropertiesPropertySource databaseProperties() throws IOException {

        Properties props = getProperties();
        if (props == null) {
            return null;
        }
        PropertiesPropertySource pps = new PropertiesPropertySource(
            "auroraDatabase[" + auroraProperties.db + "]", props);
        env.getPropertySources().addLast(pps);

        return pps;
    }

    private Properties getProperties() throws IOException {
        String envName = String.format("%s_DB_PROPERTIES", auroraProperties.db).toUpperCase();
        String databasePath = System.getenv(envName);

        if (databasePath == null) {
            logger.debug("The environment variable {} is not set.", envName);
            return null;
        }

        Properties props = new Properties();
        try (FileInputStream input = new FileInputStream(databasePath)) {
            props.load(input);
        }
        return props;
    }

    @ConfigurationProperties("aurora")
    public static class AuroraProperties {
        private String db;

        public String getDb() {
            return db;
        }

        public void setDb(String db) {
            this.db = db;
        }
    }

}