sqlx能否为PostgreSQL的JOIN查询自动推断非NULL输出列？

问题背景 #

现有两个PostgreSQL表a和b，所有列均为非NULL类型。使用Rust代码通过sqlx::query_as!宏执行INNER JOIN查询：

struct MyStruct {
  id: i32,
  name: String,
}

sqlx::query_as!(MyStruct, r"
  SELECT a.id, b.name
  FROM a
  INNER JOIN b on a.id = b.a
")

运行时出现类型不匹配错误：

mismatched types
expected type i32
   found enum std::option::Option<i32> rustcE0308
macros.rs(552, 9): Actual error occurred here
macros.rs(552, 9): Error originated from macro call here

已知可通过foo as "foo!"语法强制标记列非NULL，但希望避免这种写法：

sqlx::query_as!(MyStruct, r#"
  SELECT a.id as "id!", b.name as "name!"
  FROM a
  INNER JOIN b on a.id = b.a
"#)

sqlx文档相关说明：

在大多数情况下，数据库引擎可以告知列是否可为NULL，query!()宏会据此调整返回结构体的字段类型。
对于PostgreSQL，这仅适用于直接来自实际表的列，因为实现需要查询表元数据来判断列是否有NOT NULL约束。无NOT NULL约束的列或表达式结果列会被视为可空，因此使用Option而非T。

疑问 #

1. JOIN是否属于会阻止数据库引擎推断非空性的表达式？
2. 是否有sqlx特性可让其正确推断这些列非空？

解答 #

1. JOIN是否阻止非空推断 #

是的。JOIN查询中引用的列会被sqlx归类为表达式结果列，而非直接来自单表的列。尽管原表列带有NOT NULL约束，但sqlx当前的元数据查询逻辑无法追踪跨表关联后列的原始非空属性，因此会默认将它们标记为可空类型（Option<T>）。

2. 是否有sqlx特性支持正确推断 #

目前sqlx没有原生特性可以自动处理JOIN场景下的非空类型推断。核心原因：

• sqlx的非空推断依赖于直接查询单表的元数据，JOIN后的列属于跨表关联结果，无法通过简单的元数据查询确认其非空性（虽然INNER JOIN逻辑上保证关联列存在，但sqlx的静态分析并未深入到SQL语义层面做推导）。
• 官方文档明确说明，只有直接来自实际表的列才能触发基于元数据的非空推断，表达式（包括JOIN后的列引用）都会被视为可空。

替代方案 #

如果不想使用as "foo!"语法，可选择以下两种方式：

• 修改结构体字段为Option<T>：后续通过unwrap()或expect()做非空断言（需确保INNER JOIN逻辑确实保证数据非空）。
• 使用sqlx::query手动映射：放弃query_as!的自动映射，手动将查询结果转换为结构体：

sqlx::query(r"
  SELECT a.id, b.name
  FROM a
  INNER JOIN b on a.id = b.a
")
.map(|row: sqlx::postgres::PgRow| MyStruct {
  id: row.get(0),
  name: row.get(1),
})

内容的提问来源于stack exchange，提问作者Cypress Frankenfeld