比对两表数据差异的简易工具类

一个简易的能直接运行同结构库表数据比对的工具类。

package org.example.sql;

import java.sql.*;
import java.util.*;

/**
 * @Author: linK
 * @Date: 2022/6/28 15:07
 * @Description 测试两表比对
 */
public class Test {


    @org.junit.Test
    public void test() throws ClassNotFoundException, SQLException {

        // 获取字段sql
        ResultSet resultColumn = getResultColumn("jdbc:mysql://localhost:3306/test?useSSL=false", "root", "199911", "test", "test");
        // 获取表中所有字段集合
        List column = getColumn(resultColumn);

        // 获取数据
        ResultSet oldResultData = getResultData("jdbc:mysql://localhost:3306/test?useSSL=false", "root", "199911", "test", "test");

        ResultSet newResultData = getResultData("jdbc:mysql://localhost:3306/test?useSSL=false", "root", "199911", "test", "test1");

        Map> oldMap = getData(oldResultData, column);
        Map> newMap = getData(newResultData, column);

        // 比对两表数据差异
        List diff = getDiff(oldMap, newMap);
        diff.forEach(System.out::println);

    }

    /**
     * 比对差异
     *
     * @param oldMap 旧表数据map
     * @param newMap 新表数据map
     * @return
     */
    public List getDiff(Map> oldMap, Map> newMap) {
        List diffList = new ArrayList<>();
        oldMap.keySet().forEach(m -> {
            // 字符串差异
            StringBuilder diffStr = new StringBuilder("");
            if (newMap.containsKey(m)) {
                Map oldData = oldMap.get(m);
                Map newData = newMap.get(m);

                for (String old : oldData.keySet()) {
                    // 同时为空 跳过比对
                    if (oldData.get(old) == null && newData.get(old) == null) {
                        continue;
                    }
                    if (oldData.get(old) == null && newData.get(old) != null) {
                        diffStr.append("所在行数据:").append(m).append("，列名称：").append(old).append("   旧值为空：").append("   新值：").append(newData.get(old));
                        continue;
                    }
                    if (oldData.get(old) != null && newData.get(old) == null) {
                        diffStr.append("所在行数据:").append(m).append("，列名称：").append(old).append("   旧值：").append(oldData.get(old)).append("   新值为空");
                        continue;
                    }
                    // 比对列的数据
                    if (oldData.get(old).equals(newData.get(old))) {
                        // 结果相等跳过
                    } else {
                        diffStr.append("所在行数据:").append(m).append("，列名称：").append(old).append("   旧值：").append(oldData.get(old)).append("   新值：").append(newData.get(old));
                    }
                }
                // 数据比对信息不为空 添加进list
                if (!Objects.isNull(diffStr)) {
                    diffList.add(diffStr.toString());
                }
            } else {
                diffList.add(m + "行数据newMap为空");
            }


        });
        return diffList;
    }

    public List getColumn(ResultSet rs) throws SQLException {
        ArrayList column = new ArrayList<>();
        while (rs.next()) {
            column.add(rs.getString("column_name"));
        }
        return column;
    }

    /**
     * 获取行数据
     *
     * @param result
     * @param column
     * @return
     * @throws SQLException
     */
    public Map> getData(ResultSet result, List column) throws SQLException {
        Map> stringMapHashMap = new HashMap<>();
        ResultSetMetaData metaData = result.getMetaData();

        while (result.next()) {
            // 字段数据
            HashMap columnData = new HashMap<>();
            column.forEach(m -> {
                try {
                    columnData.put(m, result.getString(m));
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            });
            // result.getString("cur_date")  暂时字符串截取
            String cur_date = result.getString("cur_date").substring(0, 10);
            stringMapHashMap.put(cur_date.concat(result.getString("city")), columnData);

        }
        return stringMapHashMap;
    }

    /**
     * 获取字段结果集
     *
     * @param url
     * @param user
     * @param password
     * @param db
     * @param table
     * @return
     * @throws ClassNotFoundException
     * @throws SQLException
     */
    public ResultSet getResultColumn(String url, String user, String password, String db, String table) throws ClassNotFoundException, SQLException {
        //1.加载驱动程序
        Class.forName("com.mysql.jdbc.Driver");
        //2.获得数据库的连接
        Connection conn = DriverManager.getConnection(url, user, password);
        //3.通过数据库的连接操作数据库，实现增删改查
        Statement stmt = conn.createStatement();
        ResultSet rs = stmt.executeQuery("select Table_name,column_name from information_schema.columns where table_schema= '" + db + "' and table_name= " + "'" + table + "'");

        return rs;
    }

    /**
     * 查询数据
     *
     * @param url
     * @param user
     * @param password
     * @param db
     * @param table
     * @return
     * @throws ClassNotFoundException
     * @throws SQLException
     */
    public ResultSet getResultData(String url, String user, String password, String db, String table) throws ClassNotFoundException, SQLException {
        //1.加载驱动程序
        Class.forName("com.mysql.jdbc.Driver");
        //2.获得数据库的连接
        Connection conn = DriverManager.getConnection(url, user, password);
        //3.通过数据库的连接操作数据库，实现增删改查
        Statement stmt = conn.createStatement();

        ResultSet rs = stmt.executeQuery("select * from " + db + "." + table);


        return rs;
    }


}

  

用到的maven依赖

    
        
            junit
            junit
            4.13.2
        
        
            mysql
            mysql-connector-java
            5.1.44