0

我正在使用 Diesel 使用 JOIN 查询 PostgreSQL 中的数据库:

let product_id = 1;
sales::table
     .inner_join(product::table)
     .select((
         product::description,
         sales::amount,
         sales::date_sale
     ))
     .filter(sales::product_id.eq(product_id))
     .load(&diesel::PgConnection)

我的模型:

pub struct Sales {
    pub id: i32,
    pub product_id: Option<i32>,
    pub amount: Option<BigDecimal>,
    pub date_sale: Option<NaiveDateTime>
}

结果符合预期,但我需要为sales::date_salepgadmin 中使用to_char(date_sale, 'dd/mm/YYYY').

是否可以to_char在 Diesel 中使用,或者我可以通过什么方式修改 Diesel ORM 带给我的数据?

4

2 回答 2

0

当使用 ORM 时,数据以最适合 ORM 解释的表示从数据库中提取;之后,您将在目标域中对其进行操作(在本例中为 rust)。

由于date_saleis an Option<NaiveDateTime>,您可以使用chrono提供的格式化选项:

sale.date_sale.unwrap().format("%d/%m/%Y").to_string()

(当然,您的真实代码不会使用unwrap()!)

或者,如果您确实需要数据库进行格式化,您可以使用 a[sql][2]在查询中插入一些原始 SQL:

let results = sales::table.select((sales::id, sql("to_char(date_sale, 'dd/mm/YYYY')")))
    .load::<(i32, String)>(&conn);

如果您尝试实现一些在 SQL 中更容易或更有效地实现的逻辑,这将更加有用。

一个例子是一个过滤条件:假设你想从你的表中包含周数是偶数的行。虽然您可以加载整个表,然后在 rust 域中对其进行过滤,但这并不是很有效。相反,您可以这样做:

let results = sales::table
    .filter(sql("extract(week from date_sale)::smallint % 2=0"))
    .load::<Sales>(&conn);
于 2020-04-24T23:47:01.420 回答
0

除了提供的答案之外,harmic还有两种解决此问题的可能性。

sql_function!

Diesel 提供了一个接口,可以轻松地为柴油本身不提供的 sql 函数定义查询 ast 节点。鼓励用户使用此功能自行定义缺失的功能。(事实上​​,diesel 在内部使用相同的方法为开箱即用的 sql 函数定义查询 ast 节点)。定义的查询 ast 节点在其表达式类型有效的每个上下文中都可用,因此它可以在 select 和 where 子句中使用。(这基本上是上面原始 sql 解决方案的类型安全版本)

对于给定的问题,这样的事情应该有效:

#[derive(Queryable)]
pub struct Sales {
    pub id: i32,
    pub product_id: Option<i32>,
    pub amount: Option<BigDecimal>,
    pub date_sale: Option<String>,
}

sql_function! {
    fn to_char(Nullable<Timestamp>, Text) -> Nullable<Text>;
}

let product_id = 1;
sales::table
     .inner_join(product::table)
     .select((
         product::description,
         sales::amount,
         to_char(sales::date_sale, "dd/mm/YYYY")
     ))
     .filter(sales::product_id.eq(product_id))
     .load(&diesel::PgConnection);

#[derive(Queryable)]+#[diesel(deserialize_as = "Type")]

Diesels derivedQueryable提供了一种在加载时通​​过自定义属性应用某些类型操作的方法。结合由此chrono提供的解决方案harmic给出以下变体:

#[derive(Queryable)]
pub struct Sales {
    pub id: i32,
    pub product_id: Option<i32>,
    pub amount: Option<BigDecimal>,
    #[diesel(deserialize_as = "MyChronoTypeLoader")]
    pub date_sale: Option<String>
}

struct MyChronoTypeLoader(Option<String>);

impl Into<Option<String>> for MyChronoTypeLoader {
    fn into(self) -> String {
        self.0
    }
}

impl<DB, ST> Queryable<ST, DB> for MyChronoTypeLoader
where
    DB: Backend,
    Option<NaiveDateTime>: Queryable<ST, DB>,
{
    type Row = <Option<NaiveDateTime> as Queryable<ST, DB>>::Row;

    fn build(row: Self::Row) -> Self {
        MyChronoTypeLoader(Option::<NaiveDateTime>::build(row).map(|d| d.format("%d/%m/%Y").to_string()))
    }
}
于 2020-04-27T11:41:26.903 回答