自定義一個(gè)簡(jiǎn)單的JDBC連接池實(shí)現(xiàn)方法

更新時(shí)間：2017年12月20日 08:45:31 作者：samluby

下面小編就為大家分享一篇自定義一個(gè)簡(jiǎn)單的JDBC連接池實(shí)現(xiàn)方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

一、什么是JDBC連接池？

在傳統(tǒng)的JDBC連接中，每次獲得一個(gè)Connection連接都需要加載通過(guò)一些繁雜的代碼去獲取，例如以下代碼：

public static Connection getConn(){
  Connection conn = null;
  String url = "jdbc:mysql://localhost:3306/test";
  String user = "root";
  String password = "root";
  try {
   Class.forName("com.mysql.jdbc.Driver");
   conn = DriverManager.getConnection(url, user, password);
  } catch (ClassNotFoundException e) {
   e.printStackTrace();
  } catch (SQLException e) {
   e.printStackTrace();
  }
  return conn;
 }

這樣繁雜的操作只為了獲取一次連接，當(dāng)然，我們可以將其封裝成一個(gè)工具類(lèi)來(lái)訪問(wèn)（上圖以封裝好Connection的連接），但是每一次連接都需要取加載一次是不是很浪費(fèi)性能，為了優(yōu)化性能，那么就出現(xiàn)了連接池。

連接池在初始化的時(shí)候就創(chuàng)建了幾個(gè)連接供我們使用，當(dāng)我們需要連接時(shí)只需要從連接池中獲取已存在的連接，當(dāng)初始化的幾個(gè)連接都沒(méi)有時(shí)，會(huì)重新創(chuàng)建一個(gè)連接，使用完連接后不會(huì)去銷(xiāo)毀連接，而是歸還給連接池供后面需要連接的使用。（當(dāng)然，連接池不僅僅只是這么簡(jiǎn)單，這里就只做這些介紹）

常用的連接池有DBCP、C3P0，現(xiàn)在最主流是好像是阿里的Druid連接池，還有tomcat的自帶的JNDI連接池

二、自定義一個(gè)簡(jiǎn)單的連接池

對(duì)自定義連接池的分析：

1.2.因?yàn)槭沁B接池，我們需要實(shí)現(xiàn)DataSource接口，并實(shí)現(xiàn)其中的方法，基于我們的情況，我們關(guān)于與getConnection()方法；

2.既然要存放幾個(gè)連接對(duì)象，那么我們用一個(gè)集合來(lái)存放它，基于會(huì)經(jīng)常操作增加和刪除那么選用LinkedList；

3.連接的銷(xiāo)毀并不是銷(xiāo)毀連接，而是將連接歸還給連接池

編碼：

1.創(chuàng)建一個(gè)類(lèi)MyDataSource 并實(shí)現(xiàn)DataSource接口

當(dāng)此類(lèi)加載時(shí)它就需要有一個(gè)容器來(lái)存放Connection，所以定義一個(gè)靜態(tài)的屬性：

private static List<Connection> connectionList = new LinkedList<>();

2.因?yàn)樾枰～@得數(shù)據(jù)庫(kù)連接，所以我們封裝一個(gè)獲取數(shù)據(jù)庫(kù)連接的方法

public Connection getOneConnection(){
  Connection conn = null;
  try{
   //此處通過(guò)外部的properties文件來(lái)獲取的數(shù)據(jù)，這樣更加靈活。
   InputStream in = MyDataSource.class.getClassLoader().
     getResourceAsStream("jdbc/jdbc.properties");
   Properties pro = new Properties();
   pro.load(in);
   driver = pro.getProperty("driver");
   url = pro.getProperty("url");
   username = pro.getProperty("user");
   password = pro.getProperty("password");
   Class.forName(driver);
   conn = DriverManager.getConnection(url,username,password);
  }catch (Exception e){
   e.getStackTrace();
  }
  return conn;
 }

注意的是我這里通過(guò)propertie文件的獲取的數(shù)據(jù)，可根據(jù)實(shí)際情況來(lái)選擇

3.初始化幾個(gè)連接放入容器中。可以使用靜態(tài)代碼塊來(lái)實(shí)現(xiàn)，但是如果沒(méi)有使用此數(shù)據(jù)源那么就造成了資源的浪費(fèi)，所以我考慮將初始化幾個(gè)連接的實(shí)現(xiàn)放到他的構(gòu)造方法中，即當(dāng)需要此連接池的時(shí)候他才會(huì)隨之創(chuàng)建幾個(gè)連接。如下：

public MyDataSource() {
  for (int i = 0; i < 5; i++) {
   Connection conn = getOneConnection();//調(diào)用創(chuàng)建連接的方法
   connectionList.add(conn);
  }
 }

4.現(xiàn)在開(kāi)始重寫(xiě)外部從此連接池中獲取連接的方法getConnection()

@Override
 public Connection getConnection() throws SQLException {
  Connection conn = null;
  if(connectionList == null || connectionList.size() <= 0){
   Connection connection = getConnection();
   connectionList.add(connection);
  }
  conn = connectionList.remove(0);
  return conn;
 }

5.創(chuàng)建一個(gè)對(duì)象返回的方法，即將用完的連接放入歸還到連接池中

public void backConnection(Connection conn){
  connectionList.add(conn);
 }

OK，這樣就完成了一個(gè)簡(jiǎn)單的自定義連接池，測(cè)試代碼如下：

public static void main( String[] args ) throws SQLException
 {
  MyDataSource dataSource = new MyDataSource();
  Connection conn = dataSource.getConnection();
  String sql = "select * from user where u_id = ?";
  PreparedStatement ps = null;
  ResultSet rs = null;
  try {
   ps = conn.prepareStatement(sql);
   ps.setInt(1, 1);
   rs = ps.executeQuery();
   while (rs.next()) {
    System.out.println("id="+rs.getInt(1));
    System.out.println("username="+rs.getString(2));
    System.out.println("password="+rs.getString(3));
   }
  } catch (SQLException e) {
   e.printStackTrace();
  } finally {
   dataSource.backConnection(conn);
  }
 }

因?yàn)楹雎?，我的代碼中沒(méi)有關(guān)閉其他兩個(gè)對(duì)象。

現(xiàn)在有一個(gè)小問(wèn)題就是，我們的關(guān)閉連接是通過(guò)連接池的方法來(lái)實(shí)現(xiàn)的，但是，如果用戶調(diào)用Connection對(duì)象的close方法，那么連接時(shí)被銷(xiāo)毀了，并沒(méi)有返回給連接池，那么我們來(lái)優(yōu)化它，讓用戶使用close()方法不會(huì)去銷(xiāo)毀連接，而是去歸還連接。

方案有很多中，這里采用裝飾著模式的一種。

優(yōu)化：

1.新建一個(gè)類(lèi)MyConnection來(lái)實(shí)現(xiàn)Connection接口，其中他有屬性類(lèi)型為Connection conn和一個(gè)Liis<Connection>。

private Connection conn;
 private List<Connection> pool;
 
 public MyConnection(Connection conn, List<Connection> pool) {
  this.conn = conn;
  this.pool = pool;
 }

2.然后實(shí)現(xiàn)接口的close方法

 @Override
 public void close() throws SQLException {
  System.out.println("回收連接");
  pool.add(conn);
 }

3.然后實(shí)現(xiàn)他獲取Statement的方法，如果不實(shí)現(xiàn)那么獲取此Statement會(huì)出現(xiàn)空指針錯(cuò)誤,我這里就只實(shí)現(xiàn)了PreparedStatement的獲取方法

@Override
 public PreparedStatement prepareStatement(String sql) throws SQLException {
  System.out.println("獲得Statement");
  return conn.prepareStatement(sql);
 }

4.然后刪除掉MyDataSource類(lèi)中歸還連接的方法backConnection，并將構(gòu)造方法和獲取連接的方法做如下改造

public MyDataSource2() {
  for (int i = 0; i < 5; i++) {
   Connection conn = getOneConnection();
   MyConnection myConn = new MyConnection(conn, connectionList);
   connectionList.add(myConn);
  }
 }

@Override
 public Connection getConnection() throws SQLException {
  Connection conn = null;
  if(connectionList == null || connectionList.size() <= 0){
   Connection connection = getConnection();
   MyConnection myConn = new MyConnection(connection, connectionList);
   connectionList.add(myConn);
  }
  conn = connectionList.remove(0);
  return conn;
 }

好了，這樣用戶直接調(diào)用我們的Connection的close方法就不會(huì)去銷(xiāo)毀連接了，會(huì)正確的歸還給了連接池了，對(duì)測(cè)試代碼稍做修改即可測(cè)試。

以上這篇自定義一個(gè)簡(jiǎn)單的JDBC連接池實(shí)現(xiàn)方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: