jdbi返回插入的自动生成值

时间:2014-10-01 03:40:49

标签: dropwizard jdbi

我正在使用dropwizard,我想构建一个在实体中具有各种外键关系的REST应用程序。

例如,给出以下3个表:

-- table persons
CREATE TABLE PUBLIC.PERSONS(
    ID BIGINT DEFAULT NOT NULL AUTOINCREMENT,
    FIRST_NAME VARCHAR(255),
    LAST_NAME VARCHAR(255),
    BIRTHDAY DATE,
    ADDRESS_ID BIGINT NOT NULL,
    CREATED TIMESTAMP DEFAULT 'current_timestamp',
    MODIFIED TIMESTAMP
);
ALTER TABLE PUBLIC.PERSONS ADD CONSTRAINT PUBLIC.PK_PERSONS PRIMARY KEY(ID);

-- table customers
CREATE TABLE PUBLIC.CUSTOMERS(
    ID BIGINT DEFAULT NOT NULL AUTOINCREMENT,
    PERSON_ID BIGINT NOT NULL,
    STATUS_CODE VARCHAR(100) DEFAULT 'ACQUISITION' NOT NULL,
    CREATED TIMESTAMP DEFAULT 'current_timestamp',
    MODIFIED TIMESTAMP
);
ALTER TABLE PUBLIC.CUSTOMERS ADD CONSTRAINT PUBLIC.PK_CUSTOMERS PRIMARY KEY(ID);

-- table addresses
CREATE TABLE PUBLIC.ADDRESSES(
    ID BIGINT DEFAULT NOT NULL AUTOINCREMENT,
    LINE_1 VARCHAR(255),
    LINE_2 VARCHAR(255),
    ZIP VARCHAR(255),
    CITY VARCHAR(255),
    COUNTRY_CODE VARCHAR(3),
    PHONE VARCHAR(255),
    FAX VARCHAR(255),
    CREATED TIMESTAMP DEFAULT 'current_timestamp',
    MODIFIED TIMESTAMP
);
ALTER TABLE PUBLIC.ADDRESSES ADD CONSTRAINT PUBLIC.PK_ADDRESSES PRIMARY KEY(ID);

-- and following forign key constraints:
ALTER TABLE PUBLIC.PERSONS ADD CONSTRAINT 
    PUBLIC.FK_PERSON_ADDRESS FOREIGN KEY(ADDRESS_ID) 
    REFERENCES PUBLIC.ADDRESSES(ID) ON DELETE SET NULL NOCHECK;

ALTER TABLE PUBLIC.CUSTOMERS ADD CONSTRAINT 
    PUBLIC.FK_CUSTOMER_PERSON FOREIGN KEY(PERSON_ID) 
    REFERENCES PUBLIC.PERSONS(ID) ON DELETE SET NULL NOCHECK;

我已经开始实现一个customerDAO,它读取客户表的数据和带有一个SQL查询的引用表,这不是很复杂:

@RegisterMapper(CustomerResultMapper.class)
public interface CustomerDAO {

    @SqlQuery("select p.id as person_id, "
            + "p.first_name, p.last_name, p.birthday, "
            + "c.id as customer_id, c.status_code, "
            + "a.id as address_id, a.line_1, a.line_2, a.zip, "
            + "a.city, a.country_code, a.phone, a.fax "
            + "from customers c, persons p, addresses a "
            + "where c.id = :id "
            + "and c.person_id = p.id "
            + "and p.address_id = a.id")
    public Customer findCustomerById(@Bind("id") long id);
}

(为了简洁起见,我跳过了映射器,因为那不是我的实际问题)

现在我想插入一个新客户,我拥有所有必需的数据,包括属于引用表的数据。

我找不到如何使用jdbi注释执行多个查询的方法,所以我想,我必须为每个表创建一个DAO方法并从java中插入数据,手动更新外键引用。

但这也行不通,因为在插入后我无法找到读取自动生成ID值的方法。

我知道如何处理这个问题,以便我能保持参考文献正确吗?

2 个答案:

答案 0 :(得分:36)

没关系,我在此期间找到了解决方案,至少如何获取自动生成的密钥:

在你的DAO中

添加@GetGeneratedKeys注释,确保方法的返回值与生成的键的值匹配:

@SqlUpdate("insert into addresses (line_1) values ('address line 1')")
@GetGeneratedKeys
public long insertAddress();

在您的资源中,您只需使用返回值:

long id = customerDAO.insertAddress();
System.out.println("Found id " + id);

这可以获取一个语句的生成密钥,但在我的情况下,我仍然需要进行多个查询并手动填写正确的引用。

如果有人知道如何简化此过程并自动填写参考文献,我仍然有兴趣了解。

感谢。

答案 1 :(得分:16)

使用postgres,您不仅可以获得一个,甚至可以获得所有自动生成的值。

您可以通过将@SqlUpdate更改为@SqlQuery并使用INSERT ... RETURNING *来实现,假设您有一个地址映射器:

@SqlQuery("insert into addresses (line_1) values ('address line 1') returning *")
public Address insertAddress(@BindBean Address address);