gpt4 book ai didi

rust - 如何在 Diesel 中针对 Postgres 数据库执行带有子查询的删除?

转载 作者:行者123 更新时间:2023-12-03 11:27:39 24 4
gpt4 key购买 nike

我在 Postgres 数据库中有以下架构:

Table A {
ID
Name
}

Table B {
ID FOREIGN KEY (A.ID)
}

我正在尝试在 Diesel 中编写以下查询:

DELETE FROM B
WHERE B.ID in (SELECT ID from A WHERE A.Name = $VAR)
$VAR是我的应用程序传递的变量。

我第一次尝试写这篇文章如下:
fn deleteB(conn: &PgConnection, n: &str) {
use schema::A::dsl::*;
use schema::A;
use schema::B;

let res = A::table
.filter(Name.eq(n))
.select(ID);
.load(conn)
.unwrap();
assert!(res.len() < 2);
let b_id: i32 = *res.iter().nth(1).unwrap_or(&0);

let _rows = diesel::delete(
B::table
.filter(ID.eq(n_id))
)
.execute(conn)
.unwrap();
}

这编译但它不起作用: SELECT ID 的语句总是返回 0。它不匹配 A 中任何插入的记录,即使我手动检查它们在那里。我确定我在完成比赛的方式上有错误( &str vs &String 也许?),但我决定尝试不同的解决方案,因为我不喜欢这个解决方案,因为它必须执行针对数据库的两个单独的语句。

我的第二次尝试是这样的:
fn deleteB(conn: &PgConnection, n: &str) {
use schema::A::dsl::*;
use schema::A;
use schema::B;

let source = B::table.inner_join(A::table)
.filter(Name.eq(n));
let s = delete(source)
.execute(conn)
.unwrap();
}

这对我来说看起来更符合 Diesel 习惯(剧透警告,我几乎不知道框架)但可以肯定的是,它不会编译:

    |
410 | let s = delete(source)
| ^^^^^^ the trait `diesel::query_builder::IntoUpdateTarget` is not implemented for `diesel::query_builder::SelectStatement<diesel::query_source::joins::JoinOn<diesel::query_source::joins::Join<mobility2::schema::B::table...` (very long error)

当我意识到这似乎是一个微不足道的操作(删除本身)时,我正在研究上述特征,我可能最好寻求帮助。

如何正确编写惯用的 Diesel 代码来执行我要执行的删除语句?

最佳答案

由于 Diesel 语句从字面上映射到 SQL,因此您编写的查询将导致:

DELETE FROM TABLE B INNER JOIN TABLE A ON … WHERE name = $1

这不是有效的 SQL,因此会导致编译时错误。

要获得您要编写的查询,您需要执行以下操作:

#[macro_use]
extern crate diesel;

use diesel::delete;
use diesel::prelude::*;

mod schema {
table! {
A(ID) {
ID -> Integer,
name -> Text,
}
}

table! {
B(ID) {
ID -> Integer,
}
}

allow_tables_to_appear_in_same_query!(A, B);
}

fn deleteB(conn: &PgConnection, n: &str) {
use schema::A;
use schema::B;

let s = delete(B::table)
.filter(B::ID.eq_any(A::table.filter(A::name.eq(n)).select(A::ID)))
.execute(conn)
.unwrap();
}

关于rust - 如何在 Diesel 中针对 Postgres 数据库执行带有子查询的删除?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60836040/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com