JavaWeb学习-JDBC系列-7-封装一个DB工具类

绝地灬酷狼 2022-02-13 04:57 247阅读 0赞

我们前面在多个单元测试用例的时候，每个用例里都需要写加载驱动，创建连接，创建执行sql的stmt对象，创建结果集rs对象。这些很多重复的代码，我们可以提出来，写成一个静态工具类。这样，在使用的时候看起来很简洁。

1.我们尝试提取出来的连接数据库方法效果

我在项目src下创建一个com.anthony.utils的包，然后在这个包下创建DBUtils.java文件。

package com.anthony.utils;
    
    import java.sql.Connection;
    import java.sql.DriverManager;
    
    public class DBUtils {
    	//得到连接的方法
    	public static Connection getConnection() throws Exception {
    		Class.forName("com.mysql.jdbc.Driver");
    		Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/mydatabase", "root", "123456");
    		return conn;
    	}
    }

我们第一次抽取成这样，其实效果达到了，只是还不够完美。因为，我们数据库连接信息和驱动文件随时都可能发生改变，但是，这里我们采用硬编码，直接写死，这个是不合理。所以，下面我们来看看提取数据库连接信息到配置文件的效果。

2.采用配置文件保存数据库信息的优化效果

我首先在src下新建一个dbinfo.properties文件，写入下面这些内容。

classDriver=com.mysql.jdbc.Driver
    url=jdbc:mysql://localhost:3306/mydatabase
    user=root
    password=123456

有了配置文件，我们继续优化连接数据库方法。

package com.anthony.utils;
    
    import java.io.IOException;
    import java.io.InputStream;
    import java.sql.Connection;
    import java.sql.DriverManager;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    import java.sql.Statement;
    import java.util.Properties;
    
    public class DBUtils {
    	private static String classDriver;
    	private static String url;
    	private static String user;
    	private static String password;
    	
    	static {
    		Properties pro = new Properties();
    		InputStream in = DBUtils.class.getResourceAsStream("/dbinfo.properties");
    		try {
    			pro.load(in);
    			classDriver = pro.getProperty("classDriver");
    			url = pro.getProperty("url");
    			user = pro.getProperty("user");
    			password = pro.getProperty("password");
    		} catch (IOException e) {
    			e.printStackTrace();
    		}
    		try {
    			Class.forName(classDriver);
    		} catch (ClassNotFoundException e) {
    			e.printStackTrace();
    		}
    	}
    	
    	//得到连接的方法
    	public static Connection getConnection() throws Exception{
    		return DriverManager.getConnection(url, user, password);
    	}
    	
    	//关闭资源方法
    	public static void close(Connection conn, Statement stmt, ResultSet rs) {
    		if(rs != null) {
    			try {
    				rs.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    			rs = null; //赶紧垃圾回收
    		}
    		if(stmt != null) {
    			try {
    				stmt.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    			stmt = null;
    		}
    		if(conn != null) {
    			try {
    				conn.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    			conn = null;
    		}
    	}
    }

我写了一个测试类来测试下上面使用InputStream读取properties文件代码对不对。

package demo;
    
    import java.sql.Connection;
    import java.sql.ResultSet;
    import java.sql.Statement;
    
    import org.junit.Test;
    
    import com.anthony.utils.DBUtils;
    
    public class CURD {
    	
    	@Test
    	public void testSelect() {
    		Connection conn = null;
    		Statement stmt = null;
    		ResultSet rs = null;
    		
    		try {
    			conn = DBUtils.getConnection();
    			stmt = conn.createStatement();
    			rs = stmt.executeQuery("SELECT * FROM student");
    			while (rs.next()) {
    				System.out.println(
    						rs.getObject(1) + "\t" + rs.getObject(2) + "\t" + rs.getObject(3) + "\t" + rs.getObject(4));
    			} 
    			
    		} catch (Exception e) {
    			e.printStackTrace();
    		}
    	}
    }

我这边能打印查询内容，这种写法是没问题，唯一有点就是读取propeties文件这块有点啰嗦，下面我们换ResourceBundle类来读取properties文件会简单很多。

3.使用ResourceBundle类来获取配置文件key的value

这里我们来学习一个新的类，这个ResourceBundle是在G11n（国际化）开发中必须使用到的一个类。做过国际化测试的人应该知道，一个String，先是有英文，然后会翻译成日语，法语，德育，中文等不同本地化语言，靠的就是不同语言版本的资源文件。

package com.anthony.utils;
    
    import java.sql.Connection;
    import java.sql.DriverManager;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    import java.sql.Statement;
    import java.util.ResourceBundle;
    
    public class DBUtils {
    	private static String classDriver;
    	private static String url;
    	private static String user;
    	private static String password;
    	
    	static {
    		// 用于加载properties文件
    		ResourceBundle rb = ResourceBundle.getBundle("dbinfo"); //注意这里不需要文件扩展名.properties
    		classDriver = rb.getString("classDriver");
    		url = rb.getString("url");
    		user = rb.getString("user");
    		password = rb.getString("password");
    		
    		try {
    			Class.forName(classDriver);
    		} catch (ClassNotFoundException e) {
    			e.printStackTrace();
    		}
    	}
    	
    	//得到连接的方法
    	public static Connection getConnection() throws Exception{
    		return DriverManager.getConnection(url, user, password);
    	}
    	
    	//关闭资源方法
    	public static void close(Connection conn, Statement stmt, ResultSet rs) {
    		if(rs != null) {
    			try {
    				rs.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    			rs = null; //赶紧垃圾回收
    		}
    		if(stmt != null) {
    			try {
    				stmt.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    			stmt = null;
    		}
    		if(conn != null) {
    			try {
    				conn.close();
    			} catch (SQLException e) {
    				e.printStackTrace();
    			}
    			conn = null;
    		}
    	}
    }

使用了ResourceBundle类加载properties文件之后，代码少了好几行，以后如果需要Java解析properties文件，建议使用这个，这个也很好记住，比那个InputStream好记多了。DBUtils工具类封装好了，下一篇，我们来写四个Junit单元测试用例，也就是SQL的CURD，增删改查四个用例。