Java读取数据源相关信息
一、采用读取数据源配置文件的方式
package com.ofsp.utils;
import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
public class DataSourceUtil {
public static final String DATASOURCE_FILENAME ="jdbc.properties"; //连接数据库的数据源文件
public static final String DATASOURCE_URL = "jdbc.url"; //数据源文件里url的key
public static final String DATASOURCE_USERNAME = "jdbc.username";//数据源文件里用户名的key
public static final String DATASOURCE_PASSWORD = "jdbc.password";//数据源文件里密码的key
/**
* 读取配置文件信息
* @return Properties 配置文件信息
* @throws Exception
*/
public static Properties getProperties() throws Exception {
InputStream in = null;
if (in == null) {
//
Class.getResourceAsStream() 会指定要加载的资源路径与当前类所在包的路径一致。例如你写了一个MyTest类在包com.test.mycode 下，那么MyTest.class.getResourceAsStream("name") 会在com.test.mycode包下查找相应的资源。如果这个name是以 '/' 开头的，那么就会从classpath的根路径下开始查找。 ClassLoader.getResourceAsStream() 无论要查找的资源前面是否带'/' 都会从classpath的根路径下查找。 in = Thread.currentThread().getContextClassLoader().getResourceAsStream(DATASOURCE_FILENAME); //路径是 classpath的相对路径， classpath是指 WEB-INF文件夹下的classes目录
if(in == null){
throw new Exception("找不到路径为"+DATASOURCE_FILENAME+"的数据源配置文件。");
}
}
Properties properties = new Properties();
try {
properties.load(in);
} catch (IOException e) {
e.printStackTrace();
}
return properties;
}
/**
* 读取配置文件获取连接数据库的数据库url
* @return String 数据库名
* @throws Exception
*/
public static String getDatabaseURL() throws Exception {
Properties p = getProperties();
return p.getProperty(DATASOURCE_URL);
}
/**
* 读取配置文件获取连接数据库的用户名
* @return String 用户名
* @throws Exception
*/
public static String getUserOfDatabase() throws Exception {
Properties p = getProperties();
return p.getProperty(DATASOURCE_USERNAME);
}
/**
* 读取配置文件获取连接数据库的密码
* @return String 密码
* @throws Exception
*/
public static String getPasswordOfDatabase() throws Exception {
Properties p = getProperties();
return p.getProperty(DATASOURCE_PASSWORD);
}
/**
* 获取数据库Connection
* @return
* @throws Exception
*/
public static Connection getConnection() throws Exception{
Properties p = getProperties();
String url = p.getProperty(DATASOURCE_URL);
String username = p.getProperty(DATASOURCE_USERNAME);
String password = p.getProperty(DATASOURCE_PASSWORD);
return DriverManager.getConnection(url, username, password);
}
/**
* 获取指定数据库和用户的所有表名
* @param conn 连接数据库对象
* @param user 用户
* @param database 数据库名
* @return
*/
public static List<String> getAllTableNames(Connection conn,String database,String schem,String tableName) {
List<String> tableNames = new ArrayList<String>();
if (conn != null) {
try {
DatabaseMetaData metadata = conn.getMetaData();
// 表名列表
ResultSet rest = metadata.getTables(database, schem.toUpperCase(), tableName.toUpperCase(), new String[] {"TABLE"});
while (rest.next()) {
tableNames.add(rest.getString("TABLE_NAME"));
}
} catch (SQLException e) {
e.printStackTrace();
}
}
return tableNames;
}
}
二、如果采用Spring框架的话，还可以从Spring的配置文件中获取当前数据源：
数据源文件配置在spring-context.xml文件中，部分片断截取：
DruidDataSource dataSource = ApplicationContextHelper.getBean("dataSource");//获取数据源
Connection connection = dataSource.getConnection();//获取连接
String user = dataSource.getUsername();//获取用户名
注意：connection用完后一定要及时close掉，否则会出现溢出错误。
