| 
68
 | 
     1 /*
 | 
| 
 | 
     2 Copyright (c) 2008  Franklin Schmidt <fschmidt@gmail.com>
 | 
| 
 | 
     3 
 | 
| 
 | 
     4 Permission is hereby granted, free of charge, to any person obtaining a copy
 | 
| 
 | 
     5 of this software and associated documentation files (the "Software"), to deal
 | 
| 
 | 
     6 in the Software without restriction, including without limitation the rights
 | 
| 
 | 
     7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 | 
| 
 | 
     8 copies of the Software, and to permit persons to whom the Software is
 | 
| 
 | 
     9 furnished to do so, subject to the following conditions:
 | 
| 
 | 
    10 
 | 
| 
 | 
    11 The above copyright notice and this permission notice shall be included in
 | 
| 
 | 
    12 all copies or substantial portions of the Software.
 | 
| 
 | 
    13 
 | 
| 
 | 
    14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | 
| 
 | 
    15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | 
| 
 | 
    16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 | 
| 
 | 
    17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 | 
| 
 | 
    18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 | 
| 
 | 
    19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 | 
| 
 | 
    20 THE SOFTWARE.
 | 
| 
 | 
    21 */
 | 
| 
 | 
    22 
 | 
| 
 | 
    23 package fschmidt.db.pool;
 | 
| 
 | 
    24 
 | 
| 
 | 
    25 import fschmidt.db.DbDatabase;
 | 
| 
 | 
    26 import fschmidt.db.SQLRuntimeException;
 | 
| 
 | 
    27 import fschmidt.db.extend.DbDatabaseExt;
 | 
| 
 | 
    28 import fschmidt.db.extend.DbTransaction;
 | 
| 
 | 
    29 import fschmidt.db.extend.FilterDatabase;
 | 
| 
 | 
    30 import org.slf4j.Logger;
 | 
| 
 | 
    31 import org.slf4j.LoggerFactory;
 | 
| 
 | 
    32 
 | 
| 
 | 
    33 import java.sql.Array;
 | 
| 
 | 
    34 import java.sql.Blob;
 | 
| 
 | 
    35 import java.sql.CallableStatement;
 | 
| 
 | 
    36 import java.sql.Clob;
 | 
| 
 | 
    37 import java.sql.Connection;
 | 
| 
 | 
    38 import java.sql.DatabaseMetaData;
 | 
| 
 | 
    39 import java.sql.NClob;
 | 
| 
 | 
    40 import java.sql.PreparedStatement;
 | 
| 
 | 
    41 import java.sql.SQLClientInfoException;
 | 
| 
 | 
    42 import java.sql.SQLException;
 | 
| 
 | 
    43 import java.sql.SQLWarning;
 | 
| 
 | 
    44 import java.sql.SQLXML;
 | 
| 
 | 
    45 import java.sql.Savepoint;
 | 
| 
 | 
    46 import java.sql.Statement;
 | 
| 
 | 
    47 import java.sql.Struct;
 | 
| 
 | 
    48 import java.util.Map;
 | 
| 
 | 
    49 import java.util.Properties;
 | 
| 
 | 
    50 
 | 
| 
 | 
    51 
 | 
| 
 | 
    52 public class DbDatabaseImpl extends FilterDatabase {
 | 
| 
 | 
    53 	private static final Logger logger = LoggerFactory.getLogger(DbDatabaseImpl.class);
 | 
| 
 | 
    54 
 | 
| 
 | 
    55 	final Pool pool;
 | 
| 
 | 
    56 	final String user;
 | 
| 
 | 
    57 
 | 
| 
 | 
    58 	public DbDatabaseImpl(DbDatabase database,Pool pool,String user) {
 | 
| 
 | 
    59 		super((DbDatabaseExt)database);
 | 
| 
 | 
    60 		this.pool = pool;
 | 
| 
 | 
    61 		this.user = user;
 | 
| 
 | 
    62 	}
 | 
| 
 | 
    63 
 | 
| 
 | 
    64 	public DbDatabaseImpl(DbDatabase database) {
 | 
| 
 | 
    65 		this(database,new Pool(),null);
 | 
| 
 | 
    66 	}
 | 
| 
 | 
    67 
 | 
| 
 | 
    68 	DbDatabase database() {
 | 
| 
 | 
    69 		return database;
 | 
| 
 | 
    70 	}
 | 
| 
 | 
    71 
 | 
| 
 | 
    72 	@Override public boolean isInTransaction() {
 | 
| 
 | 
    73 		return pool.isInTransaction();
 | 
| 
 | 
    74 	}
 | 
| 
 | 
    75 
 | 
| 
 | 
    76 	@Override public void beginTransaction()
 | 
| 
 | 
    77 		throws SQLRuntimeException
 | 
| 
 | 
    78 	{
 | 
| 
 | 
    79 		if( isInTransaction() )
 | 
| 
 | 
    80 			throw new IllegalStateException("beginTransaction called in nested transaction");
 | 
| 
 | 
    81 		try {
 | 
| 
 | 
    82 			Connection con = getConnection();
 | 
| 
 | 
    83 			con.setAutoCommit(false);
 | 
| 
 | 
    84 		} catch(SQLException e) {
 | 
| 
 | 
    85 			throw new SQLRuntimeException(e);
 | 
| 
 | 
    86 		}
 | 
| 
 | 
    87 	}
 | 
| 
 | 
    88 
 | 
| 
 | 
    89 	@Override public void commitTransaction() {
 | 
| 
 | 
    90 		pool.commitTransaction();
 | 
| 
 | 
    91 	}
 | 
| 
 | 
    92 
 | 
| 
 | 
    93 	@Override public void endTransaction() {
 | 
| 
 | 
    94 		pool.endTransaction();
 | 
| 
 | 
    95 	}
 | 
| 
 | 
    96 
 | 
| 
 | 
    97 	@Override public void runBeforeCommit(Runnable r) {
 | 
| 
 | 
    98 		if( isInTransaction() ) {
 | 
| 
 | 
    99 			pool.runBeforeCommit(r);
 | 
| 
 | 
   100 		} else {
 | 
| 
 | 
   101 			super.runBeforeCommit(r);
 | 
| 
 | 
   102 		}
 | 
| 
 | 
   103 	}
 | 
| 
 | 
   104 
 | 
| 
 | 
   105 	@Override public void runJustAfterCommit(Runnable r) {
 | 
| 
 | 
   106 		if( isInTransaction() ) {
 | 
| 
 | 
   107 			pool.runJustAfterCommit(r);
 | 
| 
 | 
   108 		} else {
 | 
| 
 | 
   109 			super.runJustAfterCommit(r);
 | 
| 
 | 
   110 		}
 | 
| 
 | 
   111 	}
 | 
| 
 | 
   112 
 | 
| 
 | 
   113 	@Override public void runAfterCommit(Runnable r) {
 | 
| 
 | 
   114 		if( isInTransaction() ) {
 | 
| 
 | 
   115 			pool.runAfterCommit(r);
 | 
| 
 | 
   116 		} else {
 | 
| 
 | 
   117 			super.runAfterCommit(r);
 | 
| 
 | 
   118 		}
 | 
| 
 | 
   119 	}
 | 
| 
 | 
   120 
 | 
| 
 | 
   121 	@Override public void ignoreRunnablesInThisTransaction() {
 | 
| 
 | 
   122 		pool.ignoreRunnablesInThisTransaction();
 | 
| 
 | 
   123 	}
 | 
| 
 | 
   124 
 | 
| 
 | 
   125 	@Override public DbTransaction getTransaction() {
 | 
| 
 | 
   126 		return pool.getTransaction();
 | 
| 
 | 
   127 	}
 | 
| 
 | 
   128 
 | 
| 
 | 
   129 	@Override public Connection getConnection()
 | 
| 
 | 
   130 		throws SQLException
 | 
| 
 | 
   131 	{
 | 
| 
 | 
   132 		return pool.getConnection(this);
 | 
| 
 | 
   133 	}
 | 
| 
 | 
   134 }
 |