如果 Java 生产代码中不存在数据库表，则创建数据库表并在 JUnit 中确认

Question

我正在用 Java 编写一个数据库程序，如果它不存在，我想创建一个表。我从How can I detect a SQL table's existence in Java? 了解到DatabaseMetaData.getTables()，我正在尝试使用它：

private boolean tableExists() throws SQLException {
    System.out.println("tableExists()");

    DatabaseMetaData dbmd = conn.getMetaData();
    ResultSet rs = dbmd.getTables(null, null, this.getTableName(), null);

    System.out.println("TABLE_NAME: " + rs.getString("TABLE_NAME"));

    return rs.getRow() == 1;
}

问题是rs.getRow() 总是返回0，即使在表创建之后也是如此。使用rs.getString("TABLE_NAME") 会引发异常，指出结果集为空。

我想到的一个可能的解决方案是执行CREATE TABLE 语句并捕获任何抛出的异常。但是，我不喜欢在程序的控制流中使用异常的想法。

FWIW，我正在使用 HSQLDB。但是，我想编写独立于 RDMS 引擎的 Java 代码。还有其他方法可以使用DatabaseMetaData.getTables() 来做我想做的事吗？还是有其他解决方案来编写我的tableExists() 方法？

添加：

使用此处给出的建议，我找到了一个似乎适用于我的生产代码的解决方案：

private void createTable() throws SQLException {
    String sqlCreate = "CREATE TABLE IF NOT EXISTS " + this.getTableName()
            + "  (brand           VARCHAR(10),"
            + "   year            INTEGER,"
            + "   number          INTEGER,"
            + "   value           INTEGER,"
            + "   card_count           INTEGER,"
            + "   player_name     VARCHAR(50),"
            + "   player_position VARCHAR(20))";

    Statement stmt = conn.createStatement();
    stmt.execute(sqlCreate);
}

现在我也在编写一个 JUnit 测试来断言该表确实已创建：

public void testConstructor() throws Exception {
    try (BaseballCardJDBCIO bcdb = new BaseballCardJDBCIO(this.url)) {
        String query = "SELECT count(*) FROM information_schema.system_tables WHERE table_name = '" + bcdb.getTableName() + "'";
        Connection conn = DriverManager.getConnection(this.url);
        Statement stmt = conn.createStatement();
        ResultSet rs = stmt.executeQuery(query);
        Assert.assertTrue(rs.next());
        Assert.assertEquals(1, rs.getInt(1));
        Assert.assertFalse(rs.next());
    }
}

此测试在 assertEquals() 上失败，并显示以下消息：

FAILED: expected: <1> but was: <0>

Answer 1

我找到的解决方案似乎有效：

private void createTable() throws SQLException {
    String sqlCreate = "CREATE TABLE IF NOT EXISTS " + this.getTableName()
            + "  (brand           VARCHAR(10),"
            + "   year            INTEGER,"
            + "   number          INTEGER,"
            + "   value           INTEGER,"
            + "   card_count           INTEGER,"
            + "   player_name     VARCHAR(50),"
            + "   player_position VARCHAR(20))";

    Statement stmt = conn.createStatement();
    stmt.execute(sqlCreate);
}

我必须将IF NOT EXISTS 放在我的 SQL 语句中的正确位置。

Answer 2

来自 Java 文档中的 ResultSet 定义：

ResultSet 对象维护一个指向其当前行的游标 数据。最初，光标位于第一行之前。这 next 方法将光标移动到下一行，因为它返回 当 ResultSet 对象中没有更多行时为 false，它可以是 在 while 循环中用于遍历结果集。

因此，您必须始终调用next() 方法，否则getRow() 将始终返回零，因为光标位于第一行之前。

Answer 3

内置 mysql 功能可满足您的需求：http://dev.mysql.com/doc/refman/5.0/en/create-table.html

简而言之：只需在表创建查询的末尾附加 IF NOT EXISTS。

编辑：
没有通用的方法来做到这一点。大多数数据库都有一个 information_scheme 表，确定信息的查询可能如下所示：

SELECT count(*) FROM information_schema.system_tables WHERE table_schem = 'public' AND table_name = 'user';

这适用于 sqlite、mysql、msql、mariadb 和 postgres + 可能还有很多其他的。

Answer 4

我不知道这是否一定有助于实现您的目标，但是当我使用 Python 和 MySQL 遇到这个问题时，我只是在每个“创建表”语句之前添加了一个“删除表”语句，这样就可以运行该脚本会自动删除现有表，然后重建每个表。这可能无法满足您的需求，但这是我发现成功解决类似问题的一种解决方案。