RustのORMであるDieselを使ってデータの取得・削除を一括で行いたい

Question

前提

dieselを使って以下のHumansテーブルに対し、一括でfetch, deleteを行いたいです。(PostgresSQL v.14)

CREATE TABLE Humans (
    name VARCHAR(20) PRIMARY KEY,
    information JSON NOT NULL
);

#[derive(Insertable, Queryable, Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
#[table_name = "humans"]
pub struct Humans {
    pub name: String,
    pub information: serde_json::Value,
}

現在はいずれも以下のようにイテレータやfor文を使って実装しています。

pub fn fetch_humans(conn: &PgConnection, targets: &[String]) -> Vec<Option<Vec<Nodes>>> {
    targets.iter()
        .map(|target| {
            schema::humans::dsl::humans
                .filter(schema::humans::name.eq(target))
                .load::<Humans>(conn)
                .optional()
                .expect("Error fetching humans")
        })
        .collect::<Vec<_>>()
}

pub fn delete_humans(conn: &PgConnection, targets: &[String]) {
    for target in targets {
        diesel::delete(schema::humans::dsl::humans)
            .filter(schema::humans::name.eq(target))
            .execute(conn)
            .expect("Error deleting humans");
    }
}

やりたいこと

この実装でも問題ないのですが、クエリ回数を減らし、より効率的に処理したいと思っています。

diesel::insert_intoのように実現できれば嬉しいのですが、fetchやdeleteに関してはそのようなメソッドが用意されていません。(私が調べた限り)

考えたこと

diesel::sql_queryの中でWITH句を使った生のSQLを書けばできるかなと思ったのですが、targetsの中身を取り出してsql_queryの中に書き込む方法が分からず停滞しています。

もし、diesel固有の解決策やsql_queryでの解決策に知見がある方がいらっしゃれば、ご意見いただきたいです。

Answer 1

eqの代わりにeq_anyを使えば一つのSELECTまたはDELETE文で複数の値を指定できます。SQLのWHERE 〜 IN ()に対応しています。（ドキュメント）

// 注：戻り値型を変更しました
pub fn fetch_humans(conn: &mut PgConnection, targets: &[String]) -> Vec<Humans> {
    schema::humans::dsl::humans
        .filter(schema::humans::name.eq_any(targets))
        .load::<Humans>(conn)
        .expect("Error fetching humans")
}

pub fn delete_humans(conn: &mut PgConnection, targets: &[String]) {
    diesel::delete(schema::humans::dsl::humans)
        .filter(schema::humans::name.eq_any(targets))
        .execute(conn)
        .expect("Error deleting humans");
}

なお、Dieselは、PostgreSQLではIN ()の代わりに= ANY()を使うそうです。

On PostgreSQL, this method automatically performs a = ANY() query.

以下のようにしてSQLをデバッグ表示したところ、たしかに= ANY()が使われていました。

pub fn fetch_humans(conn: &mut PgConnection, targets: &[String]) -> Vec<Humans> {
    use diesel::pg::Pg;

    let query = schema::humans::dsl::humans
        .filter(schema::humans::name.eq_any(targets));
    let sql = diesel::debug_query::<Pg, _>(&query);
    println!("{}", sql);

SELECT "humans"."name", "humans"."information" FROM "humans" WHERE ("humans"."name" = ANY($1)) -- binds: [["alice", "cindy", "david"]]

Comments

[ts21]

ありがとうございます！eq_anyは見逃していました...
機能を損なわずにスマートなクエリとすることができました!

[ts21]

ちなみに戻り値をそのままにするなら
.optional().iter().collect::<Vec<_>>()とすれば近づけそうです。
(参照＆Optionとなってしまうので、何かしらの対策が必要）