org.wte4j.ui.config.StandaloneJPAConfig.java Source code

Introduction

Here is the source code for org.wte4j.ui.config.StandaloneJPAConfig.java

Source

/**
 * Copyright (C) 2015 Born Informatik AG (www.born.ch)
 *
 * Licensed 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.wte4j.ui.config;

import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;

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

import org.apache.commons.dbcp.BasicDataSource;
import org.hsqldb.persist.HsqlProperties;
import org.hsqldb.server.Server;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.BeanInitializationException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Lazy;
import org.springframework.core.env.Environment;
import org.springframework.jdbc.datasource.lookup.JndiDataSourceLookup;
import org.springframework.orm.jpa.JpaTransactionManager;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
import org.springframework.orm.jpa.vendor.OpenJpaVendorAdapter;
import org.springframework.transaction.PlatformTransactionManager;

@Configuration
public class StandaloneJPAConfig {

    private final Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    private Environment env;

    @Autowired(required = false)
    @Qualifier("wte4j")
    private DataSource externalDataSource;

    private boolean hsqlServerIsRunning = false;

    @Bean
    @Qualifier("wte4j")
    public LocalContainerEntityManagerFactoryBean wteEntityManagerFactory() {
        LocalContainerEntityManagerFactoryBean emfFactoryBean = new LocalContainerEntityManagerFactoryBean();
        emfFactoryBean.setDataSource(lookUpDataSource());
        emfFactoryBean.setJpaVendorAdapter(new OpenJpaVendorAdapter());
        emfFactoryBean.setPersistenceUnitName("wte4j-templates");
        emfFactoryBean.setJpaPropertyMap(createJpaPropertyMap());
        return emfFactoryBean;
    }

    private Map<String, ?> createJpaPropertyMap() {
        Map<String, Object> map = new HashMap<String, Object>();
        map.put("openjpa.jdbc.SynchronizeMappings", "buildSchema(ForeignKeys=true)");
        return map;
    }

    private DataSource lookUpDataSource() {
        if (externalDataSource != null) {
            logger.info("external datasource found with qualifier \"wte4j\" using this DataSource for wte4j");
            return externalDataSource;
        } else {

            return wteInternalDataSource();
        }

    }

    @Bean
    @Lazy
    public DataSource wteInternalDataSource() {
        if (env.containsProperty("wte4j.jdbc.jndi")) {
            JndiDataSourceLookup dsLookup = new JndiDataSourceLookup();
            DataSource dataSource = dsLookup.getDataSource(env.getProperty("wte4j.jdbc.jndi"));
            return dataSource;

        } else if (env.containsProperty("wte4j.jdbc.url")) {
            BasicDataSource dataSource = new BasicDataSource();
            dataSource.setUrl(env.getProperty("wte4j.jdbc.url"));
            dataSource.setDriverClassName(env.getProperty("wte4j.jdbc.driver"));
            dataSource.setUsername(env.getProperty("wte4j.jdbc.user"));
            dataSource.setPassword(env.getProperty("wte4j.jdbc.password"));
            return dataSource;

        } else {
            logger.info("no database configured");
            logger.info("try to start hsql database");
            hsqlServer().start();
            hsqlServerIsRunning = true;
            logger.info("hsql server started");

            BasicDataSource dataSource = new BasicDataSource();
            dataSource.setUrl("jdbc:hsqldb:hsql://localhost/wte4j");
            dataSource.setDriverClassName("org.hsqldb.jdbcDriver");
            dataSource.setUsername("sa");
            return dataSource;
        }

    }

    @Bean
    @Lazy
    public Server hsqlServer() {
        Path path = Paths.get(env.getProperty("java.io.tmpdir"), "hsql", "wte4j");
        String pathAsString = path.toAbsolutePath().toString();
        logger.info("dblocation: {}", pathAsString);

        HsqlProperties p = new HsqlProperties();
        p.setProperty("server.database.0", "file:" + pathAsString);
        p.setProperty("server.dbname.0", "wte4j");
        try {
            Server server = new Server();
            server.setProperties(p);
            return server;
        } catch (Exception e) {
            throw new BeanInitializationException("can not creat hsql server bean", e);
        }
    }

    @Bean
    @Qualifier("wte4j")
    public PlatformTransactionManager wteTransactionManager(@Qualifier("wte4j") EntityManagerFactory emf) {
        JpaTransactionManager transactionManager = new JpaTransactionManager();
        transactionManager.setEntityManagerFactory(emf);
        return transactionManager;
    }

    @PreDestroy
    public void stopHsqlServer() {
        if (hsqlServerIsRunning) {
            hsqlServer().stop();
        }
    }
}