package org.geoserver.monitor.hib;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Map;
import java.util.Properties;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.geoserver.config.GeoServerDataDirectory;
import org.geotools.util.logging.Logging;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.orm.jpa.AbstractEntityManagerFactoryBean;
import org.springframework.orm.jpa.vendor.Database;
import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
public class EntityManagerFactoryPostProcessor implements BeanPostProcessor {
static Logger LOGGER = Logging.getLogger("org.geoserver.monitor");
GeoServerDataDirectory data;
public EntityManagerFactoryPostProcessor(GeoServerDataDirectory data) {
this.data = data;
}
public Object postProcessBeforeInitialization(Object bean, String beanName)
throws BeansException {
if (bean instanceof AbstractEntityManagerFactoryBean) {
init((AbstractEntityManagerFactoryBean)bean);
}
return bean;
}
void init(AbstractEntityManagerFactoryBean factory) {
try {
File f = data.findDataFile("monitoring", "hibernate.properties");
if (f == null) {
//copy one out from
Properties props = new Properties();
props.putAll(factory.getJpaVendorAdapter().getJpaPropertyMap());
props.putAll(factory.getJpaPropertyMap());
FileOutputStream fout = new FileOutputStream(
data.findOrCreateDataDir("monitoring", "hibernate.properties"));
props.store(fout, "hibernate configuration");
fout.flush();
fout.close();
}
else {
//use config to overide
Properties props = new Properties();
FileInputStream fin = new FileInputStream(f);
props.load(fin);
fin.close();
HibernateJpaVendorAdapter adapter = (HibernateJpaVendorAdapter) factory.getJpaVendorAdapter();
adapter.setDatabase(Database.valueOf((String)props.get("database")));
adapter.setDatabasePlatform((String)props.get("databasePlatform"));
adapter.setShowSql(Boolean.valueOf((String)props.getProperty("showSql")));
adapter.setGenerateDdl(Boolean.valueOf((String)props.getProperty("generateDdl")));
for (Map.Entry e : props.entrySet()) {
if (((String)e.getKey()).startsWith("hibernate")) {
factory.getJpaPropertyMap().put(e.getKey(), e.getValue());
}
}
}
}
catch (IOException e) {
LOGGER.log(Level.WARNING, e.getLocalizedMessage(), e);
}
}
public Object postProcessAfterInitialization(Object bean, String beanName)
throws BeansException {
if (bean instanceof AbstractEntityManagerFactoryBean) {
init((AbstractEntityManagerFactoryBean)bean);
}
return bean;
}
}