数据库工具类JDBCUtils
什么时候自己创建工具类？如果一个公共能经常要用到，建议将这个功能做成一个工具类，可以在不同的地方重用
补充 ：通过.class类文件获取流或者ClassLoader类加载器获取当前src路径下的文件
...\src\pro.properties
url="jdbc:mysql:///jdbclearn"
user=root
password=abc123
driver=com.mysql.jdbc.Driver
Main
// 【推荐】1.在src下查找pro.properties配置文件直接作为输入流
InputStream in = Main.class.getClassLoader().getResourceAsStream("pro.properties");
Properties pro = new Properties();
pro.load(in);
// 2.获得src文件夹下对应文件的绝对路径 通过当前类获得ClassLoader！
ClassLoader classLoader = Main.class.getClassLoader();
URL source = classLoader.getResource("pro.properties");
String path = source.getPath();
Properties properties = new Properties();
properties.load(new FileReader(path));
1.创建类jdbcUtil包含3个方法
- 可以把几个字符串定义成常量
- 用户名
- 密码
- URL
- 驱动类
- 得到数据库的连接：
getConnection() - 关闭所有打开的资源
close(Connection conn,Statement stmt)close(Connection conn,Statement stmt,ResultSet rs)
2.jdbcUtil工具类实现
import java.sql.*;
public class JdbcUtils {
// 可以把几个字符串定义成常量：用户名，密码，URL，驱动类
private static final String USER = "root";
private static final String PWD = "abc123";
private static final String URL = "jdbc:mysql://localhost:3306/jdbclearn";
private static final String DRIVER = "com.mysql.jdbc.Driver";
/**
* 注册驱动(可以省略)
*/
static {
try {
Class.forName(DRIVER); }
catch (ClassNotFoundException e) { e.printStackTrace(); }
}
/**
* 得到数据库的连接
*/
public static Connection getConnection() throws SQLException {
return DriverManager.getConnection(URL, USER, PWD);
}
/**
* 关闭所有打开的资源
*/
public static void close(Connection conn, Statement stmt){
if(stmt != null) {
try { stmt.close(); } catch (SQLException e) { e.printStackTrace(); }
}
if(conn != null) {
try { conn.close(); }catch (SQLException e) { e.printStackTrace(); }
}
}
/**
* 关闭所有打开的资源 重载
*/
public static void close(Connection conn, Statement stmt, ResultSet rs) {
if(rs != null) {
try { rs.close(); } catch (SQLException e) { e.printStackTrace(); }
}
close(conn, stmt);
}
}
3.登录案例【未防范SQL注入风险;密码未进行加密】
1.数据库准备
create database if not exists jdbclearn;
use jdbclearn;
create table user (
id int primary key auto_increment,
name varchar(20),
password varchar(20)
);
insert into user values
(null, 'jack', '123'),
(null, 'rose', '456');
2.使用上面的jdbcUtil工具类进行登录实验
- Login类
import java.sql.*;
import java.util.Scanner;
public class Login {
public static void main(String[] args) {
Scanner input = new Scanner(System.in);
System.out.println("请输入用户名>> ");
String name = input.nextLine();
System.out.println("请输入密码>> ");
String password = input.nextLine();
login(name, password);
}
public static void login(String name, String password) {
Connection connection = null;
Statement statement = null;
ResultSet rs = null;
try {
connection = JdbcUtils.getConnection();
statement = connection.createStatement();
String sql = "select * from user where name='" + name +"' and password='" + password + "'";
System.out.println(sql);
rs = statement.executeQuery(sql);
if(rs.next()) {
System.out.println("登录成功，欢迎您：" + name);
} else {
System.out.println("登录失败！");
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
JdbcUtils.close(connection, statement, rs);
}
}
}
- 正常登录以及结果
请输入用户名>>
jack
请输入密码>>
123
select * from user where name='jack' and password='123'
登录成功，欢迎您：jack
- SQL注入方式登录
请输入用户名>>
someONe
请输入密码>>
a' or '1' = '1
select * from user where name='someONe' and password='a' or '1' = '1'
登录成功，欢迎您：someONe
使用PreparedStatement//www.greatytc.com/p/8553fc6b9f41@接口可以防止SQL注入的风险
