src/java/de/uapcore/lightpit/DatabaseFacade.java

Sun, 08 Apr 2018 14:40:57 +0200

author
Mike Becker <universe@uap-core.de>
date
Sun, 08 Apr 2018 14:40:57 +0200
changeset 24
8137ec335416
parent 20
bd1a76c91d5b
permissions
-rw-r--r--

updates copyright header

universe@16 1 /*
universe@16 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@16 3 *
universe@24 4 * Copyright 2018 Mike Becker. All rights reserved.
universe@16 5 *
universe@16 6 * Redistribution and use in source and binary forms, with or without
universe@16 7 * modification, are permitted provided that the following conditions are met:
universe@16 8 *
universe@16 9 * 1. Redistributions of source code must retain the above copyright
universe@16 10 * notice, this list of conditions and the following disclaimer.
universe@16 11 *
universe@16 12 * 2. Redistributions in binary form must reproduce the above copyright
universe@16 13 * notice, this list of conditions and the following disclaimer in the
universe@16 14 * documentation and/or other materials provided with the distribution.
universe@16 15 *
universe@16 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@16 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@16 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@16 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@16 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@16 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@16 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@16 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@16 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@16 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@16 26 * POSSIBILITY OF SUCH DAMAGE.
universe@16 27 *
universe@16 28 */
universe@16 29 package de.uapcore.lightpit;
universe@16 30
universe@16 31 import java.sql.Connection;
universe@16 32 import java.sql.DatabaseMetaData;
universe@16 33 import java.sql.SQLException;
universe@16 34 import java.util.Optional;
universe@16 35 import javax.naming.Context;
universe@16 36 import javax.naming.InitialContext;
universe@16 37 import javax.naming.NamingException;
universe@16 38 import javax.servlet.ServletContext;
universe@16 39 import javax.servlet.ServletContextEvent;
universe@16 40 import javax.servlet.ServletContextListener;
universe@16 41 import javax.servlet.annotation.WebListener;
universe@16 42 import javax.sql.DataSource;
universe@16 43 import org.slf4j.Logger;
universe@16 44 import org.slf4j.LoggerFactory;
universe@16 45
universe@16 46 /**
universe@16 47 * Provides access to different privilege layers within the database.
universe@16 48 */
universe@16 49 @WebListener
universe@16 50 public final class DatabaseFacade implements ServletContextListener {
universe@16 51
universe@16 52 private static final Logger LOG = LoggerFactory.getLogger(DatabaseFacade.class);
universe@16 53
universe@16 54 /**
universe@16 55 * Timeout in seconds for the validation test.
universe@16 56 */
universe@16 57 private static final int DB_TEST_TIMEOUT = 10;
universe@16 58
universe@19 59 public static enum Dialect {
universe@19 60 Postgres;
universe@19 61 }
universe@19 62
universe@19 63 /**
universe@19 64 * The database dialect to use.
universe@19 65 *
universe@19 66 * May be override by context parameter.
universe@19 67 *
universe@19 68 * @see Constants#CTX_ATTR_DB_DIALECT
universe@19 69 */
universe@19 70 private Dialect dialect = Dialect.Postgres;
universe@19 71
universe@16 72 /**
universe@16 73 * The default schema to test against when validating the connection.
universe@16 74 *
universe@16 75 * May be overridden by context parameter.
universe@19 76 *
universe@19 77 * @see Constants#CTX_ATTR_DB_SCHEMA
universe@16 78 */
universe@16 79 private static final String DB_DEFAULT_SCHEMA = "lightpit";
universe@16 80
universe@16 81 /**
universe@19 82 * The attribute name in the Servlet context under which an instance of this class can be found.
universe@16 83 */
universe@16 84 public static final String SC_ATTR_NAME = DatabaseFacade.class.getName();
universe@16 85 private ServletContext sc;
universe@16 86
universe@19 87 private static final String DS_JNDI_NAME = "jdbc/lightpit/app";
universe@19 88 private Optional<DataSource> dataSource;
universe@16 89
universe@16 90 /**
universe@19 91 * Returns the data source.
universe@16 92 *
universe@16 93 * The Optional returned should never be empty. However, if something goes
universe@16 94 * wrong during initialization, the data source might be absent.
universe@16 95 * Hence, users of this data source are forced to check the existence.
universe@16 96 *
universe@19 97 * @return a data source
universe@16 98 */
universe@19 99 public Optional<DataSource> getDataSource() {
universe@19 100 return dataSource;
universe@16 101 }
universe@19 102
universe@19 103 public Dialect getSQLDialect() {
universe@19 104 return dialect;
universe@16 105 }
universe@16 106
universe@16 107 private static void checkConnection(DataSource ds, String testSchema, String errMsg) {
universe@16 108 try (Connection conn = ds.getConnection()) {
universe@16 109 if (!conn.isValid(DB_TEST_TIMEOUT)) {
universe@16 110 throw new SQLException("Validation check failed.");
universe@16 111 }
universe@16 112 if (conn.isReadOnly()) {
universe@16 113 throw new SQLException("Connection is read-only and thus unusable.");
universe@16 114 }
universe@16 115 if (!conn.getSchema().equals(testSchema)) {
universe@16 116 throw new SQLException(String.format("Connection is not configured to use the schema %s.", testSchema));
universe@16 117 }
universe@16 118 DatabaseMetaData metaData = conn.getMetaData();
universe@16 119 LOG.info("Connections as {} to {}/{} ready to go.", metaData.getUserName(), metaData.getURL(), conn.getSchema());
universe@16 120 } catch (SQLException ex) {
universe@16 121 LOG.error(errMsg, ex);
universe@16 122 }
universe@16 123 }
universe@16 124
universe@19 125 private static Optional<DataSource> retrieveDataSource(Context ctx) {
universe@16 126 DataSource ret = null;
universe@16 127 try {
universe@19 128 ret = (DataSource)ctx.lookup(DS_JNDI_NAME);
universe@19 129 LOG.info("Data source retrieved.");
universe@16 130 } catch (NamingException ex) {
universe@19 131 LOG.error("Data source {} not available.", DS_JNDI_NAME);
universe@16 132 LOG.error("Reason for the missing data source: ", ex);
universe@16 133 }
universe@16 134 return Optional.ofNullable(ret);
universe@16 135 }
universe@16 136
universe@16 137 @Override
universe@16 138 public void contextInitialized(ServletContextEvent sce) {
universe@16 139 sc = sce.getServletContext();
universe@16 140
universe@19 141 dataSource = null;
universe@16 142
universe@16 143 final String contextName = Optional
universe@16 144 .ofNullable(sc.getInitParameter(Constants.CTX_ATTR_JNDI_CONTEXT))
universe@16 145 .orElse("java:comp/env");
universe@16 146 final String dbSchema = Optional
universe@16 147 .ofNullable(sc.getInitParameter(Constants.CTX_ATTR_DB_SCHEMA))
universe@16 148 .orElse(DB_DEFAULT_SCHEMA);
universe@19 149 final String dbDialect = sc.getInitParameter(Constants.CTX_ATTR_DB_DIALECT);
universe@19 150 if (dbDialect != null) {
universe@19 151 try {
universe@19 152 dialect = Dialect.valueOf(dbDialect);
universe@19 153 } catch (IllegalArgumentException ex) {
universe@20 154 LOG.error("Unknown or unsupported database dialect {}. Defaulting to {}.", dbDialect, dialect);
universe@19 155 }
universe@19 156 }
universe@16 157
universe@16 158 try {
universe@16 159 LOG.debug("Trying to access JNDI context {}...", contextName);
universe@16 160 Context initialCtx = new InitialContext();
universe@16 161 Context ctx = (Context) initialCtx.lookup(contextName);
universe@16 162
universe@19 163 dataSource = retrieveDataSource(ctx);
universe@16 164
universe@19 165 dataSource.ifPresent((ds) -> checkConnection(ds, dbSchema, "Checking database connection failed"));
universe@16 166 } catch (NamingException | ClassCastException ex) {
universe@16 167 LOG.error("Cannot access JNDI resources.", ex);
universe@16 168 }
universe@16 169
universe@16 170 sc.setAttribute(SC_ATTR_NAME, this);
universe@16 171 LOG.info("Database facade injected into ServletContext.");
universe@16 172 }
universe@16 173
universe@16 174 @Override
universe@16 175 public void contextDestroyed(ServletContextEvent sce) {
universe@19 176 dataSource = null;
universe@16 177 }
universe@16 178 }

mercurial