JDBC入门
在做实例前需要下载一个mysql的jar包
1. 实例:
``package com.imooc.jdbc.demo1;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import org.junit.Test;
public class JDBCDemo2 {
@Test
/*
* 查询一条记录
*/
public void demo4() {
Connection conn=null;
Statement stmt=null;
ResultSet rs=null;
try {
//注册驱动
Class.forName("com.mysql.jdbc.Driver");
//获得链接
conn =DriverManager.getConnection("jdbc:mysql:///jdbc", "root", "123456");
//获得执行SQL语句的对象
stmt=conn.createStatement();
//编写SQL
String sql="select * from user where uid=2";
//执行SQL语句
rs=stmt.executeQuery(sql);
//遍历结果集
while(rs.next()) {
System.out.println(rs.getInt("uid")+" "+rs.getString("username")+" "+rs.getString("password")+" "+rs.getString("name"));
}
} catch (Exception e) {
e.printStackTrace();
}finally {
//释放资源
if(rs!= null) {
try {
rs.close();
} catch (Exception e) {
e.printStackTrace();
}
rs=null;
}
if(stmt!=null) {
try {
stmt.close();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
stmt=null;
}
if(conn!=null) {
try {
} catch (Exception e) {
e.printStackTrace();
}
conn=null;
}
}
}
@Test
/*
* 查询所有记录
*/
public void demo3() {
Connection conn=null;
Statement stmt=null;
ResultSet rs=null;
try {
//注册驱动
Class.forName("com.mysql.jdbc.Driver");
//获得链接
conn =DriverManager.getConnection("jdbc:mysql:///jdbc", "root", "123456");
//获得执行SQL语句的对象
stmt=conn.createStatement();
//编写SQL
String sql="select * from user";
//执行SQL语句
rs=stmt.executeQuery(sql);
//遍历结果集
while(rs.next()) {
System.out.println(rs.getInt("uid")+" "+rs.getString("username")+" "+rs.getString("password")+" "+rs.getString("name"));
}
} catch (Exception e) {
e.printStackTrace();
}finally {
//释放资源
if(rs!= null) {
try {
rs.close();
} catch (Exception e) {
e.printStackTrace();
}
rs=null;
}
if(stmt!=null) {
try {
stmt.close();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
stmt=null;
}
if(conn!=null) {
try {
} catch (Exception e) {
e.printStackTrace();
}
conn=null;
}
}
}
@Test
/*
* 删除操作
*/
public void demo2() {
Connection conn = null;
Statement stmt = null;
try {
//注册驱动
Class.forName("com.mysql.jdbc.Driver");
//获得链接
conn=DriverManager.getConnection("jdbc:mysql:///jdbc", "root", "123456");
//获得执行SQL语句的对象
stmt=conn.createStatement();
//编写SQL
String sql="delete from user where uid=4";
//执行SQL语句
int i=stmt.executeUpdate(sql);
if(i>0) {
System.out.println("删除成功!");
}
} catch (Exception e) {
e.printStackTrace();
}finally {
//释放资源
if(stmt!=null) {
try {
stmt.close();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
stmt=null;
}
if(conn!=null) {
try {
} catch (Exception e) {
e.printStackTrace();
}
conn=null;
}
}
}
@Test
/*
* 修改操作
*/
public void demo1() {
Connection conn = null;
Statement stmt = null;
try{
// 注册驱动:
Class.forName("com.mysql.jdbc.Driver");
// 获得连接:
conn = DriverManager.getConnection("jdbc:mysql:///jdbc", "root", "123456");
// 获得执行SQL语句的对象:
stmt = conn.createStatement();
// 编写SQL:
String sql = "insert into user values (null,'eee','123','张三')";
// 执行SQL:
int i = stmt.executeUpdate(sql);
if(i > 0){
System.out.println("保存成功!");
}
}catch(Exception e){
e.printStackTrace();
}finally{
// 释放资源:
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;
}
}}
从上面的操作来看,这样重复的代码很多,所以我们创建了一个工具类,专门用来存储这些重复的代码:(ps:如果把url,username,和password这些值都放在工具类中的话,每次要修改的时候都需要改源代码不好,所以我们又创建了一个配置文件jdbc.properties)
package com.imooc.jdbc.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;
/*
* JDBC的工具类
*/
public class JDBCUtils {
private static final String driverClass;
private static final String url;
private static final String username;
private static final String password;
static {
//加载属性文件并解析
Properties props=new Properties();
//获得属性文件的输入流
//通常使用类的加载器的方式进行获取
InputStream is=JDBCUtils.class.getClassLoader().getResourceAsStream("jdbc.properties");
try {
props.load(is);
} catch (IOException e) {
e.printStackTrace();
}
driverClass=props.getProperty("driverClass");
url=props.getProperty("url");
username=props.getProperty("username");
password=props.getProperty("password");
}
/*
* 注册驱动的方法
*/
public static void loadDriver() throws ClassNotFoundException {
Class.forName(driverClass);
}
/*
* 获得连接的方法
*/
public static Connection getConnection() throws Exception {
loadDriver();
Connection conn=DriverManager.getConnection(url,username,password);
return conn;
}
/*
* 资源的释放
*/
public static void release(Statement stmt,Connection conn) {
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;
}
}
public static void release(ResultSet rs,Statement stmt,Connection conn) {
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;
}
}
}
jdbc.properties代码如下:
driverClass=com.mysql.jdbc.Driver
url=jdbc:mysql:///jdbc
username=root
password=123456
这样的话,等我们需要用到的时候通过(类名 . 方法名)直接调用就可以了。最终版如下:
package com.imooc.jdbc.demo1;
import java.sql.Connection;
import java.sql.Statement;
import org.junit.Test;
import com.imooc.jdbc.utils.JDBCUtils;
public class JDBCDemo3 {
@Test
//保存记录
public void demo1() {
Connection conn=null;
Statement stmt=null;
try {
//获得连接
conn=JDBCUtils.getConnection();
//创建执行SQL的对象
stmt=conn.createStatement();
//编写SQL
String sql="insert into user values(null,'ggg','123','小六')";
//执行SQL
int num=stmt.executeUpdate(sql);
if(num>0) {
System.out.println("保存成功");
}
} catch (Exception e) {
e.printStackTrace();
}finally {
//释放资源
JDBCUtils.release(stmt, conn);
}
}
}
这样子,代码量就少了很多了~
JDBC的SQL注入漏洞
原本的测试是这样的:
package com.imooc.jdbc.demo1;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.Statement;
import org.junit.Test;
import com.imooc.jdbc.utils.JDBCUtils;
/*
*演示JDBC的注入漏洞
*/
public class JDBCDemo4 {
/*
*测试JDBC的注入漏洞方法
*/
@Test
public void demo1(){
boolean flag = JDBCDemo4.login("aaa", "111");
if(flag == true){
System.out.println("登录成功!");
}else{
System.out.println("登录失败!");
}
}
/**
* 产生SQL注入漏洞的方法
* @param username
* @param password
* @return
*/
public static boolean login(String username,String password){
Connection conn = null;
Statement stmt = null;
ResultSet rs = null;
boolean flag = false;
try{
conn = JDBCUtils.getConnection();
// 创建执行SQL语句的对象:
stmt = conn.createStatement();
// 编写SQL:
String sql = "select * from user where username = '"+username+"' and password = '"+password+"'";
// 执行SQL:
rs = stmt.executeQuery(sql);
// 判断结果集中是否有数据。
if(rs.next()){
flag = true;
}else{
flag = false;
}
}catch(Exception e){
e.printStackTrace();
}finally{
JDBCUtils.release(rs, stmt, conn);
}
return flag;
}
}
演示结果:
如果输入的账号或者密码与mysql中的数据不一样的话,就会提示登录失败。那么当把第21 行的代码修改以下两个代码中的任意一条时:
boolean flag = JDBCDemo4.login("aaa'or '1=1", "adfaf");
boolean flag = JDBCDemo4.login("aaa' -- ", "adfaf");
结果仍然是登录成功。因为 or 是sql的关键字,在计算的时候or后面的全部是错的,而前面的“aaa”是正确的,那正确or错误,结果还是正确的,所以这就是sql的注入漏洞问题。
引入了PreparedStatement这个预处理对象,解决方法如下,其中的?是占位符,表示一个变量,不能被修改,所以上面那种添加一个or的就不可行了。setString中的1代表第一个占位符,2代表第二个
public class JDBCDemo4 {
/*
*测试JDBC的注入漏洞方法
*/
@Test
public void demo1(){
boolean flag = JDBCDemo4.login2("aaa", "111");
if(flag == true){
System.out.println("登录成功!");
}else{
System.out.println("登录失败!");
}
}
/**
* 避免SQL注入漏洞的方法
*/
public static boolean login2(String username,String password){
Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;
boolean flag = false;
try{
// 获得连接:
conn = JDBCUtils.getConnection();
// 编写SQL:
String sql = "select * from user where username = ? and password = ?";
// 预处理SQL:
pstmt = conn.prepareStatement(sql);
// 设置参数:
pstmt.setString(1, username);
pstmt.setString(2, password);
// 执行SQL:
rs = pstmt.executeQuery();
// 判断结果街
if(rs.next()){
flag = true;
}else{
flag = false;
}
}catch(Exception e){
e.printStackTrace();
}finally{
JDBCUtils.release(rs, pstmt, conn);
}
return flag;
}
}
}
再测试后,就不会有注入漏洞的问题啦~
PreaparedStatement的使用
查询数据,修改数据,增加数据,删除数据
package com.imooc.jdbc.demo2;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import org.junit.Test;
import com.imooc.jdbc.utils.JDBCUtils;
/**
* PreparedStatement的使用
* @author jt
*
*/
public class JDBCDemo5 {
@Test
/**
* 查询一条记录
*/
public void demo5(){
Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;
try{
// 获得连接:
conn = JDBCUtils.getConnection();
// 编写SQL:
String sql = "select * from user where uid = ?";
// 预编译SQL:
pstmt = conn.prepareStatement(sql);
// 设置参数:
pstmt.setObject(1, 3);
// 执行SQL:
rs = pstmt.executeQuery();
// 判断结果集:
if(rs.next()){
System.out.println(rs.getInt("uid")+" "+rs.getString("username")+" "+rs.getString("password")+" "+rs.getString("name"));
}
}catch(Exception e){
e.printStackTrace();
}finally{
JDBCUtils.release(rs, pstmt, conn);
}
}
@Test
/**
* 查询所有数据
*/
public void demo4(){
Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;
try{
// 获得连接:
conn = JDBCUtils.getConnection();
// 编写SQL:
String sql = "select * from user";
// 预编译SQL:
pstmt = conn.prepareStatement(sql);
// 设置参数
// 执行SQL:
rs = pstmt.executeQuery();
while(rs.next()){
System.out.println(rs.getInt("uid")+" "+rs.getString("username")+" "+rs.getString("password")+" "+rs.getString("name"));
}
}catch(Exception e){
e.printStackTrace();
}finally{
JDBCUtils.release(rs, pstmt, conn);
}
}
@Test
/**
* 删除数据
*/
public void demo3(){
Connection conn = null;
PreparedStatement pstmt = null;
try{
// 获得连接:
conn = JDBCUtils.getConnection();
// 编写SQL:
String sql = "delete from user where uid = ?";
// 预编译SQL:
pstmt = conn.prepareStatement(sql);
// 设置参数:
pstmt.setInt(1, 6);
// 执行SQL:
int num = pstmt.executeUpdate();
if(num > 0){
System.out.println("删除成功!");
}
}catch(Exception e){
e.printStackTrace();
}finally{
JDBCUtils.release(pstmt, conn);
}
}
@Test
/**
* 修改数据
*/
public void demo2(){
Connection conn = null;
PreparedStatement pstmt = null;
try{
// 获得连接:
conn = JDBCUtils.getConnection();
// 编写SQL:
String sql = "update user set username = ?,password = ?,name = ? where uid = ?";
// 预编译SQL:
pstmt = conn.prepareStatement(sql);
// 设置参数:
pstmt.setString(1, "www");
pstmt.setString(2, "123456");
pstmt.setString(3, "张六");
pstmt.setInt(4, 6);
// 执行SQL:
int num = pstmt.executeUpdate();
if(num > 0){
System.out.println("修改成功!");
}
}catch(Exception e){
e.printStackTrace();
}finally{
JDBCUtils.release(pstmt, conn);
}
}
@Test
/**
* 保存数据
*/
public void demo1(){
Connection conn = null;
PreparedStatement pstmt = null;
try{
// 获得连接:
conn = JDBCUtils.getConnection();
// 编写SQL:
String sql = "insert into user values (null,?,?,?)";
// 预处理SQL:
pstmt = conn.prepareStatement(sql);
// 设置参数的值:
pstmt.setString(1, "qqq");
pstmt.setString(2, "123");
pstmt.setString(3, "张武");
// 执行SQL:
int num = pstmt.executeUpdate();
if(num > 0){
System.out.println("保存成功!");
}
}catch(Exception e){
e.printStackTrace();
}finally{
// 释放资源
JDBCUtils.release(pstmt, conn);
}
}
}
数据库连接池
C3P0连接池
首先还是导包,c3p0-config.xml到src目录下就可以了,然后抽取方法出来做成工具类
package com.imooc.jdbc.utils;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import com.mchange.v2.c3p0.ComboPooledDataSource;
public class JDBCUtils2 {
private static final ComboPooledDataSource dataSource=new ComboPooledDataSource();
/*
* 获得连接的方法
*/
public static Connection getConnection() throws Exception {
Connection conn=dataSource.getConnection();
return conn;
}
/*
* 资源的释放
*/
public static void release(Statement stmt,Connection conn) {
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;
}
}
public static void release(ResultSet rs,Statement stmt,Connection conn) {
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;
}
}
}
package com.imooc.jdbc.demo2;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import org.junit.Test;
import com.imooc.jdbc.utils.JDBCUtils;
import com.imooc.jdbc.utils.JDBCUtils2;
import com.mchange.v2.c3p0.ComboPooledDataSource;
/*
* 连接池的测试类
*/
public class DataSourceDemo1 {
@Test
/*
* 使用配置文件的方式
*/
public void demo2() {
Connection conn=null;
PreparedStatement pstmt=null;
ResultSet rs=null;
try {
//获得连接
conn=JDBCUtils2.getConnection();
//编写sql
String sql="select * from user";
//预编译sql
pstmt=conn.prepareStatement(sql);
//执行sql
rs=pstmt.executeQuery();
while(rs.next()) {
System.out.println(rs.getInt("uid")+rs.getString("username")+rs.getString("password")+rs.getString("name"));
}
} catch (Exception e) {
e.printStackTrace();
} finally {
JDBCUtils2.release(rs, pstmt, conn);
}
}
@Test
/*
* 手动设置连接池
*/
public void demo1() {
//获得连接
Connection conn=null;
PreparedStatement pstmt=null;
ResultSet rs=null;
try {
//创建连接池:
ComboPooledDataSource dataSource=new ComboPooledDataSource();
//设置连接池的参数
dataSource.setDriverClass("com.mysql.jdbc.Driver");
dataSource.setJdbcUrl("jdbc:mysql:///jdbc");
dataSource.setUser("root");
dataSource.setPassword("123456");
dataSource.setMaxPoolSize(20);
dataSource.setInitialPoolSize(3);
//获得连接
conn=dataSource.getConnection();
//编写sql
String sql="select * from user";
//预编译sql
pstmt=conn.prepareStatement(sql);
//执行sql
rs=pstmt.executeQuery();
while(rs.next()) {
System.out.println(rs.getInt("uid")+rs.getString("username")+rs.getString("password")+rs.getString("name"));
}
} catch (Exception e) {
e.printStackTrace();
} finally {
JDBCUtils.release(rs, pstmt, conn);
}
}
}