在柴油机中执行右连接

时间:2020-01-23 13:32:14

标签: postgresql rust rust-diesel

我有一个函数,它接受两个可选的过滤参数,如果提供了这些参数,则过滤表中的数据。为此,我创建了一个框式查询。我想使用此框式查询创建右联接。 Diesel的当前文档没有提及正确的联接,但似乎更喜欢belonging_to方法。我将代码简化为一个示例:

#[macro_use] extern crate diesel;

use diesel::prelude::*;

table! {
    groups (id) {
        id -> Int4,
        name -> Text,
    }
}

table! {
    user_groups (id) {
        id -> Int4,
        user_id -> Int4,
        group_id -> Int4,
    }
}

allow_tables_to_appear_in_same_query!(groups, user_groups);

#[derive(Debug, Queryable)]
struct Group {
    id: i32,
    name: String,
}

#[derive(Debug, Queryable, Associations)]
#[belongs_to(Group)]
struct UserGroup {
    id: i32,
    user_id: i32,
    group_id: i32,
}

fn filter(
    name: Option<&str>,
    user_id: Option<i32>,
    conn: &diesel::PgConnection,
) -> Result<Vec<(Group, Vec<UserGroup>)>, Box<dyn std::error::Error>> {
    let mut query = groups::table
        .right_join(user_groups::table.on(groups::id.eq(user_groups::group_id))) // this method does not exist
        .into_boxed();
    if let Some(name) = name {
        query = query.filter(groups::name.eq(name));
    }
    if let Some(user_id) = user_id {
        query = query.filter(user_groups::user_id.contains(user_id)); // so this is just a guess
    }
    Ok(query.get_results(conn)?)
}

fn main() {
    let url = "postgres://question_usr:question_pwd:localhost:5432/question_db";
    let conn = diesel::PgConnection::establish(url).unwrap();
    let _ = filter(Some("groupname"), Some(4), &conn).unwrap();
}

目的是,如果指定了参数user_id,则仅返回Groups的行,这些行至少具有一个UserGroup这样user_group.user_id == user_id。如何运行此查询?我需要以某种方式使用belonging_to函数吗?

1 个答案:

答案 0 :(得分:8)

在每种情况下,右联接都等效于具有反向表顺序的左联接。 (例如,请参见here)。因此,Diesel仅提供构造LEFT JOIN的功能。

您的示例如下所示:

fn filter(
    name: Option<&str>,
    user_id: Option<i32>,
    conn: &diesel::PgConnection,
) -> Result<Vec<(UserGroup, Option<Group>)>, Box<dyn std::error::Error>> {
    let mut query = user_groups::table
        .left_join(groups::table.on(groups::id.eq(user_groups::group_id))) /
        .into_boxed();
    if let Some(name) = name {
        query = query.filter(groups::name.eq(name));
    }
    if let Some(user_id) = user_id {
        query = query.filter(user_groups::user_id.eq(user_id)); 
    }
    Ok(query.get_results(conn)?)
}

请注意,默认情况下会返回Vec<(UserGroup, Option<Group>)>。如果您需要其他内容,则需要编写一个自定义的select子句,以指定所需的结果形状。