创建Connection的多种方法

时间:2022-01-24 22:13:49

一、利用ThreadLocal

package day01;

import java.io.InputStream;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

public class DBUtil {
private static String url;
private static String user;
private static String password;
//用于管理不同线程所获取的连接
private static ThreadLocal<Connection> tl
= new ThreadLocal<Connection>();
static{
try{
/**
* properties类用于读取properties文件
* 类似Map的形式读取配置文件中的内容
* user=scott
* 等号左边就是key,等号右边就是value
*/
Properties p = new Properties();
InputStream is = DBUtil.class.getClassLoader().getResourceAsStream("wq/config.properties");
/**
* p.load(InputStream/InputReader)可传入的参数为字节流获字符流
*/
p.load(is);
/**
* 当通过Properties读取文件后,要关闭这个流
*/
is.close();
String driver = p.getProperty("driver").trim();
url = p.getProperty("url").trim();
user = p.getProperty("user").trim();
password = p.getProperty("password").trim();
//注册驱动
Class.forName(driver);
}catch(Exception e){
e.printStackTrace();
}
}
/**
* 获取连接
* @throws Exception
*/
public static Connection getConnection() throws Exception{
try{

Connection conn =
DriverManager.getConnection(url,user,password);
/**
* ThreadLocal的set方法会将当前线程作为key,并将给定
* 的值作为value存入内部的map中保存
*/
tl.set(conn);
return conn;
}catch(SQLException e){
e.printStackTrace();
//通知调用者,创建连接出错
throw e;
}
}
/**
* 关闭给定连接
*/
public static void closeConnection(){
try{
//将当前线程作为key获取连接,若直接调用此方法,没有调用set,则会返回null
Connection conn = tl.get();
if(conn != null){
conn.close();
tl.remove();//将当前连接删除 (根据value保存的参数视情况而定是否要删除该参数)
}
}catch(SQLException e){
e.printStackTrace();
}
}
}

此方法可以跨方法传递参数,在此例中关闭连接时就不用传入参数,利用ThreadLocal来获取其他方法中的参数

二、普通做法

package util;

import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

public class DBTool {

private static String driver;
private static String url;
private static String user;
private static String pwd;

static {
//在类加载时只读取一次配置文件
Properties p = new Properties();
try {
p.load(DBTool.class.getClassLoader()
.getResourceAsStream("db.properties"));
driver = p.getProperty("driver");
url = p.getProperty("url");
user = p.getProperty("user");
pwd = p.getProperty("pwd");
//只需要加载一次驱动
Class.forName(driver);
} catch (IOException e) {
e.printStackTrace();
throw new RuntimeException(
"加载配置文件失败", e);
} catch (ClassNotFoundException e) {
e.printStackTrace();
throw new RuntimeException(
"加载驱动类失败", e);
}
}

/**
* 此处直接抛出SQLException,是为了强制
* 调用者去处理异常,从而避免其忘记写finally,
* 以及在finally当中关闭连接.
*/
public static Connection getConnection()
throws SQLException {
return DriverManager
.getConnection(url, user, pwd);
}

public static void close(Connection conn) {
if(conn != null) {
try {
conn.close();
} catch (SQLException e) {
e.printStackTrace();
throw new RuntimeException(
"关闭连接失败", e);
}
}
}
}