Subversion Repositories SmartDukaan

package com.spice.profitmandi.dao.config;

import com.mysql.cj.jdbc.AbandonedConnectionCleanupThread;
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.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
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.annotation.PreDestroy;
import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Enumeration;
import java.util.Properties;

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

        private static final Logger LOGGER = LogManager.getLogger(WebDBContextConfigure.class);

        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();

        private Properties loadAllProperties() {
                Properties properties = new Properties();
                try {
                        Resource sharedRes = AppConfig.getSharedResource();
                        if (sharedRes != null && sharedRes.exists()) {
                                properties.load(sharedRes.getInputStream());
                        }
                        properties.load(resource.getInputStream());
                } catch (IOException e) {
                        LOGGER.error("Failed to load properties", e);
                }
                return properties;
        }

        @Bean(name = "dataSource", destroyMethod = "close")
        public DataSource dataSource() {
                Properties properties = loadAllProperties();

                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")));
                config.setLeakDetectionThreshold(30000);

                return new HikariDataSource(config);
        }

        @Bean
        public Properties getHibernateProperties() {
                Properties dbProperties = new Properties();
                Properties properties = loadAllProperties();
                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"));
                dbProperties.put("hibernate.order_inserts", properties.getProperty("hibernate.order_inserts", "true"));
                dbProperties.put("hibernate.order_updates", properties.getProperty("hibernate.order_updates", "true"));
                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(destroyMethod = "close")
        public Mongo mongoClient(SessionFactory sessionFactory) {
                Properties properties = loadAllProperties();
                return new Mongo(properties.getProperty("mongo.host"),
                                properties.getProperty("content.mongo.host"));
        }

        /**
         * Cleanup method to prevent memory leaks on shutdown.
         * Deregisters JDBC drivers and stops MySQL cleanup thread.
         */
        @PreDestroy
        public void cleanup() {
                LOGGER.info("WebDBContextConfigure cleanup started...");

                // Stop MySQL AbandonedConnectionCleanupThread
                try {
                        AbandonedConnectionCleanupThread.checkedShutdown();
                        LOGGER.info("MySQL AbandonedConnectionCleanupThread stopped");
                } catch (Exception e) {
                        LOGGER.error("Error stopping MySQL cleanup thread", e);
                }

                // Deregister JDBC drivers loaded by this webapp's classloader
                ClassLoader cl = Thread.currentThread().getContextClassLoader();
                Enumeration<Driver> drivers = DriverManager.getDrivers();
                while (drivers.hasMoreElements()) {
                        Driver driver = drivers.nextElement();
                        if (driver.getClass().getClassLoader() == cl) {
                                try {
                                        DriverManager.deregisterDriver(driver);
                                        LOGGER.info("Deregistered JDBC driver: {}", driver);
                                } catch (SQLException e) {
                                        LOGGER.error("Error deregistering JDBC driver {}", driver, e);
                                }
                        }
                }

                LOGGER.info("WebDBContextConfigure cleanup completed");
        }
}