com.wms.multitenant.config.MasterDatabaseConfig.java Source code

Introduction

Here is the source code for com.wms.multitenant.config.MasterDatabaseConfig.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.wms.multitenant.config;

import org.springframework.context.annotation.PropertySource;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.orm.jpa.JpaTransactionManager;
import org.springframework.orm.jpa.JpaVendorAdapter;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import org.hibernate.jpa.HibernatePersistenceProvider;

import javax.persistence.EntityManagerFactory;
import javax.sql.DataSource;

import java.util.Properties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.env.Environment;

/**
 *
 * @author Mazen
 * @company WeMake{}Stuff
 */
@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = "com.wms.multitenant.repository.master", entityManagerFactoryRef = "masterEntityManager", transactionManagerRef = "masterTransactionManager")
@PropertySource("classpath:application.properties")
public class MasterDatabaseConfig {

    @Autowired
    private Environment springEnvironment;

    @Bean
    public DataSource dataSource() {
        DriverManagerDataSource dataSource = new DriverManagerDataSource();
        dataSource.setDriverClassName(
                springEnvironment.getProperty("master.datasource.classname", "com.mysql.jdbc.Driver"));
        dataSource
                .setUrl(springEnvironment.getProperty("master.datasource.url", "jdbc:mysql://localhost:3306/master")
                        + "?createDatabaseIfNotExist=true");
        dataSource.setUsername(springEnvironment.getProperty("master.datasource.user", "root"));
        dataSource.setPassword(springEnvironment.getProperty("master.datasource.password", "root"));
        return dataSource;
    }

    @Bean(name = "masterEntityManager")
    public LocalContainerEntityManagerFactoryBean masterEntityManagerFactory() {
        LocalContainerEntityManagerFactoryBean entityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean();
        entityManagerFactoryBean.setDataSource(dataSource());
        entityManagerFactoryBean.setPersistenceProviderClass(HibernatePersistenceProvider.class);
        JpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
        entityManagerFactoryBean.setJpaVendorAdapter(vendorAdapter);
        entityManagerFactoryBean.setPackagesToScan(new String[] { "com.wms.multitenant.model.master" });
        entityManagerFactoryBean.setJpaProperties(getHibernateProperties());
        entityManagerFactoryBean.setPersistenceUnitName("master");
        return entityManagerFactoryBean;
    }

    private Properties getHibernateProperties() {
        Properties properties = new Properties();
        properties.put("hibernate.dialect",
                springEnvironment.getProperty("hibernate.dialect", "org.hibernate.dialect.MySQLDialect"));
        properties.put("hibernate.show_sql", springEnvironment.getProperty("hibernate.show_sql", "true"));
        properties.put("hibernate.format_sql", springEnvironment.getProperty("hibernate.format_sql", "true"));
        properties.put("hibernate.hbm2ddl.auto", springEnvironment.getProperty("hibernate.hbm2ddl.auto", "update"));
        return properties;
    }

    @Bean(name = "masterTransactionManager")
    public JpaTransactionManager transactionManager(EntityManagerFactory masterEntityManager) {
        JpaTransactionManager transactionManager = new JpaTransactionManager();
        transactionManager.setEntityManagerFactory(masterEntityManager);
        return transactionManager;
    }
}