JDBCIO调用Postgres例程(存储过程),该例程将自定义对象类型作为参数

时间:2019-03-26 22:00:54

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

我正在尝试调用Postgres例程,该例程将自定义Object类型作为参数。

create type person_type as
(
  first        varchar,
  second varchar,
  is_real     boolean
);

我的例程(存储的proc):

create function person_routine(person person_type)
returns void
language plpgsql
as $$

BEGIN
  INSERT INTO person(first, second, is_real) VALUES 
(person.first,person.second,person.is_real);

END;
$$;

然后我尝试创建一个Java类来表示自定义类型:

import java.sql.SQLData;
import java.sql.SQLException;
import java.sql.SQLInput;
import java.sql.SQLOutput;

public class PersonType implements SQLData {

public String first;
public String second;
public boolean is_real;
private String sql_type;

@Override
public String getSQLTypeName() throws SQLException {
    return sql_type;
}

@Override
public void readSQL(SQLInput stream, String typeName) throws SQLException {
    sql_type = typeName;
    second = stream.readString();
    first = stream.readString();
    is_real = stream.readBoolean();
}

@Override
public void writeSQL(SQLOutput stream) throws SQLException {
    stream.writeString(first);
    stream.writeBoolean(is_real);
    stream.writeString(second);
}
}

然后我尝试执行如下代码:

.apply(JdbcIO.<Person>write()
                            .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(
                                    "org.postgresql.Driver", configuration.GetValue("postgres_host"))
                                    .withUsername(configuration.GetValue("postgres_username"))
                                    .withPassword(configuration.GetValue("postgres_password")))
                            .withStatement("SELECT person_routine(?)")
                            .withPreparedStatementSetter(new JdbcIO.PreparedStatementSetter<Person>() {
                                public void setParameters(Person element, PreparedStatement query)
                                        throws SQLException {
                                    PersonType dto = new PersonType();
                                    dto.first = element.first;
                                    dto.second = element.second;
                                    dto.is_real = element.is_real;
                                    query.setObject(1, dto);
                                }
                            })
                    );

不幸的是,这给了我一个例外:

java.lang.RuntimeException: org.postgresql.util.PSQLException: Can't infer the SQL type to use for an instance of dto.PersonType. Use setObject() with an explicit Types value to specify the type to use.

任何帮助都会很棒。

1 个答案:

答案 0 :(得分:0)

因此,它与使用PGobject()有关。这就是我实现它的方式,并且看起来工作得很好。

 PGobject person = new PGobject();
 person.setType("person_type");
 person.setValue(String.format("(%s,%s,%s)","something","something","FALSE"));
 query.setObject(1, person);