如何在执行JDBC语句时收集警告/通知?

时间:2017-06-23 12:40:19

标签: java postgresql jdbc postgresql-9.4

我遇到了最新版本的PostgreSQL JDBC驱动程序的问题:在准备好的语句仍在执行时,我无法收集警告/通知。警告列表仅在语句返回后变为可用。

我将此功能与之前的一些驱动程序版本(我猜9.1)一起使用,但PgPreparedStatement的实现可能从那时起发生了变化。

在返回结果之前,我有哪些收集警告的选项?

我创建了这个简单的测试:

public class WarningTest {

    @Test
    public void readWarnings() throws Exception {
        Connection con = DriverManager.getConnection("jdbc:postgresql://localhost:5432/test_db", "test_user", "test_password");
        createTestFunction(con);

        PreparedStatement statement = con.prepareStatement("SELECT test_function();");
        SQLWarning[] warnings = new SQLWarning[1];
        new Timer().schedule(new TimerTask() {
            @Override public void run() {
                try {
                    warnings[0] = statement.getWarnings();
                } catch (SQLException e) {
                    Assert.fail("Exception thrown: " + e.getMessage());
                }
            }
        }, 3500);
        statement.executeQuery();
        Thread.sleep(1000);
        statement.close();

        Assert.assertNotNull(warnings[0]);
        Assert.assertFalse(warnings[0].getMessage().isEmpty());
    }

    private void createTestFunction(Connection con) throws SQLException {
        final PreparedStatement statement = con.prepareStatement(
            "CREATE OR REPLACE FUNCTION test_function() RETURNS VOID AS\n"
            + "$BODY$\n"
            + "BEGIN\n"
            + "\tFOR i IN 1..3 LOOP \n"
            + "\t\tRAISE NOTICE 'Tick %', i;\n"
            + "\t\tEXECUTE pg_sleep(1);\n"
            + "\tEND LOOP;\n"
            + "END\n"
            + "$BODY$\n"
            + "\tLANGUAGE plpgsql STABLE;");
        statement.execute();
        statement.close();
    }
}

它创建一个运行3秒的函数,并在每秒开始时写一个勾号。

在返回结果后,测试通过这样的计时器。但是,如果将计时器延迟从3500减少到1500,则即使数据库在此时发出2个通知,测试也会失败。

1 个答案:

答案 0 :(得分:7)

此问题是在REL9.4.1210个版本REL9.4.1209中引入的,而且工作情况与预期不符。

Line 244 of this commit之所以不再有效,警告会在收到后立即直接添加到声明中,但它已经重新计算,现在只有在完成后才可用。

I have raised an issue for this,以及PR to address it.