java 自己实现DataSource实现实例

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对象放入到缓存池中，而不是关闭对象。

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