如何使用JDBCIO执行存储过程/例程(Apache Beam)

时间:2019-03-21 10:06:51

标签: java postgresql jdbc apache-beam apache-beam-io

我正在尝试使用JDBCIO的Apache Beam执行postgres例程。

到目前为止,我已经尝试过:

 .apply(JdbcIO.<MyData>write()
                            .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(
                                    "org.postgresql.Driver", "jdbc:postgresql://localhost:5432/postgres")
                                    .withUsername("postgres")
                                    .withPassword("password"))

                            .withStatement("do $$\n" +
                                    "begin " +
                                    "perform test_routine(first := ?, second := ?, age := ?) " +
                                    "end\n" +
                                    "$$;")
                            .withPreparedStatementSetter(new JdbcIO.PreparedStatementSetter<MyData>() {
                                public void setParameters(MyData element, PreparedStatement query)
                                        throws SQLException {
                                    query.setString(1,element.mfirst);
                                    query.setString(2, element.second);
                                    query.setInt(3, element.age);
                                }
                            })
                    );

不幸的是,这给了我错误:

org.postgresql.util.PSQLException: The column index is out of range: 1, number of columns: 0

我已经设法通过一个简单的插入语句使它工作,但是理想情况下希望调用一个例程。任何帮助,将不胜感激。

1 个答案:

答案 0 :(得分:1)

非常简单:

.withStatement("select test_routine(first := ?, second := ?, age := ?))