Subversion Repositories SmartDukaan

package com.spice.profitmandi.dao.config;

import com.spice.profitmandi.dao.repository.dtr.Mongo;
import com.spice.profitmandi.web.config.AppConfig;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.hibernate.SessionFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.Resource;
import org.springframework.orm.hibernate5.HibernateTransactionManager;
import org.springframework.orm.hibernate5.LocalSessionFactoryBuilder;
import org.springframework.transaction.annotation.EnableTransactionManagement;

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

@Configuration
@EnableTransactionManagement
@ComponentScan({ "com.spice.profitmandi.dao.*" })
public class WebDBContextConfigure {

        private static final String HIBERNATE_DRIVER_CLASS = "hibernate.driver.class";
        private static final String HIBERNATE_URL = "hibernate.url";
        private static final String HIBERNATE_USER_NAME = "hibernate.user.name";
        private static final String HIBERNATE_PASSWORD = "hibernate.password";
        private static final String HIKARI_MAX_POOL_SIZE = "hikari.maximumPoolSize";
        private static final String HIKARI_MIN_IDLE = "hikari.minimumIdle";
        private static final String HIKARI_IDLE_TIMEOUT = "hikari.idleTimeout";
        private static final String HIKARI_MAX_LIFETIME = "hikari.maxLifetime";
        private static final String HIKARI_CONNECTION_TIMEOUT = "hikari.connectionTimeout";

        private Resource resource = AppConfig.getResource();

        @Bean(name = "dataSource")
        public DataSource dataSource() {
                Properties properties = new Properties();
                try {
                        properties.load(resource.getInputStream());
                } catch (IOException e) {
                        e.printStackTrace();
                }

                HikariConfig config = new HikariConfig();
                config.setDriverClassName(properties.getProperty(HIBERNATE_DRIVER_CLASS));
                config.setJdbcUrl(properties.getProperty(HIBERNATE_URL));
                config.setUsername(properties.getProperty(HIBERNATE_USER_NAME));
                config.setPassword(properties.getProperty(HIBERNATE_PASSWORD));
                config.setMaximumPoolSize(Integer.parseInt(properties.getProperty(HIKARI_MAX_POOL_SIZE, "20")));
                config.setMinimumIdle(Integer.parseInt(properties.getProperty(HIKARI_MIN_IDLE, "2")));
                config.setIdleTimeout(Long.parseLong(properties.getProperty(HIKARI_IDLE_TIMEOUT, "30000")));
                config.setMaxLifetime(Long.parseLong(properties.getProperty(HIKARI_MAX_LIFETIME, "1800000")));
                config.setConnectionTimeout(Long.parseLong(properties.getProperty(HIKARI_CONNECTION_TIMEOUT, "30000")));

                return new HikariDataSource(config);
        }

        @Bean
        public Properties getHibernateProperties() {
                Properties dbProperties = new Properties();
                Properties properties = new Properties();
                try {
                        properties.load(resource.getInputStream());
                } catch (IOException e) {
                        e.printStackTrace();
                }
                dbProperties.put("hibernate.dialect", properties.getProperty("hibernate.dialect"));
                dbProperties.put("hibernate.show_sql", properties.getProperty("hibernate.show_sql"));
                dbProperties.put("hibernate.format_sql", properties.getProperty("hibernate.format_sql"));
                dbProperties.put("hibernate.jdbc.batch_size", properties.getProperty("hibernate.jdbc.batch_size"));
                return dbProperties;
        }

        @Primary
        @Bean(name = "sessionFactory")
        public SessionFactory getSessionFactory(DataSource dataSource) {
                LocalSessionFactoryBuilder sessionBuilder = new LocalSessionFactoryBuilder(dataSource);
                sessionBuilder.addProperties(getHibernateProperties());
                sessionBuilder.scanPackages("com.spice.profitmandi.dao.*");
                return sessionBuilder.buildSessionFactory();
        }

        @Primary
        @Bean(name = "transactionManager")
        public HibernateTransactionManager getTransactionManager(SessionFactory sessionFactory) {
                return new HibernateTransactionManager(sessionFactory);
        }

        @Bean
        public Mongo mongoClient(SessionFactory sessionFactory) {
                Properties properties = new Properties();
                try {
                        properties.load(resource.getInputStream());
                } catch (IOException e) {
                        e.printStackTrace();
                }
                return new Mongo(properties.getProperty("content.mongo.host"),
                                properties.getProperty("content.mongo.host"));
        }
}