packagecom.rocky.pool;importjava.sql.Connection;importjava.sql.SQLException;importjava.util.concurrent.CountDownLatch;importjava.util.concurrent.atomic.AtomicInteger;public classConnectionPoolTest {static ConnectionPool pool = new ConnectionPool(10);//保证所有runner能够同时运行
static CountDownLatch start = new CountDownLatch(1);staticCountDownLatch end ;public static void main(String[] args) throwsException {int threadCount = 20;
end= newCountDownLatch(threadCount);int count = 20;
AtomicInteger got= newAtomicInteger();
AtomicInteger notGot= newAtomicInteger();for(int i=0; i
Thread thread= new Thread(new ConnectionRunner(count, got, notGot), "ConectionRunnerThread"+i);
thread.start();
}
start.countDown();
end.await();
System.out.println("total invoke: "+ (threadCount) *count);
System.out.println("got connection: "+got);
System.out.println("not got connection "+notGot);
}static class ConnectionRunner implementsRunnable{intcount ;
AtomicInteger got;
AtomicInteger notGot;public ConnectionRunner(intcount, AtomicInteger got, AtomicInteger notGot){this.count =count;this.got =got;this.notGot =notGot;
}
@Overridepublic voidrun() {try{
start.await();
}catch(InterruptedException e) {
e.printStackTrace();
}while(count > 0){try{
Connection connection= pool.fetchConnection(1000);if(connection != null){try{
connection.createStatement();
connection.commit();
}finally{
pool.releaseConnection(connection);
got.incrementAndGet();
}
}else{
notGot.incrementAndGet();
}
}catch (InterruptedException |SQLException e) {
e.printStackTrace();
}finally{
count--;
}
}
end.countDown();
}
}
}
扫一扫
1603
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
