Question

我有一项任务。我需要随机生成介于1和1000000之间的数字，并将其插入数据库中名为“数字”的表中。表'数字'有两列：'数字'和'数量' 如果数据库中已经存在的数字，我需要在表中增加数量。我已经用这种方式尝试过了：

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

您过分依赖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

Answer 2

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

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

你的第二个问题是隔离。您需要将连接隔离设置为可能的最高级别：SERIALIZABLE。您无法正确交错读写。

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

2 个答案: