【Java】+操作数据库

一、操作MySQL数据库

已封装好 可直接导入使用

import org.testng.annotations.Test;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author 
 * @version 1.0
 * @time 2020/1/6 16:30
 */
public class Data {
    // MySQL 8.0 以下版本 - JDBC 驱动名及数据库 URL
    static private String JDBC_DRIVER = "com.mysql.jdbc.Driver";
    static private String DB_URL = "jdbc:mysql://数据库IP:3306/库";

    // MySQL 8.0 以上版本 - JDBC 驱动名及数据库 URL
//    static final String JDBC_DRIVER = "com.mysql.cj.jdbc.Driver";
//    static final String DB_URL = "jdbc:mysql://数据库IP:3306/RUNOOB?useSSL=false&serverTimezone=UTC";

    // 数据库的用户名与密码，需要根据自己的设置
    static private String USER = "账号";
    static private String PASS = "密码";

    /**
     * 更新数据（增、删、改）
     *
     * @param updatesSql sql语句
     */
    public static void dataUpdate(List<String> updatesSql) {
        Connection conn = null;
        Statement stmt = null;
        try {
            // step1 注册 JDBC 驱动
            Class.forName(JDBC_DRIVER);

            // step2 连接数据库 返回一个指针 用于操作数据库
            conn = DriverManager.getConnection(DB_URL, USER, PASS);

            // step3 创建一个实例化对象 返回一个对象指针 用于执行sql语句
            stmt = conn.createStatement();

            // step4 执行sql语句
            String sql = "";
            int rs = 0;
            for (int i = 0; i < updatesSql.size(); i++) {
                sql = updatesSql.get(i);
                rs = stmt.executeUpdate(sql);
                System.out.println(String.format("执行第 %s 条sql：%s", i + 1, sql));
            }

            // step5 关闭所有指针
            stmt.close();
            conn.close();
        } catch (SQLException se) {
            // 处理 JDBC 错误
            se.printStackTrace();
        } catch (Exception e) {
            // 处理 Class.forName 错误
            e.printStackTrace();
        } finally {
            // 关闭资源
            try {
                if (stmt != null) stmt.close();
            } catch (SQLException se2) {
            }// 什么都不做
            try {
                if (conn != null) conn.close();
            } catch (SQLException se) {
                se.printStackTrace();
            }
        }
        System.out.println("Goodbye!");
    }

    /**
     * 查询数据（查）
     *
     * @param searchSql sql语句
     * @return 返回查询后的结果（每一行、每一列）
     */
    public static List<ArrayList> dataSearch(String searchSql) {
        List<ArrayList> resultList = new ArrayList<>();

        Connection conn = null;
        Statement stmt = null;
        try {
            // step1 注册 JDBC 驱动
            Class.forName(JDBC_DRIVER);

            // step2 连接数据库 返回一个指针 用于操作数据库
            conn = DriverManager.getConnection(DB_URL, USER, PASS);

            // step3 创建一个实例化对象 返回一个对象指针 用于执行sql语句
            stmt = conn.createStatement();

            // step4 执行sql语句
            String sql = searchSql;
            ResultSet resultSet = stmt.executeQuery(sql);
            System.out.println(String.format("查询后的结果总列数：%s", resultSet.getMetaData().getColumnCount()));
            int columnCount = resultSet.getMetaData().getColumnCount();
            System.out.println(String.format("查询后的结果总列数：%s", columnCount));

            while (resultSet.next()) {
                System.out.println(resultSet.getObject(1));
                ArrayList result = new ArrayList();
                // 组装一行数据
                for (int i = 0; i < columnCount; i++) {
                    result.add(resultSet.getObject(i + 1));
                }
                resultList.add(result);

            }

            // step5 关闭所有指针
            stmt.close();
            conn.close();
        } catch (SQLException se) {
            // 处理 JDBC 错误
            se.printStackTrace();
        } catch (Exception e) {
            // 处理 Class.forName 错误
            e.printStackTrace();
        } finally {
            // 关闭资源
            try {
                if (stmt != null) stmt.close();
            } catch (SQLException se2) {
            }// 什么都不做
            try {
                if (conn != null) conn.close();
            } catch (SQLException se) {
                se.printStackTrace();
            }
        }
        System.out.println("Goodbye!");

        return resultList;
    }

    @Test
    public void test() {
        String sql = "sql查询语句";
        List<ArrayList> arrayLists = dataSearch(sql);
        for (int i = 0; i < arrayLists.size(); i++) {
            System.out.println(String.format("第%s行数据：%s", i, arrayLists.get(i)));
        }
    }
}