我正在尝试手动实现sqlx::FrowRow
而不是使用derive
,因为需要一些自定义初始化逻辑(例如将整数转换i16
为枚举)。
默认 impl 生成如下:
use sqlx::FromRow;
impl FromRow for User {
fn from_row(row: &'r R) -> Result<Self, sqlx::Error> {
todo!()
}
}
但目前尚不清楚,我应该使用什么生命周期和类型。
该FromRow
特征非常通用,因为它用于可能支持不同类型的不同后端数据库。您可以通过查看#[derive(FromRow)]
生成的内容(通过cargo expand
)来了解实现一个简单结构需要多少通用约束:
use sqlx::FromRow;
#[derive(FromRow)]
struct User {
name: String,
status: i16,
}
// generates
impl<'a, R: ::sqlx::Row> ::sqlx::FromRow<'a, R> for User
where
&'a ::std::primitive::str: ::sqlx::ColumnIndex<R>,
String: ::sqlx::decode::Decode<'a, R::Database>,
String: ::sqlx::types::Type<R::Database>,
i16: ::sqlx::decode::Decode<'a, R::Database>,
i16: ::sqlx::types::Type<R::Database>,
{
fn from_row(row: &'a R) -> ::sqlx::Result<Self> {
let name: String = row.try_get("name")?;
let status: i16 = row.try_get("status")?;
::std::result::Result::Ok(User { name, status })
}
}
它必须限制列可以按名称索引,并且限制列String
是i16
有效的并且可以从数据库中解码。
自己做,你最好选择你打算使用的数据库Row
类型AnyRow
( , MssqlRow
, MySqlRow
, PgRow
, SqliteRow
)并为此实现它。这里PgRow
用于 postgres:
use sqlx::{Row, FromRow, Error};
use sqlx::postgres::PgRow;
struct User {
name: String,
status: i16,
}
impl<'r> FromRow<'r, PgRow> for User {
fn from_row(row: &'r PgRow) -> Result<Self, Error> {
let name = row.try_get("name")?;
let status = row.try_get("status")?;
Ok(User{ name, status })
}
}
我确实想知道是否有不同的建议来进行自定义转换(通过 DTO 或其他机制)。