Java tutorial
/* * Copyright 2006-2007 the original author or authors. * * 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 com.khs.test.jdbc.datasource; import java.io.IOException; import java.util.List; import javax.sql.DataSource; import org.apache.commons.io.IOUtils; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.springframework.beans.factory.BeanInitializationException; import org.springframework.beans.factory.DisposableBean; import org.springframework.beans.factory.InitializingBean; import org.springframework.core.io.Resource; import org.springframework.dao.DataAccessException; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.jdbc.datasource.DataSourceTransactionManager; import org.springframework.transaction.TransactionStatus; import org.springframework.transaction.support.TransactionCallback; import org.springframework.transaction.support.TransactionTemplate; import org.springframework.util.Assert; import org.springframework.util.StringUtils; public class DSInitializer implements InitializingBean, DisposableBean { private static final Log logger = LogFactory.getLog(DSInitializer.class); private Resource[] initScripts; private Resource[] destroyScripts; private DataSource dataSource; private boolean ignoreFailedDrop = true; private static boolean initialized = false; /** * @throws Throwable * @see java.lang.Object#finalize() */ protected void finalize() throws Throwable { super.finalize(); initialized = false; logger.debug("finalize called"); } public void destroy() { if (destroyScripts == null) return; for (int i = 0; i < destroyScripts.length; i++) { Resource destroyScript = destroyScripts[i]; try { doExecuteScript(destroyScript); } catch (Exception e) { if (logger.isDebugEnabled()) { logger.warn("Could not execute destroy script [" + destroyScript + "]", e); } else { logger.warn("Could not execute destroy script [" + destroyScript + "]"); } } } } public void afterPropertiesSet() throws Exception { Assert.notNull(dataSource); initialize(); } private void initialize() { if (!initialized) { destroy(); if (initScripts != null) { for (int i = 0; i < initScripts.length; i++) { Resource initScript = initScripts[i]; logger.debug("Initializing with script: " + initScript); doExecuteScript(initScript); } } initialized = true; } } private void doExecuteScript(final Resource scriptResource) { if (scriptResource == null || !scriptResource.exists()) return; TransactionTemplate transactionTemplate = new TransactionTemplate( new DataSourceTransactionManager(dataSource)); transactionTemplate.execute(new TransactionCallback() { @SuppressWarnings("unchecked") public Object doInTransaction(TransactionStatus status) { JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource); String[] scripts; try { scripts = StringUtils.delimitedListToStringArray( stripComments(IOUtils.readLines(scriptResource.getInputStream())), ";"); } catch (IOException e) { throw new BeanInitializationException("Cannot load script from [" + scriptResource + "]", e); } for (int i = 0; i < scripts.length; i++) { String script = scripts[i].trim(); if (StringUtils.hasText(script)) { try { jdbcTemplate.execute(script); } catch (DataAccessException e) { if (ignoreFailedDrop && script.toLowerCase().startsWith("drop")) { logger.debug("DROP script failed (ignoring): " + script); } else { throw e; } } } } return null; } }); } private String stripComments(List<String> list) { StringBuffer buffer = new StringBuffer(); for (String line : list) { if (!line.startsWith("//") && !line.startsWith("--")) { buffer.append(line + "\n"); } } return buffer.toString(); } public void setInitScripts(Resource[] initScripts) { this.initScripts = initScripts; } public void setDestroyScripts(Resource[] destroyScripts) { this.destroyScripts = destroyScripts; } public void setDataSource(DataSource dataSource) { this.dataSource = dataSource; } public void setIgnoreFailedDrop(boolean ignoreFailedDrop) { this.ignoreFailedDrop = ignoreFailedDrop; } }