AWS Amplify 中 PostgreSQL 自动生成类型的 list() 命令问题解析

问题背景

在使用 AWS Amplify 的 Gen 2 版本时，开发者遇到了一个关于 PostgreSQL 数据库连接的有趣现象。当通过 npx ampx generate schema-from-database 命令自动生成数据模型后，发现对于某些特定类型的模型，list() 命令会返回空数组，而其他操作如 create、update 和 delete 则工作正常。

问题现象

具体表现为：

1. 当模型的主键标识符(identifier)为字符串类型(a.string().required())时，list() 命令返回空数组
2. 当模型的主键标识符为整数类型(a.integer().required())时，list() 命令正常工作
3. 通过 Data Manager 可以确认数据库中确实存在数据

技术分析

深入分析发现，问题的根源在于自动生成的 GraphQL 查询语句的差异：

对于字符串类型主键的模型，生成的查询会包含主键作为参数：

query ($user_id: String, $sortDirection: ModelSortDirection, $filter: ModelAuthFilterInput, $limit: Int, $nextToken: String) {
  listAuths(
    user_id: $user_id
    sortDirection: $sortDirection
    filter: $filter
    limit: $limit
    nextToken: $nextToken
  ) {
    items {
      user_id
      username
      password
      access
      created
      updated
    }
    nextToken
    __typename
  }
}

而对于整数类型主键的模型，查询则不会包含主键参数：

query ($filter: ModelOne_offFilterInput, $limit: Int, $nextToken: String) {
  listOne_offs(filter: $filter, limit: $limit, nextToken: $nextToken) {
    items {
      id
      token
      amount
      used
      expiry_date
      created
      updated
    }
    nextToken
    __typename
  }
}

这种差异导致字符串主键模型的查询总是尝试匹配一个空的 user_id 参数，从而返回空结果。

解决方案

经过 AWS Amplify 团队的验证，最新版本的 @aws-amplify/backend 和 @aws-amplify/backend-cli 包已经修复了这个问题。开发者可以通过以下步骤解决：

1. 更新相关包到最新版本：

npm i @aws-amplify/backend @aws-amplify/backend-cli

2. 特别更新数据模式包：

npm update @aws-amplify/data-schema

最佳实践建议

1. 定期更新 Amplify 相关依赖，特别是当使用预览功能时
2. 对于自动生成的数据库模型，建议进行全面的 CRUD 操作测试
3. 遇到类似问题时，检查生成的 GraphQL 查询语句与实际需求是否匹配
4. 考虑在开发环境中使用 Data Manager 验证数据是否存在，以排除数据库连接问题

总结

这个问题展示了在使用 ORM 或自动生成的数据库模型时可能遇到的微妙问题。AWS Amplify 团队已经在新版本中修复了这个问题，开发者只需保持依赖更新即可避免此类问题。这也提醒我们，在使用自动化工具时，理解其生成的代码逻辑对于问题排查至关重要。