java 自己实现DataSource实现代码

DataSource 对象所表示的物理数据源的连接。作为 DriverManager 工具的替代项。DataSource能提供最高性能的对数据库的并发访问,数据源技术是Java操作数据库的一个很关键技术，流行的持久化框架都离不开数据源的应用。

数据源提供了一种简单获取数据库连接的方式，并能在内部通过一个池的机制来复用数据库连接，这样就大大减少了创建数据库连接的次数，提高了系统性能。下面，我们自己动手实现个精简的数据源，代码如下：

public class MyDataSource { private LinkedList<Connection> connectionPool = new LinkedList<Connection>(); public MyDataSource() { for (int i = 0; i < 10; i++) { connectionPool.add(new MyConnection(creatConnection(),this)); } } private Connection creatConnection() { try { return DriverManager.getConnection( "jdbc:mysql://localhost:3306/test", "root", "root"); } catch (SQLException e) { // TODO Auto-generated catch block throw new ExceptionInInitializerError(); } } public Connection getConnection(){ System.out.println(connectionPool.size()); return connectionPool.removeFirst(); } public void freeConnection(Connection conn){ System.out.println("DataSource Close Connection"); connectionPool.addLast(conn); } }

Java代码

public class MyConnection implements Connection{ private Connection connection; private MyDataSource datasource; public Connection getConnection() { return connection; } public void setConnection(Connection connection) { this.connection = connection; } public MyConnection(Connection realconnection,MyDataSource datasource){ this.connection=realconnection; this.datasource=datasource; } @Override public void close() throws SQLException { // TODO Auto-generated method stub System.out.println("MyConnection Close"); datasource.freeConnection(this); } ... }

我们通过DataSource获得的Connection是经过包裹后的对象，这里应用到了代理模式。下面我们使用jdk的动态代理来改写MyDataSource：

Java代码

public class MyDataSource { private LinkedList<Connection> connectionPool = new LinkedList<Connection>(); public MyDataSource() { for (int i = 0; i < 10; i++) { connectionPool.add(GetProxy(creatConnection())); } } private Connection GetProxy(final Connection connection) { // TODO Auto-generated method stub return (Connection)Proxy.newProxyInstance(this.getClass().getClassLoader(), new Class[]{Connection.class}, new InvocationHandler(){ @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { // TODO Auto-generated method stub Object value; if(method.getName().equalsIgnoreCase("close")){ connectionPool.addLast((Connection)proxy); System.out.println(connectionPool.size()); return null; }else{ value=method.invoke(connection, args); } System.out.println(connectionPool.size()); return value; } }); } private Connection creatConnection() { try { return DriverManager.getConnection( "jdbc:mysql://localhost:3306/test", "root", "root"); } catch (SQLException e) { // TODO Auto-generated catch block throw new ExceptionInInitializerError(); } } public Connection getConnection(){ System.out.println(connectionPool.size()); return connectionPool.removeFirst(); } public void freeConnection(Connection conn){ System.out.println("DataSource Close Connection"); connectionPool.addLast(conn); } }

通过这种方式获得的Connection是通过jdk的动态代理生成的一个代理对象，该代理对象实现了Connection接口。

以上两种方式实现了我们自己的DataSource，在我们通过DataSource获得的Connection对象的close方法都已经被改写过了。在Connection对象调用close方法时，会将该Connection对象放入到缓存池中，而不是关闭对象。

感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！