Java tutorial
/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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.syncope.buildtools; import java.io.File; import java.sql.Connection; import java.sql.SQLException; import java.sql.Statement; import javax.servlet.ServletContext; import javax.servlet.ServletContextEvent; import javax.servlet.ServletContextListener; import javax.sql.DataSource; import org.h2.tools.Server; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.jdbc.datasource.DataSourceUtils; import org.springframework.web.context.WebApplicationContext; import org.springframework.web.context.support.WebApplicationContextUtils; /** * Utility serlvet context listener managing H2 test server instance (to be used * as external resource). */ public class H2StartStopListener implements ServletContextListener { /** * Logger. */ private static final Logger LOG = LoggerFactory.getLogger(H2StartStopListener.class); private static final String H2_TESTDB = "h2TestDb"; @Override public void contextInitialized(final ServletContextEvent sce) { ServletContext context = sce.getServletContext(); File workDir = (File) sce.getServletContext().getAttribute("javax.servlet.context.tempdir"); try { Server h2TestDb = new Server(); h2TestDb.runTool("-baseDir", workDir.getAbsolutePath(), "-tcp", "-tcpDaemon", "-web", "-webDaemon", "-webPort", sce.getServletContext().getInitParameter("testdb.webport")); context.setAttribute(H2_TESTDB, h2TestDb); } catch (SQLException e) { LOG.error("Could not start H2 test db", e); } WebApplicationContext ctx = WebApplicationContextUtils.getWebApplicationContext(context); DataSource datasource = ctx.getBean(DataSource.class); Connection conn = null; Statement stmt = null; try { conn = DataSourceUtils.getConnection(datasource); stmt = conn.createStatement(); stmt.executeUpdate("RUNSCRIPT FROM 'classpath:/testdb.sql'"); } catch (Exception e) { LOG.error("While loading data into testdb", e); } finally { if (stmt != null) { try { stmt.close(); } catch (SQLException e) { } } if (conn != null) { try { conn.close(); } catch (SQLException e) { } } DataSourceUtils.releaseConnection(conn, datasource); } } @Override public void contextDestroyed(final ServletContextEvent sce) { ServletContext context = sce.getServletContext(); Server h2TestDb = (Server) context.getAttribute(H2_TESTDB); if (h2TestDb != null) { h2TestDb.shutdown(); } } }