如何在多线程中插入数据库 java

Question

我有一个任务。我需要随机生成 1 到 1000000 之间的数字，并将其插入数据库中名为 'numbers' 的 table。 Table 'numbers' 有两列：'number' 和 'quantity' 如果数据库中已经存在数字，我需要在 table 中增加数量。我试过这样：

public void insertRow(String number) throws SQLException {
        int cnt = getCount(number);
        if (cnt == 0) {
            insert(number);
        } else if (cnt > 0) {
            update(number);
        }
    }

getCount 检查数据库中是否存在数字：

private int getCount(String number) throws SQLException {
        int cnt = 0;
        String sql = "select count(number) as cnt from \"PUBLIC\".UNIQUE_NUMBER where number='" + number + "'";
        Statement sta = getConnection().createStatement();
        ResultSet rs = sta.executeQuery(sql);
        if (rs.next()) {
            cnt = rs.getInt("cnt");
        }

        return cnt;
    }

正在插入数据库：

private boolean insert(String number) throws SQLException {
        String sql = "insert into \"PUBLIC\".UNIQUE_NUMBER (number, qty) values(?, ?)";
        PreparedStatement ps = getConnection().prepareStatement(sql);
        synchronized (this) {
            try {
                getConnection().setAutoCommit(false);
                ps.setString(1, number);
                ps.setInt(2, 0);
                ps.addBatch();
                ps.executeBatch();
                getConnection().commit();

            } catch (Exception e) {
                LOGGER.error(e.toString());
                try {
                    getConnection().rollback();
                } catch (Exception ex) {
                    LOGGER.error(ex.toString());
                }
                return false;
            } finally {
                if (ps != null) {
                    ps.close();
                }
                getConnection().setAutoCommit(true);
            }
        }
        return true;
    }

并更新：

private boolean update(String number) throws SQLException {
        String sql = "update \"PUBLIC\".UNIQUE_NUMBER set (qty) = (?) where number = ?";
        int qty = selectQtyByNumber(number) + 1;
        PreparedStatement ps = getConnection().prepareStatement(sql);
        synchronized (this) {
            try {
                getConnection().setAutoCommit(false);
                ps.setInt(1, qty);
                ps.setString(2, number);
                ps.executeUpdate();
                getConnection().commit();
            } catch (Exception e) {
                LOGGER.error(e.toString());
                try {
                    getConnection().rollback();
                } catch (Exception ex) {
                    LOGGER.error(ex.toString());
                }
                return false;
            } finally {
                if (ps != null) {
                    ps.close();
                }
                getConnection().setAutoCommit(true);
            }
        }
        return true;
    }

SelectQtyByNumber方法获取当前数量

private int selectQtyByNumber(String number) throws SQLException {
        int qty = 0;
        String sql = "select qty from \"PUBLIC\".UNIQUE_NUMBER where number='" + number + "'";
        Statement sta = getConnection().createStatement();
        ResultSet rs = sta.executeQuery(sql);
        if (rs.next()) {
            qty = rs.getInt("qty");
        }
        return qty;
    }

当我运行这段代码时，我总是得到 BatchUpdateException：违反完整性约束。但是一些数字的数量增加了。为什么会这样？如何处理？数据库是 HSQLDB。

这是我的数据库构造函数和 getConnection 方法

public Database(String url, String user_name, String password) {
        try {
            Class.forName("org.hsqldb.jdbc.JDBCDriver");

            this.connection = DriverManager.getConnection(url, user_name, password);

        } catch (ClassNotFoundException | SQLException e) {
            // TODO Auto-generated catch block
            LOGGER.error("Database initialization exception: " + e.toString());
        }
    }

    public Connection getConnection() {
        return this.connection;
    }

Answer 1

你的问题是 JDBC 包类不是线程安全的。如果您在线程之间共享连接，就会遇到问题。

有关连接获取的细节至关重要。您将它们隐藏在 getConnection() 方法中。他们汇集了吗？每个操作都会得到一个全新的 Connection 实例吗？那些被关闭的在哪里？

你的第二个问题是孤立。您需要将连接隔离设置为可能的最高级别：SERIALIZABLE。你不能正确地交错读取和写入。

Answer 2

您过于依赖 Java 代码。您的代码可以简化，但是您可以使用 MERGE 语句的完全不同的方法并允许 HSQLDB 处理整个事情。此语句可与自动提交一起使用并且是线程安全的。

MERGE INTO public.unique_number
USING VALUES (CAST(? AS VARCHAR(100)), CAST(? AS INTEGER)) AS vals(number, quantity)
ON unique_number.number = vals.number
WHEN MATCHED THEN
UPDATE SET number = vals.number
WHEN NOT MATCHED THEN
INSERT (number, quantity) VALUES vals.number, valse.quantity;

查看指南 http://hsqldb.org/doc/guide/dataaccess-chapt.html#dac_merge_statement

如何在多线程中插入数据库 java

How to insert in database inside multi threads java

java

multithreading

insert

hsqldb