如何通过Trait简化Sqlx类型转自定义Enum的重复代码并兼容多数据库？

现有Rust代码通过匹配PgRow的type_info.name()，将Sqlx查询结果转换为自定义枚举MyDynamicValue，但每个数据库类型都需单独编写match分支，代码重复度极高。期望将重复逻辑简化为row.try_get(col_index).into()的形式，此前尝试实现From<dyn Decode<'a, T, DB>>未成功，需解决代码重复问题并实现对任意数据库类型的兼容。

当前示例代码：

fn get_row_value(row: &PgRow, col_index: usize) -> Result<Option<MyDynamicValue>> {
    let value_ref = row
        .try_get_raw(col_index)
        .map_err(|err| Error::RunQueryError)?;
    Ok(match value_ref.type_info().name() {
        "VARCHAR" => row
            .try_get::<Option<String>, usize>(col_index)
            .map_err(|err| Error::RunQueryError)?
            .map(|value| MyDynamicValue::String(value)),
        "INT4" => row
            .try_get::<Option<i32>, usize>(col_index)
            .map_err(|err| Error::RunQueryError)?
            .map(|value| MyDynamicValue::Integer(value)),
        "BYTEA" => row
            .try_get::<Option<&[u8]>, usize>(col_index)
            .map_err(|err| Error::RunQueryError)?
            .map(|value| MyDynamicValue::Bytea(value.to_owned())),
            // ... 更多重复分支
        name => panic!("UNHANDLED DATABASE TYPE: {name}"),
    })
}

#[derive(PartialEq, Debug, Clone)]
pub enum MyDynamicValue {
    String(String),
    Integer(i32),
    Bytea(Vec<u8>),
}

重复逻辑片段示例：

"VARCHAR" => row
    .try_get::<Option<String>, usize>(col_index)
    .map_err(|err| Error::RunQueryError)?
    .map(|value| MyDynamicValue::String(value)),

1. 定义转换Trait #

创建Trait抽象类型到MyDynamicValue的转换逻辑，同时封装Sqlx的Decode约束：

use sqlx::{Decode, Postgres, PgRow};

// 自定义错误类型，根据实际项目调整
#[derive(Debug)]
pub enum MyError {
    RunQueryError(sqlx::Error),
    UnsupportedType(String),
}

impl From<sqlx::Error> for MyError {
    fn from(err: sqlx::Error) -> Self {
        MyError::RunQueryError(err)
    }
}

trait TryIntoMyDynamicValue<'a>: Decode<'a, Postgres> + Sized {
    fn into_dynamic(self) -> MyDynamicValue;
}

2. 为目标类型实现Trait #

为需要支持的数据库类型实现上述Trait，对应到MyDynamicValue的不同变体：

impl<'a> TryIntoMyDynamicValue<'a> for String {
    fn into_dynamic(self) -> MyDynamicValue {
        MyDynamicValue::String(self)
    }
}

impl<'a> TryIntoMyDynamicValue<'a> for i32 {
    fn into_dynamic(self) -> MyDynamicValue {
        MyDynamicValue::Integer(self)
    }
}

impl<'a> TryIntoMyDynamicValue<'a> for &'a [u8] {
    fn into_dynamic(self) -> MyDynamicValue {
        MyDynamicValue::Bytea(self.to_owned())
    }
}

// 新增类型时只需添加对应实现

3. 构建类型映射表 #

用静态数组存储数据库类型名称到转换函数的映射，避免重复的match分支：

type ValueConverter<'a> = Box<dyn Fn(&PgRow, usize) -> Result<Option<MyDynamicValue>, MyError> + 'a>;

fn get_converters<'a>() -> &'a [(&'static str, ValueConverter<'a>)] {
    static CONVERTERS: &[(&str, ValueConverter)] = &[
        ("VARCHAR", Box::new(|row, idx| {
            row.try_get::<Option<String>, _>(idx)
                .map(|opt| opt.map(TryIntoMyDynamicValue::into_dynamic))
                .map_err(MyError::from)
        })),
        ("INT4", Box::new(|row, idx| {
            row.try_get::<Option<i32>, _>(idx)
                .map(|opt| opt.map(TryIntoMyDynamicValue::into_dynamic))
                .map_err(MyError::from)
        })),
        ("BYTEA", Box::new(|row, idx| {
            row.try_get::<Option<&[u8]>, _>(idx)
                .map(|opt| opt.map(TryIntoMyDynamicValue::into_dynamic))
                .map_err(MyError::from)
        })),
        // 新增类型时添加新的映射项
    ];
    CONVERTERS
}

4. 重构核心函数 #

利用映射表统一处理类型转换，消除重复代码：

fn get_row_value(row: &PgRow, col_index: usize) -> Result<Option<MyDynamicValue>, MyError> {
    let type_name = row
        .try_get_raw(col_index)
        .map_err(MyError::from)?
        .type_info()
        .name();

    get_converters()
        .iter()
        .find(|(name, _)| *name == type_name)
        .ok_or_else(|| MyError::UnsupportedType(type_name.to_string()))?
        .1(row, col_index)
}

关键说明 #

• 为何From<dyn Decode>不可行：Decode trait包含关联类型Database，且方法签名存在泛型约束，不符合对象安全要求，无法直接使用dyn Decode作为 trait 对象。
• 扩展性：新增数据库类型时，只需两步：为对应Rust类型实现TryIntoMyDynamicValue，在CONVERTERS数组中添加类型名称与转换函数的映射，无需修改核心逻辑。

内容的提问来源于stack exchange，提问作者Guillaume Geoffrey Attia