simpleconnetionpool.java

java编写的数据路连接池 以及数据库连接池的控制方法

package com.mec.dbConnections;

import java.io.PrintStream;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.TimerTask;

public class SimpleConnetionPool {
	// û���õ���l��
	private static LinkedList m_notUsedConnection = new LinkedList();
	// ʹ���е�l��
	private static HashSet m_usedUsedConnection = new HashSet();
	//��ݿ���
	private static String m_driver = "";
	// ��ݿ�l�ӵ�ַ
	private static String m_url = "";
	// ��ݿ��û���
	private static String m_user = "";
	// ��ݿ�����
	private static String m_password = "";
	static final boolean DEBUG = true;
	static private long m_lastClearClosedConnection = System
			.currentTimeMillis();
	//public static   long CHECK_CLOSED_CONNECTION_TIME = 4 * 60 * 60 * 1000; // 4hours
	public static long CHECK_CLOSED_CONNECTION_TIME = 2000; // 4

	// hours

	public static synchronized int getNotUsedConnectionCount() {
		return m_notUsedConnection.size();
	}

	public static synchronized int getUsedConnectionCount() {
		return m_usedUsedConnection.size();
	}

	public static String getUrl() {
		return m_url;
	}

	public static void setUrl(String url) {
		if (url == null) {
			return;
		}
		m_url = url.trim();
	}

	public static String getUser() {
		return m_user;
	}

	public static void setUser(String user) {
		if (user == null) {
			return;
		}
		m_user = user.trim();
	}

	public static String getPassword() {
		return m_password;
	}

	public static void setPassword(String password) {
		if (password == null) {
			return;
		}
		m_password = password.trim();
	}

	/**
	 * ��ʼ����ݿ���l��
	 */
	private static void initDriver() {
		Driver driver = null;
		// load mysql driver
		try {
			driver = (Driver) Class.forName(m_driver)
					.newInstance();
			installDriver(driver);
		} catch (Exception e) {

		}
	}

	/**
	 * �� DriverManager ע��������
	 * 
	 * @param driver
	 *            ������
	 */
	public static void installDriver(Driver driver) {
		try {
			DriverManager.registerDriver(driver);
		} catch (Exception e) {
			// TODO: handle exception
		}
	}

	/**
	 * ��ȡ��ݿ�l��
	 * 
	 * @return ����һ����ݿ�l��
	 */
	public static synchronized Connection getConnection() {

		// ���û��ʹ�õ�l�� > 0
		while (m_notUsedConnection.size() > 0) {
			ConnectionWrapper wrapper = (ConnectionWrapper) m_notUsedConnection
					.removeFirst();
			try {
				if (wrapper.connection.isClosed()) {
					continue;
				}
			} catch (SQLException e) {

			}
			m_usedUsedConnection.add(wrapper);

			if (DEBUG) {
				wrapper.debugInfo = new Throwable(
						"Connection initial statement");
			}
			//System.out.println("连接"+ wrapper.connection.hashCode()+"被使用"); 
			return wrapper.connection;
			
		}

		// ���û��l��
		int newCount = getIncreasingConnectionCount();
		LinkedList list = new LinkedList();
		ConnectionWrapper wrapper = null;
		for (int i = 0; i < newCount; i++) {
			wrapper = getNewConnection();
			if (wrapper != null) {
				list.add(wrapper);
			}
		}
		if (list.size() == 0) {
			return null;
		}
		wrapper = (ConnectionWrapper) list.removeFirst();
		m_usedUsedConnection.add(wrapper);

		m_notUsedConnection.addAll(list);
		list.clear();
		System.out.println("连接"+ wrapper.connection.hashCode()+"被创建");
		return wrapper.connection;
	}

	/**
	 * get increasing connection count, not just add 1 connection
	 * 
	 * @return
	 */

	public static int getIncreasingConnectionCount() {
		int count = 1;
		int current = getConnectionCount();
		count = current / 4;
		if (count < 1) {
			count = 1;
		}
		return count;
	}

	/**
	 * ��ȡ�ܹ���l����
	 * 
	 * @return
	 */
	public static synchronized int getConnectionCount() {
		return m_notUsedConnection.size() + m_usedUsedConnection.size();
	}

	/**
	 * ��ȡһ���µ�l��
	 * 
	 * @return
	 */
	private static ConnectionWrapper getNewConnection() {
		try {
			// Class.forName("com.mysql.jdbc.Driver");

			initDriver();

			Connection con = DriverManager.getConnection(m_url, m_user,
					m_password);
			ConnectionWrapper wrapper = new ConnectionWrapper(con);
			return wrapper;
		} catch (Exception e) {
			//e.printStackTrace();
		}
		return null;
	}

	/**
	 * ��ʹ�õ�l��ת�Ƶ����õ�l��
	 * 
	 * @param con
	 */
	static synchronized void pushConnectionBackToPool(ConnectionWrapper con) {
		boolean exist = m_usedUsedConnection.remove(con);
		if (exist) {
			m_notUsedConnection.addLast(con);
		}
	}

	/**
	 * �ر�����l��
	 * 
	 * @return �������е�l����
	 */
	public static int close() {
		int count = 0;

		Iterator iterator = m_notUsedConnection.iterator();
		while (iterator.hasNext()) {
			try {
				((ConnectionWrapper) iterator.next()).close();
				count++;
			} catch (Exception e) {
			}
		}
		m_notUsedConnection.clear();

		iterator = m_usedUsedConnection.iterator();
		while (iterator.hasNext()) {
			try {
				ConnectionWrapper wrapper = (ConnectionWrapper) iterator.next();
				wrapper.close();
				if (DEBUG) {
					wrapper.debugInfo.printStackTrace();
				}
				count++;
			} catch (Exception e) {
			}
		}
		m_usedUsedConnection.clear();

		return count;
	}

	/**
	 * �������еĹرյ�l��
	 */
	public static  TimerTask clearClosedConnection() {
		long time = System.currentTimeMillis();
		// sometimes user change system time,just return
		if (time < m_lastClearClosedConnection) {
			time = m_lastClearClosedConnection;
			return null;
		}
		// no need check very often
		if (time - m_lastClearClosedConnection < CHECK_CLOSED_CONNECTION_TIME) {
			return null;
		}
		m_lastClearClosedConnection = time;

		// begin check
		Iterator iterator = m_notUsedConnection.iterator();
		while (iterator.hasNext()) {
			ConnectionWrapper wrapper = (ConnectionWrapper) iterator.next();
			try {
				if (wrapper.connection.isClosed()) {
					iterator.remove();
					System.out.println("连接"+ wrapper.connection.hashCode()+"被移除");
				}
			} catch (Exception e) {
				iterator.remove();
				if (DEBUG) {
					System.out
							.println("connection is closed, this connection initial StackTrace");
					wrapper.debugInfo.printStackTrace();
				}
			}
		}

		// make connection pool size smaller if too big
		int decrease = getDecreasingConnectionCount();
		if (m_notUsedConnection.size() < decrease) {
			return null;
		}

		while (decrease-- > 0) {
			ConnectionWrapper wrapper = (ConnectionWrapper) m_notUsedConnection
					.removeFirst();
			try {
				wrapper.connection.close();
				System.out.println("连接"+ wrapper.connection.hashCode()+"被关闭");
			} catch (Exception e) {

			}
		}
		return null;
	}

	/**
	 * * get decreasing connection count, not just remove 1 connection *
	 * 
	 * @return count
	 */
	public static int getDecreasingConnectionCount() {
		int count = 0;
		int current = getConnectionCount();
		if (current < 10) {
			return 0;
		}
		return current / 3;
	}

	/**
	 * �����ô���,��ӡ��Ϣ
	 */
	public synchronized static void printDebugMsg() {
		printDebugMsg(System.out);
	}

	public synchronized static void printDebugMsg(PrintStream out) {
		if (DEBUG == false) {
			return;
		}
		StringBuffer msg = new StringBuffer();
		msg.append("debug message in " + SimpleConnetionPool.class.getName());
		msg.append("\r\n");
		msg.append("total count is connection pool: " + getConnectionCount());
		msg.append("\r\n");
		msg.append("not used connection count: " + getNotUsedConnectionCount());
		msg.append("\r\n");
		msg.append("used connection, count: " + getUsedConnectionCount());
		out.println(msg);
		Iterator iterator = m_usedUsedConnection.iterator();
		while (iterator.hasNext()) {
			ConnectionWrapper wrapper = (ConnectionWrapper) iterator.next();
			wrapper.debugInfo.printStackTrace(out);
		}
		out.println();
	}

	public static String getM_driver() {
		return m_driver;
	}

	public static void setM_driver(String m_driver) {
		SimpleConnetionPool.m_driver = m_driver;
	}
}

class ConnectionWrapper implements InvocationHandler {

	private final static String CLOSE_METHOD_NAME = "close";
	public Connection connection = null;
	private Connection m_originConnection = null;
	public long lastAccessTime = System.currentTimeMillis();
	Throwable debugInfo = new Throwable("Connection initial statement");

	ConnectionWrapper(Connection con) {
		this.connection = (Connection) Proxy.newProxyInstance(con.getClass()
				.getClassLoader(), con.getClass().getInterfaces(), this);
		m_originConnection = con;
	}

	void close() throws SQLException {
		m_originConnection.close();
	}

	public Object invoke(Object proxy, Method m, Object[] args)
			throws Throwable {
		Object obj = null;
		if (CLOSE_METHOD_NAME.equals(m.getName())) {
			SimpleConnetionPool.pushConnectionBackToPool(this);
		} else {
			obj = m.invoke(m_originConnection, args);
		}
		lastAccessTime = System.currentTimeMillis();
		return obj;
	}

}