如何从仅 returns 1 或 0 条记录的 Diesel 查询中获取 Option<T> 而不是 Option<Vec<T>> ？

Question

我正在查询名为 messages 的 table 中的现有记录；然后将此查询用作 'find or create' 函数的一部分：

fn find_msg_by_uuid<'a>(conn: &PgConnection, msg_uuid: &Uuid) -> Option<Vec<Message>> {
    use schema::messages::dsl::*;
    use diesel::OptionalExtension;

    messages.filter(uuid.eq(msg_uuid))
        .limit(1)
        .load::<Message>(conn)
        .optional().unwrap()
}

我将此设为可选，因为在这种情况下查找记录和查找 none 都是有效结果，因此此查询可能 return a Vec一个 Message 或一个空的 Vec，所以我总是最终检查 Vec 是否为空或不使用这样的代码：

let extant_messages = find_msg_by_uuid(conn, message_uuid);

if !extant_messages.unwrap().is_empty() { ... }

然后如果它不为空，则将 Vec 中的第一个 Message 作为我使用

之类的代码找到的消息

let found_message = find_msg_by_uuid(conn, message_uuid).unwrap()[0];

我总是取 Vec 中的第一个元素，因为记录是唯一的，所以查询只会 return 1 或 0 条记录。

感觉有点乱，好像步骤太多了，感觉好像有查询记录那么应该return Option<Message> 不是Option<Vec<Message>>或 None 如果没有匹配查询的记录。

Answer 1

如评论中所述，使用first:

Attempts to load a single record. Returns Ok(record) if found, and Err(NotFound) if no results are returned. If the query truly is optional, you can call .optional() on the result of this to get a Result<Option<U>>.

fn find_msg_by_uuid<'a>(conn: &PgConnection, msg_uuid: &Uuid) -> Option<Message> {
    use schema::messages::dsl::*;
    use diesel::OptionalExtension;

    messages
        .filter(uuid.eq(msg_uuid))
        .first(conn)
        .optional()
        .unwrap()
}

如何从仅 returns 1 或 0 条记录的 Diesel 查询中获取 Option<T> 而不是 Option<Vec<T>> ？

How do I get an Option<T> instead of an Option<Vec<T>> from a Diesel query which only returns 1 or 0 records?

rust

rust-diesel