JavaWeb学习-数据库连接池-2-装饰模式实现close

女爷i 2022-01-23 08:55 201阅读 0赞

前面一篇介绍了数据库连接池，也就是说以后，我们一个业务处理完了，不应该直接调用mysql实现的close方法，而是要把这个连接对象放回到连接池中取，供下一个请求使用。本篇就来看看如何解决这个close方法的问题。前面一篇代码中模拟了连接池pool的实现，就是因为每个人都可以写自己的连接池实现，造成了不规范，为了规范这种行为，sun公司提供了一个接口javax.sql.DataSource，这样大家都去实现这个接口，别人看到你实现了这个接口的代码就明白了这是一个连接池的实现。

1.MyDataSource.java

写一个类实现接口javax.sql.DataSource

package com.anthony.pool;
    
    import java.io.PrintWriter;
    import java.sql.Connection;
    import java.sql.SQLException;
    import java.sql.SQLFeatureNotSupportedException;
    import java.util.Collections;
    import java.util.LinkedList;
    import java.util.logging.Logger;
    
    import javax.sql.DataSource;
    
    import com.anthony.utils.DBUtils;
    
    public class MyDataSource implements DataSource {
    	
    	//创建一个存放连接的池子，大小为10
    	private static LinkedList<Connection> pool = (LinkedList<Connection>) Collections.synchronizedCollection(new LinkedList());
    	
    	static {
    		try {
    			for (int i = 0; i < 10; i++) {
    				Connection conn = DBUtils.getConnection();
    				pool.add(conn);
    			}
    		} catch (Exception e) {
    			throw new ExceptionInInitializerError("初始化数据库连接失败，请检查配置文件。");
    		}
    	}
    	
    	public Connection getConnection() throws SQLException {
    		Connection conn = null;
    		if(pool.size() > 0) {
    			conn = pool.removeFirst();
    			//对conn进行包装
    			MyConnection myConn = new MyConnection(conn, pool); //得到一个包装后的MyConnection对象
    			return myConn;
    		}else {
    			// 1 等待
    			// 2.等待超时，新创建一个连接
    			throw new RuntimeException("服务器忙");
    		}
    	}
    
    	public Connection getConnection(String username, String password) throws SQLException {
    		return null;
    	}
    	
    	@Override
    	public PrintWriter getLogWriter() throws SQLException {
    		return null;
    	}
    
    	@Override
    	public void setLogWriter(PrintWriter out) throws SQLException {
    		
    	}
    
    	@Override
    	public void setLoginTimeout(int seconds) throws SQLException {
    		
    	}
    
    	@Override
    	public int getLoginTimeout() throws SQLException {
    		return 0;
    	}
    
    	@Override
    	public Logger getParentLogger() throws SQLFeatureNotSupportedException {
    		return null;
    	}
    
    	@Override
    	public <T> T unwrap(Class<T> iface) throws SQLException {
    		return null;
    	}
    
    	@Override
    	public boolean isWrapperFor(Class<?> iface) throws SQLException {
    		return false;
    	}
    
    	
    }

我们重点实现getConnection()方法，其他方法默认就好。

2.MyConnection.java

MyConnection类包装mysql中实现的Connection。

package com.anthony.pool;
    
    import java.sql.Array;
    import java.sql.Blob;
    import java.sql.CallableStatement;
    import java.sql.Clob;
    import java.sql.Connection;
    import java.sql.DatabaseMetaData;
    import java.sql.NClob;
    import java.sql.PreparedStatement;
    import java.sql.SQLClientInfoException;
    import java.sql.SQLException;
    import java.sql.SQLWarning;
    import java.sql.SQLXML;
    import java.sql.Savepoint;
    import java.sql.Statement;
    import java.sql.Struct;
    import java.util.LinkedList;
    import java.util.Map;
    import java.util.Properties;
    import java.util.concurrent.Executor;
    
    /*1）编写一个类，实现与被包装类相同的接口（具备相同的行为）
    2）定义一个被包装类类型的变量
    3）定义构造方法，把被包装类的对象注入，给包装类变量赋值。
    4）对于不需要改写的方法，调用原有的方法。
    5）对于需要改写的方法，写自己的代码。*/
    
    public class MyConnection implements Connection {
    	
    	private Connection oldConnection;// com.mysql.jdbc.Connection
    	private LinkedList<Connection> pool;
    	
    	public MyConnection(Connection oldConnection, LinkedList<Connection> pool) {
    		this.oldConnection = oldConnection; //得到com.mysql.jdbc.Connection
    		this.pool = pool; //得到连接池对象
    	}
    	
    	@Override
    	public void close() throws SQLException {
    		pool.addLast(oldConnection); //把用完connection添加回集合中
    	}
    	
    	@Override
    	public PreparedStatement prepareStatement(String sql) throws SQLException {
    		return oldConnection.prepareStatement(sql);
    	}
    	
    	@Override
    	public PreparedStatement prepareStatement(String sql, int resultSetType, int resultSetConcurrency)
    			throws SQLException {
    		return oldConnection.prepareStatement(sql,resultSetConcurrency);
    	}
    	
    	//以下省略多个实现方法
    
    }

重点看构造方法和close方法。构造方法就是对conn对象进行包装，close方法就是把用户结束的连接对象进行放回到集合里。

3.TestJDBC.java

写一个测试类。

package com.anthony.pool;
    
    import java.sql.Connection;
    import java.sql.PreparedStatement;
    import java.sql.SQLException;
    
    import javax.sql.DataSource;
    
    import org.junit.Test;
    
    public class TestJDBC {
    	
    	@Test
    	public void test() {
    		DataSource ds = new MyDataSource();
    		Connection conn = null;
    		PreparedStatement ps = null;
    		
    		try {
    			conn = ds.getConnection(); //从池中取出一个连接
    			ps = conn.prepareStatement("select * from users");
    		} catch (SQLException e) {
    			e.printStackTrace();
    		} finally {
    			try {
    				conn.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    		}
    	}
    
    }