package com.mchange.v2.c3p0.test;
import java.util.*;
import java.sql.*;
import javax.sql.*;
import com.mchange.v2.c3p0.*;
import com.mchange.v1.db.sql.*;
import com.mchange.v2.c3p0.DriverManagerDataSource;
public final class LoadPoolBackedDataSource
{
final static int NUM_THREADS = 100;
final static int ITERATIONS_PER_THREAD = 1000;
static DataSource ds;
public static void main(String[] argv)
{
if (argv.length > 0)
{
System.err.println( LoadPoolBackedDataSource.class.getName() +
" now requires no args. Please set everything in standard c3p0 config files.");
return;
}
String jdbc_url = null;
String username = null;
String password = null;
try
{
DataSource ds_unpooled = DataSources.unpooledDataSource();
ds = DataSources.pooledDataSource( ds_unpooled );
Connection con = null;
Statement stmt = null;
try
{
con = ds.getConnection();
stmt = con.createStatement();
stmt.executeUpdate("CREATE TABLE testpbds ( a varchar(16), b varchar(16) )");
System.err.println( "LoadPoolBackedDataSource -- TEST SCHEMA CREATED" );
}
catch (SQLException e)
{
e.printStackTrace();
System.err.println("relation testpbds already exists, or something " +
"bad happened.");
}
finally
{
StatementUtils.attemptClose( stmt );
ConnectionUtils.attemptClose( con );
}
Thread[] threads = new Thread[NUM_THREADS];
for (int i = 0; i < NUM_THREADS; ++i)
{
Thread t = new ChurnThread(i);
threads[i] = t;
t.start();
System.out.println("THREAD MADE [" + i + "]");
Thread.sleep(500);
}
for (int i = 0; i < NUM_THREADS; ++i)
threads[i].join();
}
catch (Exception e)
{ e.printStackTrace(); }
finally
{
Connection con = null;
Statement stmt = null;
try
{
con = ds.getConnection();
stmt = con.createStatement();
stmt.executeUpdate("DROP TABLE testpbds");
System.err.println( "LoadPoolBackedDataSource -- TEST SCHEMA DROPPED" );
}
catch (Exception e)
{
e.printStackTrace();
}
finally
{
StatementUtils.attemptClose( stmt );
ConnectionUtils.attemptClose( con );
}
}
}
static class ChurnThread extends Thread
{
Random random = new Random();
int num;
public ChurnThread(int num)
{ this.num = num; }
public void run()
{
try
{
for( int i = 0; i < ITERATIONS_PER_THREAD; ++i )
{
Connection con = null;
try
{
con = ds.getConnection();
int select = random.nextInt(3);
switch (select)
{
case 0:
executeSelect( con );
break;
case 1:
executeInsert( con, random );
break;
case 2:
executeDelete( con );
break;
}
PooledDataSource pds = (PooledDataSource) ds;
System.out.println("iteration: (" + num + ", " + i + ')');
System.out.println( pds.getNumConnectionsDefaultUser() );
System.out.println( pds.getNumIdleConnectionsDefaultUser() );
System.out.println( pds.getNumBusyConnectionsDefaultUser() );
System.out.println( pds.getNumConnectionsAllUsers() );
Thread.sleep(1);
}
finally
{ ConnectionUtils.attemptClose( con ); }
}
}
catch (Exception e)
{ e.printStackTrace(); }
}
}
static void executeInsert(Connection con, Random random) throws SQLException
{
Statement stmt = null;
try
{
stmt = con.createStatement();
stmt.executeUpdate("INSERT INTO testpbds VALUES ('" +
random.nextInt() + "', '" +
random.nextInt() + "')");
System.out.println("INSERTION");
}
finally
{
StatementUtils.attemptClose( stmt );
}
}
static void executeDelete(Connection con) throws SQLException
{
Statement stmt = null;
try
{
stmt = con.createStatement();
stmt.executeUpdate("DELETE FROM testpbds;");
System.out.println("DELETION");
}
finally
{
StatementUtils.attemptClose( stmt );
}
}
static void executeSelect(Connection con) throws SQLException
{
long l = System.currentTimeMillis();
Statement stmt = null;
ResultSet rs = null;
try
{
stmt = con.createStatement();
rs = stmt.executeQuery("SELECT count(*) FROM testpbds");
rs.next();
System.out.println("SELECT [count=" + rs.getInt(1) + ", time=" +
(System.currentTimeMillis() - l) + " msecs]");
}
finally
{
ResultSetUtils.attemptClose( rs );
StatementUtils.attemptClose( stmt );
}
}
private static void usage()
{
System.err.println("java " +
"-Djdbc.drivers=<comma_sep_list_of_drivers> " +
LoadPoolBackedDataSource.class.getName() +
" <jdbc_url> [<username> <password>]" );
System.exit(-1);
}
}