java攻城狮之路--复习JDBC(PrepareStatement)

2024-01-08 14:12:16

PreparedStatement：

1、可以通过调用 Connection 对象的 preparedStatement() 方法获取 PreparedStatement 对象

2、PreparedStatement 接口是 Statement 的子接口，它表示一条预编译过的 SQL 语句

2、PreparedStatement 对象所代表的 SQL 语句中的参数用问号(?)来表示，

调用 PreparedStatement 对象的 setXXX() 方法来设置这些参数. setXXX() 方法有两个参数，

第一个参数是要设置的 SQL 语句中的参数的索引(从 1 开始)，第二个是设置的 SQL 语句中的参数的值.

PreparedStatement vs Statement：

1、代码的可读性和可维护性.

2、PreparedStatement 能最大可能提高性能：

–-DBServer会对预编译语句提供性能优化。因为预编译语句有可能被重复调用，所以语句在被DBServer

的编译器编译后的执行代码被缓存下来，那么下次调用时只要是相同的预编译语句就不需要编译，

只要将参数直接传入编译过的语句执行代码中就会得到执行。

–-在statement语句中,即使是相同操作但因为数据内容不一样,所以整个语句本身不能匹配,没有缓存语句的

意义.事实是没有数据库会对普通语句编译后的执行代码缓存.这样每执行一次都要对传入的语句编译一次.

–-(语法检查，语义检查，翻译成二进制命令，缓存)

3、PreparedStatement 可以防止 SQL 注入 .

package com.shellway.jdbc;

import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;

public class JDBCTools {

    /**
     * 执行 SQL 语句, 使用 PreparedStatement
     * @param sql
     * @param args: 填写 SQL 占位符的可变参数
     */
    public static void update(String sql, Object ... args){
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        
        try {
            connection = JDBCTools.getConnection();
            preparedStatement = connection.prepareStatement(sql);
            
            for(int i = 0; i < args.length; i++){
                preparedStatement.setObject(i + 1, args[i]);
            }
            
            preparedStatement.executeUpdate();
            
        } catch (Exception e) {
            e.printStackTrace();
        } finally{
            JDBCTools.releaseDB(null, preparedStatement, connection);
        }
    }
    
    /**
     * 执行 SQL 的方法
     * 
     * @param sql: insert, update 或 delete。 而不包含 select
     */
    public static void update(String sql) {
        Connection connection = null;
        Statement statement = null;

        try {
            // 1. 获取数据库连接
            connection = getConnection();

            // 2. 调用 Connection 对象的 createStatement() 方法获取 Statement 对象
            statement = connection.createStatement();

            // 4. 发送 SQL 语句: 调用 Statement 对象的 executeUpdate(sql) 方法
            statement.executeUpdate(sql);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 5. 关闭数据库资源: 由里向外关闭.
            releaseDB(null, statement, connection);
        }
    }

    /**
     * 释放数据库资源的方法
     * 
     * @param resultSet
     * @param statement
     * @param connection
     */
    public static void releaseDB(ResultSet resultSet, Statement statement,
            Connection connection) {

        if (resultSet != null) {
            try {
                resultSet.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        if (statement != null) {
            try {
                statement.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        if (connection != null) {
            try {
                connection.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

    }

    /**
     * 获取数据库连接的方法
     */
    public static Connection getConnection() throws IOException,
            ClassNotFoundException, SQLException {
        // 0. 读取 jdbc.properties
        /**
         * 1). 属性文件对应 Java 中的 Properties 类 2). 可以使用类加载器加载 bin 目录(类路径下)的文件
         */
        Properties properties = new Properties();
        InputStream inStream = ReviewTest.class.getClassLoader()
                .getResourceAsStream("jdbc.properties");
        properties.load(inStream);

        // 1. 准备获取连接的 4 个字符串: user, password, jdbcUrl, driverClass
        String user = properties.getProperty("user");
        String password = properties.getProperty("password");
        String jdbcUrl = properties.getProperty("jdbcUrl");
        String driverClass = properties.getProperty("driverClass");

        // 2. 加载驱动: Class.forName(driverClass)
        Class.forName(driverClass);

        // 3. 调用
        // DriverManager.getConnection(jdbcUrl, user, password)
        // 获取数据库连接
        Connection connection = DriverManager.getConnection(jdbcUrl, user,
                password);
        return connection;
    }
}

JDBCTools.java

    /**
     * 使用 PreparedStatement 将有效的解决 SQL 注入问题.
     */
    @Test
    public void testSQLInjection2() {
        String username = "a‘ OR PASSWORD = ";
        String password = " OR ‘1‘=‘1";

        String sql = "SELECT * FROM users WHERE username = ? "
                + "AND password = ?";

        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;

        try {
            connection = JDBCTools.getConnection();
            preparedStatement = connection.prepareStatement(sql);

            preparedStatement.setString(1, username);
            preparedStatement.setString(2, password);

            resultSet = preparedStatement.executeQuery();

            if (resultSet.next()) {
                System.out.println("登录成功!");
            } else {
                System.out.println("用户名和密码不匹配或用户名不存在. ");
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(resultSet, preparedStatement, connection);
        }
    }

    /**
     * SQL 注入.
     */
    @Test
    public void testSQLInjection() {
        String username = "a‘ OR PASSWORD = ";
        String password = " OR ‘1‘=‘1";

        String sql = "SELECT * FROM users WHERE username = ‘" + username
                + "‘ AND " + "password = ‘" + password + "‘";

        System.out.println(sql);

        Connection connection = null;
        Statement statement = null;
        ResultSet resultSet = null;

        try {
            connection = JDBCTools.getConnection();
            statement = connection.createStatement();
            resultSet = statement.executeQuery(sql);

            if (resultSet.next()) {
                System.out.println("登录成功!");
            } else {
                System.out.println("用户名和密码不匹配或用户名不存在. ");
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(resultSet, statement, connection);
        }
    }

使用 PreparedStatement 将有效的解决 SQL 注入问题

练习：

•插入一个新的 student 信息

1)请输入考生的详细信息

Type:

IDCard:

ExamCard:

StudentName:

Location:

Grade:

提示：信息录入成功!

2).在 eclipse 中建立 java 程序：输入身份证号或准考证号可以查询到学生的基本信息。结果如下：

java攻城狮之路--复习JDBC(PrepareStatement)

    @Test
    public void testGetStudent() {
        // 1. 得到查询的类型
        int searchType = getSearchTypeFromConsole();

        // 2. 具体查询学生信息
        Student student = searchStudent(searchType);

        // 3. 打印学生信息
        printStudent(student);
    }

    /**
     * 打印学生信息: 若学生存在则打印其具体信息. 若不存在: 打印查无此人.
     * 
     * @param student
     */
    private void printStudent(Student student) {
        if (student != null) {
            System.out.println(student);
        } else {
            System.out.println("查无此人!");
        }
    }

    /**
     * 具体查询学生信息的. 返回一个 Student 对象. 若不存在, 则返回 null
     * 
     * @param searchType
     *            : 1 或 2.
     * @return
     */
    private Student searchStudent(int searchType) {

        String sql = "SELECT flowid, type, idcard, examcard,"
                + "studentname, location, grade " + "FROM examstudent "
                + "WHERE ";

        Scanner scanner = new Scanner(System.in);

        // 1. 根据输入的 searchType, 提示用户输入信息:
        // 1.1 若 searchType 为 1, 提示: 请输入身份证号. 若为 2 提示: 请输入准考证号
        // 2. 根据 searchType 确定 SQL
        if (searchType == 1) {
            System.out.print("请输入准考证号:");
            String examCard = scanner.next();
            sql = sql + "examcard = ‘" + examCard + "‘";
        } else {
            System.out.print("请输入身份证号:");
            String examCard = scanner.next();
            sql = sql + "idcard = ‘" + examCard + "‘";
        }

        // 3. 执行查询
        Student student = getStudent(sql);

        // 4. 若存在查询结果, 把查询结果封装为一个 Student 对象

        return student;
    }

    /**
     * 根据传入的 SQL 返回 Student 对象
     * 
     * @param sql
     * @return
     */
    private Student getStudent(String sql) {

        Student stu = null;

        Connection connection = null;
        Statement statement = null;
        ResultSet resultSet = null;

        try {
            connection = JDBCTools.getConnection();
            statement = connection.createStatement();
            resultSet = statement.executeQuery(sql);

            if (resultSet.next()) {
                stu = new Student(resultSet.getInt(1), resultSet.getInt(2),
                        resultSet.getString(3), resultSet.getString(4),
                        resultSet.getString(5), resultSet.getString(6),
                        resultSet.getInt(7));
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(resultSet, statement, connection);
        }

        return stu;
    }

    /**
     * 从控制台读入一个整数, 确定要查询的类型
     * 
     * @return: 1. 用身份证查询. 2. 用准考证号查询 其他的无效. 并提示请用户重新输入.
     */
    private int getSearchTypeFromConsole() {

        System.out.print("请输入查询类型: 1. 用身份证查询. 2. 用准考证号查询 ");

        Scanner scanner = new Scanner(System.in);
        int type = scanner.nextInt();

        if (type != 1 && type != 2) {
            System.out.println("输入有误请重新输入!");
            throw new RuntimeException();
        }

        return type;
    }

    @Test
    public void testAddNewStudent() {
        Student student = getStudentFromConsole();
        addNewStudent2(student);
    }

    /**
     * 从控制台输入学生的信息
     */
    private Student getStudentFromConsole() {

        Scanner scanner = new Scanner(System.in);

        Student student = new Student();

        System.out.print("FlowId:");
        student.setFlowId(scanner.nextInt());

        System.out.print("Type: ");
        student.setType(scanner.nextInt());

        System.out.print("IdCard:");
        student.setIdCard(scanner.next());

        System.out.print("ExamCard:");
        student.setExamCard(scanner.next());

        System.out.print("StudentName:");
        student.setStudentName(scanner.next());

        System.out.print("Location:");
        student.setLocation(scanner.next());

        System.out.print("Grade:");
        student.setGrade(scanner.nextInt());

        return student;
    }

    public void addNewStudent2(Student student) {
        String sql = "INSERT INTO examstudent(flowid, type, idcard, "
                + "examcard, studentname, location, grade) "
                + "VALUES(?,?,?,?,?,?,?)";

        JDBCTools.update(sql, student.getFlowId(), student.getType(),
                student.getIdCard(), student.getExamCard(),
                student.getStudentName(), student.getLocation(),
                student.getGrade());
    }

    public void addNewStudent(Student student) {
        // 1. 准备一条 SQL 语句:
        String sql = "INSERT INTO examstudent VALUES(" + student.getFlowId()
                + "," + student.getType() + ",‘" + student.getIdCard() + "‘,‘"
                + student.getExamCard() + "‘,‘" + student.getStudentName()
                + "‘,‘" + student.getLocation() + "‘," + student.getGrade()
                + ")";

        System.out.println(sql);

        // 2. 调用 JDBCTools 类的 update(sql) 方法执行插入操作.
        JDBCTools.update(sql);
    }

}

练习代码

在练习中要注意的问题：

1、在执行这一语句ps = conn.prepareStatement(sql);的时候，如果提示要强制转换成prepareStatement类型，
则不要转换，只需要把包导入就行：import java.sql.PreparedStatement;
2、遇到插入日期语句的时候，比如：ps.setDate(3, new Date(new java.util.Date().getTime()));
前面的new Date（）是要声明为java.sql包中的，里面的new Date（）则要声明为java.util包中的。
在使用Preparement的时候，最好在主程序中写函数然后调用工具类JDBCTools.java中的更新方法：

public void addStudent(Student stu) throws Exception {
    String sql = "insert into examstudent values(?,?,?,?,?,?,?)";
    JDBCTools.testUpdate(sql, stu.getFlowID(), stu.getType(),
            stu.getIdCard(), stu.getExamCard(), stu.getStudentName(),
            stu.getLocation(), stu.getGrade());
    }

测试函数

在工具类JDBCTools.java中写更新方法，传入 sql 和可变参数args

public static void testUpdate(String sql, Object... args) throws Exception {
    Connection conn = null;
    PreparedStatement ps = null;

    try {
        conn = JDBCTools.getConnection();
        ps = conn.prepareStatement(sql);
        for (int i = 0; i < args.length; i++) {
            ps.setObject(i + 1, args[i]);
        }
        System.out.println(sql);
        ps.executeUpdate();
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        JDBCTools.release(null, ps, conn);
    }
}

工具类JDBCTools.java中的更新方法

下面写个简单而小巧的PrepareStatement例子：

    @Test
    public void testPreparedStatement() {
        Connection connection = null;
        PreparedStatement preparedStatement = null;

        try {
            connection = JDBCTools.getConnection();
            String sql = "INSERT INTO customers (name, email, birth) "
                    + "VALUES(?,?,?)";

            preparedStatement = connection.prepareStatement(sql);
            preparedStatement.setString(1, "shellway");
            preparedStatement.setString(2, "shellway@163.com");
            preparedStatement.setDate(3,
                    new Date(new java.util.Date().getTime()));

            preparedStatement.executeUpdate();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(null, preparedStatement, connection);
        }
    }

PrepareStatement例子

java攻城狮之路--复习JDBC(PrepareStatement)

码农公寓

相关文章