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.apache.openejb.resource.jdbc.dbcp; import org.apache.commons.dbcp.ConnectionFactory; import org.apache.openejb.OpenEJB; import org.apache.openejb.cipher.PasswordCipher; import org.apache.openejb.cipher.PasswordCipherFactory; import org.apache.openejb.loader.SystemInstance; import org.apache.openejb.resource.jdbc.BasicDataSourceUtil; import org.apache.openejb.resource.jdbc.IsolationLevels; import org.apache.openejb.resource.jdbc.plugin.DataSourcePlugin; import org.apache.openejb.resource.jdbc.pool.XADataSourceResource; import javax.sql.DataSource; import java.io.File; import java.sql.SQLException; import java.sql.SQLFeatureNotSupportedException; import java.util.Properties; import java.util.concurrent.locks.ReentrantLock; import java.util.logging.Logger; @SuppressWarnings({ "UnusedDeclaration" }) public class BasicManagedDataSource extends org.apache.commons.dbcp.managed.BasicManagedDataSource { private static final ReentrantLock lock = new ReentrantLock(); private Logger logger; /** * The password codec to be used to retrieve the plain text password from a * ciphered value. * <p/> * <em>The default is no codec.</em>. In other words, it means password is * not ciphered. The {@link org.apache.openejb.cipher.PlainTextPasswordCipher} can also be used. */ private String passwordCipher; private JMXBasicDataSource jmxDs; public BasicManagedDataSource(final String name) { registerAsMbean(name); } @Override protected ConnectionFactory createConnectionFactory() throws SQLException { final String xaDataSource = getXADataSource(); if (xaDataSource != null & getXaDataSourceInstance() == null) { try { try { Thread.currentThread().getContextClassLoader().loadClass(xaDataSource); } catch (final ClassNotFoundException | NoClassDefFoundError cnfe) { setJndiXaDataSource(xaDataSource); } } catch (final Throwable th) { // no-op } } return super.createConnectionFactory(); } private void setJndiXaDataSource(final String xaDataSource) { setXaDataSourceInstance( // proxy cause we don't know if this datasource was created before or not the delegate XADataSourceResource.proxy(getDriverClassLoader() != null ? getDriverClassLoader() : Thread.currentThread().getContextClassLoader(), xaDataSource)); if (getTransactionManager() == null) { setTransactionManager(OpenEJB.getTransactionManager()); } } private void registerAsMbean(final String name) { try { jmxDs = new JMXBasicDataSource(name, this); } catch (final Exception | NoClassDefFoundError e) { jmxDs = null; } } /** * Returns the password codec class name to use to retrieve plain text * password. * * @return the password codec class */ public String getPasswordCipher() { final ReentrantLock l = lock; l.lock(); try { return this.passwordCipher; } finally { l.unlock(); } } /** * <p> * Sets the {@link #passwordCipher}. * </p> * * @param passwordCipher password codec value */ public void setPasswordCipher(final String passwordCipher) { final ReentrantLock l = lock; l.lock(); try { this.passwordCipher = passwordCipher; } finally { l.unlock(); } } public String getUserName() { final ReentrantLock l = lock; l.lock(); try { return super.getUsername(); } finally { l.unlock(); } } public void setUserName(final String string) { final ReentrantLock l = lock; l.lock(); try { super.setUsername(string); } finally { l.unlock(); } } public String getJdbcDriver() { final ReentrantLock l = lock; l.lock(); try { return super.getDriverClassName(); } finally { l.unlock(); } } public void setJdbcDriver(final String string) { final ReentrantLock l = lock; l.lock(); try { super.setDriverClassName(string); } finally { l.unlock(); } } public String getJdbcUrl() { final ReentrantLock l = lock; l.lock(); try { return super.getUrl(); } finally { l.unlock(); } } public void setJdbcUrl(final String string) { final ReentrantLock l = lock; l.lock(); try { super.setUrl(string); } finally { l.unlock(); } } public void setDefaultTransactionIsolation(final String s) { final ReentrantLock l = lock; l.lock(); try { if (s == null || s.equals("")) { return; } final int level = IsolationLevels.getIsolationLevel(s); super.setDefaultTransactionIsolation(level); } finally { l.unlock(); } } public void setMaxWait(final int maxWait) { final ReentrantLock l = lock; l.lock(); try { super.setMaxWait((long) maxWait); } finally { l.unlock(); } } protected DataSource createDataSource() throws SQLException { final ReentrantLock l = lock; l.lock(); try { if (super.dataSource != null) { return super.dataSource; } // check password codec if available if (null != passwordCipher) { final PasswordCipher cipher = PasswordCipherFactory.getPasswordCipher(passwordCipher); final String plainPwd = cipher.decrypt(password.toCharArray()); // override previous password value super.setPassword(plainPwd); } // get the plugin final DataSourcePlugin helper = BasicDataSourceUtil.getDataSourcePlugin(getUrl()); // configure this if (helper != null) { final String currentUrl = getUrl(); final String newUrl = helper.updatedUrl(currentUrl); if (!currentUrl.equals(newUrl)) { super.setUrl(newUrl); } } wrapTransactionManager(); // create the data source if (helper == null || !helper.enableUserDirHack()) { try { return super.createDataSource(); } catch (final Throwable e) { throw BasicDataSource.toSQLException(e); } } else { // wrap super call with code that sets user.dir to openejb.base and then resets it final Properties systemProperties = System.getProperties(); final String userDir = systemProperties.getProperty("user.dir"); try { final File base = SystemInstance.get().getBase().getDirectory(); systemProperties.setProperty("user.dir", base.getAbsolutePath()); try { return super.createDataSource(); } catch (final Throwable e) { throw BasicDataSource.toSQLException(e); } } finally { systemProperties.setProperty("user.dir", userDir); } } } finally { l.unlock(); } } protected void wrapTransactionManager() { //TODO? } public void close() throws SQLException { //TODO - Prevent unuathorized call final ReentrantLock l = lock; l.lock(); try { try { unregisterMBean(); } catch (final Exception ignored) { // no-op } super.close(); } finally { l.unlock(); } } private void unregisterMBean() { if (jmxDs != null) { jmxDs.unregister(); } } public Logger getParentLogger() throws SQLFeatureNotSupportedException { final ReentrantLock l = lock; l.lock(); try { if (null == this.logger) { this.logger = (Logger) DataSource.class.getDeclaredMethod("getParentLogger") .invoke(super.dataSource); } return this.logger; } catch (final Throwable e) { throw new SQLFeatureNotSupportedException(); } finally { l.unlock(); } } }